diff --git a/Modify.ipynb b/Modify.ipynb
index 6ddedca..4e97a67 100644
--- a/Modify.ipynb
+++ b/Modify.ipynb
@@ -3,8 +3,8 @@
   {
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2025-06-25T06:49:47.268865Z",
-     "start_time": "2025-06-25T06:49:25.453526Z"
+     "end_time": "2025-06-26T01:24:15.197491Z",
+     "start_time": "2025-06-26T01:24:15.193517Z"
     }
    },
    "cell_type": "code",
@@ -42,20 +42,20 @@
     "\n",
     "    # Seed all GPUs with the same seed if using multi-GPU\n",
     "    torch.cuda.manual_seed_all(seed)\n",
-    "    torch.backends.cudnn.benchmark = True\n",
+    "    torch.backends.cudnn.benchmark = False\n",
     "    torch.backends.cudnn.deterministic = True\n",
     "\n",
     "seed_torch(42)  # 使用更常见的随机种子值"
    ],
-   "id": "9cf2e2b5d8a6892d",
+   "id": "ce18f89946d8c18b",
    "outputs": [],
-   "execution_count": 1
+   "execution_count": 16
   },
   {
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2025-06-25T06:49:47.819194Z",
-     "start_time": "2025-06-25T06:49:47.811528Z"
+     "end_time": "2025-06-26T01:24:15.226309Z",
+     "start_time": "2025-06-26T01:24:15.218309Z"
     }
    },
    "cell_type": "code",
@@ -97,7 +97,7 @@
     "    patience = 10  # 早停耐心值\n",
     "    counter = 0   # 计数器\n",
     "\n",
-    "    scheduler = optim.lr_scheduler.ReduceLROnPlateau(optimizer, mode='max', factor=0.5, patience=10)\n",
+    "    scheduler = optim.lr_scheduler.ReduceLROnPlateau(optimizer, mode='max', factor=0.5, patience=5)\n",
     "\n",
     "    with tqdm(total=num_epochs) as pbar:\n",
     "        for epoch in range(num_epochs):\n",
@@ -188,15 +188,15 @@
     "        print(f'Test Acc: {test_acc:.3f}')\n",
     "    return test_acc"
    ],
-   "id": "3a0bcd81cba9b9d4",
+   "id": "68d3cc00181d6b0f",
    "outputs": [],
-   "execution_count": 2
+   "execution_count": 17
   },
   {
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2025-06-25T06:49:47.839016Z",
-     "start_time": "2025-06-25T06:49:47.826008Z"
+     "end_time": "2025-06-26T01:24:15.272943Z",
+     "start_time": "2025-06-26T01:24:15.238946Z"
     }
    },
    "cell_type": "code",
@@ -206,7 +206,6 @@
     "    transforms.RandomHorizontalFlip(),  # 随机水平翻转\n",
     "    transforms.RandomRotation(10),     # 随机旋转±10度\n",
     "    transforms.ColorJitter(brightness=0.2, contrast=0.2),  # 颜色调整\n",
-    "    transforms.Resize((18, 18)),       # 调整大小为18x18\n",
     "    transforms.ToTensor(),           # 转换为张量\n",
     "    transforms.Normalize((0.5,), (0.5,))  # 归一化到[-1, 1]\n",
     "])\n",
@@ -215,11 +214,10 @@
     "    transforms.RandomHorizontalFlip(),  # 随机水平翻转\n",
     "    transforms.RandomRotation(10),     # 随机旋转±10度\n",
     "    transforms.ColorJitter(brightness=0.2, contrast=0.2),  # 颜色调整\n",
-    "    transforms.Resize((16, 16)),       # 调整大小为16x16\n",
     "    transforms.ToTensor(),           # 转换为张量\n",
     "    transforms.Normalize((0.5,), (0.5,))  # 归一化到[-1, 1]\n",
     "])\n",
-    "train_dataset = FashionMNIST(root='./data/notebook2', train=False, transform=trans1,download=True)\n",
+    "train_dataset = FashionMNIST(root='./data/notebook2', train=True, transform=trans1,download=True)\n",
     "test_dataset = FashionMNIST(root='./data/notebook2', train=False, transform=trans1,download=True)\n",
     "\n",
     "# 定义训练集和测试集的比例\n",
@@ -237,15 +235,15 @@
     "valid_loader = DataLoader(valid_dataset, batch_size=64, shuffle=False, drop_last=True)\n",
     "test_loader = DataLoader(test_dataset, batch_size=64, shuffle=False, drop_last=True)"
    ],
-   "id": "7228ef013f3a1fdd",
+   "id": "1a2fb592a20d3fe1",
    "outputs": [],
-   "execution_count": 3
+   "execution_count": 18
   },
   {
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2025-06-25T06:49:47.845266Z",
-     "start_time": "2025-06-25T06:49:47.841750Z"
+     "end_time": "2025-06-26T01:24:15.287304Z",
+     "start_time": "2025-06-26T01:24:15.284079Z"
     }
    },
    "cell_type": "code",
@@ -253,15 +251,15 @@
     "singlegate_list = ['rx', 'ry', 'rz', 's', 't', 'p', 'u3']\n",
     "doublegate_list = ['rxx', 'ryy', 'rzz', 'swap', 'cnot', 'cp', 'ch', 'cu', 'ct', 'cz']"
    ],
-   "id": "d9e483c48405660",
+   "id": "8cdb66e075ac60ed",
    "outputs": [],
-   "execution_count": 4
+   "execution_count": 19
   },
   {
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2025-06-25T06:49:47.895431Z",
-     "start_time": "2025-06-25T06:49:47.889872Z"
+     "end_time": "2025-06-26T01:24:15.305490Z",
+     "start_time": "2025-06-26T01:24:15.300543Z"
     }
    },
    "cell_type": "code",
@@ -310,15 +308,15 @@
     "        exps = exps.reshape(x.shape[0], 3, w, w)\n",
     "        return exps"
    ],
-   "id": "f24b62bd70ab89eb",
+   "id": "248357d84b938ba9",
    "outputs": [],
-   "execution_count": 5
+   "execution_count": 20
   },
   {
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2025-06-25T06:49:48.381594Z",
-     "start_time": "2025-06-25T06:49:47.897435Z"
+     "end_time": "2025-06-26T01:24:15.436208Z",
+     "start_time": "2025-06-26T01:24:15.327298Z"
     }
    },
    "cell_type": "code",
@@ -326,27 +324,27 @@
     "net = RandomQuantumConvolutionalLayer(nqubit=4, num_circuits=3, seed=1024)\n",
     "net.cirs[0].draw()"
    ],
-   "id": "f97107a549dee68",
+   "id": "22520c5a291ddfb9",
    "outputs": [
     {
      "data": {
       "text/plain": [
        "<Figure size 1207.22x367.889 with 1 Axes>"
       ],
-      "image/png": "iVBORw0KGgoAAAANSUhEUgAAA7UAAAEvCAYAAACaO+Y5AAAAOnRFWHRTb2Z0d2FyZQBNYXRwbG90bGliIHZlcnNpb24zLjEwLjAsIGh0dHBzOi8vbWF0cGxvdGxpYi5vcmcvlHJYcgAAAAlwSFlzAAAPYQAAD2EBqD+naQAAZPBJREFUeJzt3XlYVOXbB/DvDAPDsMgmCoKIgoCQW+5b7qampqlZabbZXpaallqZabZopVm9mfbLzCXNPXNPS1xxS3EBEdn3HQaGgVneP8hJAhWQmcOc8/1cl1dxzjMz98C572fuOZvMaDQaQURERERERGSF5EIHQERERERERFRXbGqJiIiIiIjIarGpJSIiIiIiIqvFppaIiIiIiIisFptaIiIiIiIislpsaomIiIiIiMhqsaklIiIiIiIiq8WmloiIiIiIiKwWm1oiIiIiIiKyWmxqiYiIiIiIyGqxqSUiIiIiIiKrxaaWiIiIiIiIrBabWiIiIiIiIrJabGqJiIiIiIjIarGpJSIiIiIiIqvFppaIiIiIiIisFptaIiIiIiIislpsaomIiIiIiMhqsaklIiIiIiIiq8WmloiIiIiIiKwWm1oiIiIiIiKyWmxqiYiIiIiIyGqxqSUiIiIiIiKrxaaWiIiIiIiIrBabWiIiIiIiIrJabGqJiIiIiIjIarGpJSIiIiIiIqvFppaIiIiIiIislkLoAIjuxenTp2s1Pjs7G1u3bsUjjzyCxo0b1+gxXbp0qUtoRGQhtakDdakBAOsANWycC4lI6nMh99SSpGRnZ2PVqlXIzs4WOhQiEgBrABHzgEjqxFgD2NQSERERERGR1WJTS0RERERERFaLTS0RERERERFZLTa1JCnOzs4YOnQonJ2dhQ6FiATAGkDEPCCSOjHWAJnRaDQKHQRRXdX2io910ZCv9EZErANEzAEiknod4J5akhStVoukpCRotVqhQyEiAbAGEDEPiKROjDWATS1JSlxcHMaOHYu4uDihQyEiAbAGEDEPiKROjDVAIXQAVD2j0Qidxnq+PVGolJDJZEKHIXrWtl2IFbd3sgRrzHfmBjVU1phPltJQ8lasf6OG8vsVOza1DZROo8W6gElCh1FjE2PXwtbBXugwRM/atgux4vZOlmCN+c7coIbKGvPJUhpK3or1b9RQfr9ix8OPiYiIiIiIyGqxqSUiIiIiIiKrxcOPSVJCQkIQEREhdBhEJBDWACLmAZHUibEGcE8tERERERERWS02tSQpCQkJePbZZ5GQkCB0KEQkANYAIuYBkdSJsQawqSVJ0Wg0uHTpEjQajdChEJEAWAOImAdEUifGGsCmloiIiIiIiKwWLxQlIl49wjB06/xKy8qLNSi8kYbYzUdw9YfdMOoNAkVHUhT4aD/0XvYajr7xNa5v+rPKeidfT4w7/X+4vvEwjr75jeUDJBIZzgNEDQ/nQiLzY1MrQnHbjyHp4BlAJoPK0xWB4/ui6/yn4dLaBydmrhA6PCIiMjPOA0REJCVsakUo93IcbmwJN/0cvXofRocvRdATA3Hukw3Q5hQKGJ2wvL29MX/+fHh7ewsdChEJQCo1gPMA3YlU8oCIqifGGsBzaiVAp9Ei+9x1yORyNGrRVOhwBOXi4oJhw4bBxcVF6FCISABSrQGcB+hWUs0DIqogxhrAplYinP0rPsSU5hUJHImw8vLy8OuvvyIvL0/oUIhIAFKuAZwH6CYp5wERibMGsKkVIRuVEkp3Zyg9GsE1xA/dFk2BR9tWyDofg6K4dKHDE1RGRgYWL16MjIwMoUMhIgFIpQZwHqA7kUoeEFH1xFgDJHFObXZ2Nj755BNs27YNKSkp8PT0xKOPPoqFCxfi+eefx7p167By5UpMmTJF6FDrRYfp49Fh+vhKyxL2nMLJt1cKFBEREVkS5wEiIpIS0Te1Fy9exJAhQ5CRkQFHR0eEhoYiJSUFX3zxBeLj45GYmAgAaN++vcCR1p9r6w4ibscxyBU2cA32Q9vXR0PV2BW60jLTGLmdAiP3fYYb244i8qutpuW9l74Ke09XHJz4kRChk0QZjUahQyASFc4DRNaHc2HNtBzdC2Evj4Jra1/oNFqk/nURZxetRXFyttChkYBEffhxdnY2RowYgYyMDMyaNQvp6ek4d+4cMjIysGDBAmzduhV///035HI57rvvPqHDrTdF8elIC49EyuG/cfm7nfjjyY/R+P5A9PjkedMYQ5kO4VOXo93UMXALbQEA8BvaBb6DO+PY9G+FCp1E5uYHaBt7u2rX2zgoAQD6Wz5oE9G94zxA1HBwLqw/Ic8MRd//mwZ9aRlOf/ATrqz8Hc36tsPwnR9B1dRN6PBIQKJuaqdOnYqkpCRMnz4dn376KZycnEzr5s6di5CQEOh0OgQFBUGlUgkYqXllnYvBjS3haPVIH3je39q0PDcyDpe+2YE+X70OB2939PjsRZyaswqaDPGcNP5fDg4O6NatGxwcHIQORRLUiZkAANcg32rXu7auWF70zzgic5NqDeA8QLeSah4IhXNh/VC6OaHTnInIvhiLvY/MQ/Sa/bi4dAsOPPERHJq6oePMCUKHaDXEWANE29ReuXIFGzduRJMmTbBgwYIq62UyGTp16gSg6qHHcXFxGDVqFJydneHm5obJkycjJyfHInGby4UvN8Og06PjrMcqLb/41VYYynUYdWAx0o9fRtyOYwJFaBl+fn5Yvnw5/Pz8hA5FEnIib0CdkoWWo3tV+QZVbqtAm2eHwWgwIGn/GYEiJKmRcg3gPEA3STkPhMC5sH40f7ArbJ1UuLpqN4x6g2l5zoVYZJy8Cv9RPSG3Ff2ZlfVCjDVAtE3tunXrYDAYMGnSpNt+C6FUVhzucWtTW1RUhP79+yM5ORkbNmzA999/j/DwcIwYMQIGg6Ha57EGRfHpiNtxDM36tkeTbm1My416AzIiomDv4YLrGw8LGKFl6PV6qNVq6PV6oUORBKPegJNvr4StswMePvQFOs2dhKBJg9Bu2jiM3P8ZvHqGIfLr7SiMTRU6VJIIKdcAzgN0k5TzQAicC+uHZ8dAAEDmmegq6zLPRMPO2QEugT6WDssqibEGiLap/eOPPwAA/fv3v+2Y5ORkAJWb2u+//x4pKSnYvn07RowYgfHjx2PdunU4efIkdu7cad6gzezisi0w6PWVDs9o0iUYrR8fgKs/7EbXD5+57fkeYhETE4MBAwYgJiZG6FAkI/mPc9g96l2kH7uEwEf7ovuiKbjvxZEozSnEny98jnMfrxc6RJIQqdcAzgMEMA+EwLnw3jl4uQMAStJyq6wrTq04otLB292iMVkrMdYA0e6jT0hIAAD4+/tXu16r1eLUqVMAKje1u3btQu/evSvtju/Zsyf8/f3x22+/YfTo0bWOpXPnzkhPr919AW2NcsxD11o9Jv3EZaz2Hnfb9QUxKVjj++8HGYVKid7LXsO5ResR9dM+DNv2ITrNnYiI936s1esCQFDrIJTLLL8ne9y427/f6mRmVpyvsmfPHpw9e7ZGjxkzZkyt4zKXumwXDUHOhVj8+cLnQodRb4Ta3ql6takDdakBgDB1wNrmAYC5IRSpzYV10RDmz4Y6FzaUvL3b38hG9c8FtbTlVdbdXKb4Z0xDYqnfrxjmQi8vL5w5U7fD8EXb1BYXFwMASktLq12/du1aFBQUwMPDAz4+/x6qcOXKFYwfP77K+NDQUFy5cqVOsaSnpyMlJaVWj7GT2QBN6/RyNdZl/lNQJ2UhavVeAMDRN77GqINLkLg7AuknLtfquVLTUlFmtPwhDDf/zjWl0WhM/63pY2v7tzMnS2wXdHdCbe9UvdrUgbrUAECYOmBt8wDA3BCK1ObCuuD8eXsNJW/v9jfSa7QAABulbZUrRSv+OcJE98+YhsRSv1+xzoU1Jdqm1svLC0VFRThz5gw6d+5caV1SUhJmz54NAGjXrl2ldXl5eXB1da3yfG5ubrh+/XqdY6ktW6McMOOXOj79O6DlqF7YMXCGaVlRQgbOfrQOvZa+gh0DZkBXXP0XAtVp5t1MkG/5HB0dazX+ZuKqVKoaP/bWLz2EZu7tgmpGqO2dqlebOlCXGgAIUwesbR4AmBtCkdpcWBecP2+voeTt3f5GJekVhx07eLujKK7yEZA3Dzuu7tBkoVnq9yuGubAuPdNNom1qBw0ahJiYGCxatAgDBgxAUFAQAODkyZN48sknkZ+fD6DqlY+Biisj/9e93BC7LrvRy0tKsS5gUp1f825SDv+N9SFPVVketXqv6Rv72rgWcw22Dvb1EVqtnD59ulbjo6KisGHDBgwbNgwhISE1eszSpUvrEJl5mHu7oJoRanun6tWmDtSlBgDC1AFrmwcA5oZQpDYX1gXnz9trKHl7t79R9t/XETx5CJp0Dq7S1DbpHIxytQYF1xvenkRL/X7FOhfWlGgvFPXOO+/Azc0NSUlJCAsLQ9u2bREYGIgePXqgTZs26N27N4CqTa2bmxvy8qreny8/Px/u7jz53NoFBgZi3759CAwMFDoUIhIAawAR84CsU+K+0ygvKUWbKcMhs/m3hfFoH4Cm3dsgbudxGMp1AkZoPcRYA0Tb1Pr5+SE8PBzDhw+Hvb09EhIS4OnpiVWrVmHHjh2IiooCULWpbdOmTbXnzl65cgVt2rSpspysi0KhgJubGxQK0R6kQER3wBpAxDwg66TNLcK5j9ejcbsADN06H0FPDka7Nx7B4PVzockqwN+LNwodotUQYw0QbVMLAGFhYfj9999RVFSEwsJCnDhxAs899xzy8vKQlpYGhUKB0NDQSo8ZMWIEjh49iqSkJNOykydPIj4+HiNHjrT0W6B6lpycjBkzZphu50RE0sIaQMQ8IOt1ddVuHHl1GRQqJbrOfxqhL4xE2pFI7B4513TOLd2dGGuAqJva24mMjAQAhISEQKmsfOnvF154Ad7e3hg9ejR27dqFzZs344knnkDXrl3x8MMPCxEu1SO1Wo3w8HCo1WqhQyEiAbAGEDEPyLrd2BqO34bMwtpWE/FL2DP46+UvoU7KFDosqyLGGiDppra6i0Q1atQIhw4dgre3Nx577DFMmTIFPXv2xK5duyCXS/LXRURERERE1GCJ50DqWrhTUwsAAQEB2LVrlyVDumfjIr6FXltuum/XxeXbEL/zeKUxTr6e6L3sNbjf54/CuHTsGvq2aZ13n7bo8v5k08/2jV2gycrHb0NmWeYN0D1xbumFPsteh9LdGWWFxTj65jcouFb9ISWuIX7o/tFzsPd0gUwmw9mP1yNx9yk4+XrikRNfIz8q0TT28JQlKErIAAD0WzkDTToHw8HLHWsDJkFXUrtbfbSfPh4dZ07A9n7TkB+dVO2Yzu89CZ+B98OoN0CbV4Tjb32Hovj0asfe5NS8CfqtmgG5XA6ZjRwF11NxfOZ3KCuo5r5rMhm6LXgGPgPvB4xGXF6xC9E/7as0ROnRCKMPf4GMU1fx5/Of1+o9EgnFrpEDhm6Zb/rZRqWEc4um+KXtcyjLr/xNfOvHB6Dta2MAuQxpRyNx8p2VMOoNnAdIUmyUtuj73TS4tPaFXqOFJisfJ2Z9D3VyVpWxAeP7IuyFEaafHZp5IOPkVRx+bvEdX0PhYI/+P7wFj3atAAC/hD1b4/jsXBzRbdEUeHYIhEGvR9Le0zi7aF2VcR7tA9BtwbNwv88fyX+cqzRv3elzH5HYSLqp/e89aq3d4SlLbtssAECZWoNzn26AnbMDOsycUGldWngkdg6eafp54JrZSD92yWyxUv3q+dmLuLb2AK5v+hMtHuqOXp+/jN0j51YZZ6Oyw4AfZ+HoG18jMyIKMhs57FydTOvLCosrbQe3il6zHyffWYnHLv2v1vG5t20Jz05B1X5YuKn5g13QtHsodg56C0adHu3eHIv7Zz+Bv1784o7PXZKRiz0Pv2f6Qqfrh0+j/bRxOP3BT1XGBox7AC5BzbGt11TYNnLAqP2fIf1YJAqup5rG9Pj4eST/cQ62Tqpav08ioZQVllTK3bCXRsGrR2iVhtapeRN0nPUYdg6eidLsAgxY/TZaPzEQ134+wHmAJCf65wNIOXQeABDyzFD0WPISDjy2oMq42F//Quyvf5l+fvjQ57ix9chdn9+g0+HSNzugzSvCkE3v1yq2Xl++gsyIaIS/ugwAoGriWu04TWYeIt7/Ee5tW6LZA5U/197pcx+R2EjyeNrjx4/DaDTiwQcfFDoUiyrLVyMzIgq6Eu0dx6mausG7932I3fzXHcdZI09PT7zxxhvw9PQUOpR6Y+/RCB5tWyF2S8UEm/D7STj7NYGTb9X32GpMH2SdvYbMiIqrfxv1BmhzCmv0OmnhkSit4dhbye0U6L5oCk6+sxK4y/2ebewUsFHaAgBsnVQoScu56/MbynSmhlYml0PhYA+jofrX8R/VE9Fr9sNoMKAsX424ncfR8uHepvWtHukDTVY+Mk5UvQI6iYMYa0B1Ah/rj5gNf1RZ3mJEdyTsiUBpdgGAii+rWo3uXWWcmOcBkk4e3IleW25qaAEg61wMnP2a3PVxjTsEwt7TFYn7ztx1rKFMh7SjkdUfOXQHzv5e8GjbCpdX/GZapsnMr3ZsSVousv++DoO2vMq6mn7uI+kRYw2Q5J5asXrgmzcgk8uQdS4GZz9eX+Nm5b8CH+2H5EPn69TANHQeHh6YOHGi0GHUK0efxijJyIVRbzAtU6dkw9G3cZU9o65BvtCXlmHgmtlw9HZH7tVEnJ7/k2lbsXVSYcSeTyCzkSNxTwQuLtsKo8GAe9Fx5mO4sSX8rhdxSNp/Bl49wzDh4iro1BoUp+di75h5NXoNua0CI3Z/DEdfT+RdScAfT31S7Tgnn8YovuV3ok7KQuP2AQAqPsSHvjgCe8fMg/+I7jV8d2RtxFgD/suzUxDs3ZyQdOBslXVVciA5C44+jauME/M8QNLIg9pq89zwanPmv1o/MQCxm/+CUac3WyyuQb4oTs1Gj89eQOP2ASjNLcLZhWuReynObK9J0iLGGiDJPbVitGfM+9g56C3sHDIL2jw1+ix7rc7PFTihP2LWV/2GXwwKCwtx8OBBFBaK64Paf3eAymSyasfJFTbw6dcBJ2atwM7BM1Gcmo3ui6YAAEoy8/Dr/S9i17B3sO/RD9G0WxuEvXRvt7Hy7BSExh0CELV6713HerRrBZdAH/za8QVs7PAC0sIj0X3RczV6HUO5DjsHz8TGdlNQEJuC4MlDbjvWeMsv69bfU88lL+HsgrW1PleYrItYa8CtWj8+ANd//avSF123qpQDt3kOMc8DJI08qI22Ux9Bo5ZeOPfJ+juOs1HZoeWoXojZcMis8chtbeDZKQhx247ityGzcPm7nRj40zuQ2fBjO9UPMdYAZodIFKdkAwCMOj2urNyFpt3a1Ol5mnZvA4VKidQ/L9RneA1Gamoq5syZg9TU1LsPthLFKdlw9HavNNk5NvNAcXJ2lbHq5GykHb9supfbja3h8OwYCKDiMKmbe2XK8tWI+eVQnbejm7x6hMIl0AfjIr7FuIhv4eDtgcEb3oXPgI5VxgY+2g/pxy6hrLAEMBoRu+lPePW6r1avZyjX4fovhxEw7oFq16tTsuHU/N/Dyxx9G5typ0mnIPT64mWMi/gWnedNhs+Ajhi84d1avT41fGKsAbdSqJTwH9UT13+p/kN31RzwNOXATWKfB0j8eVAbYS+NQovh3XBw4kfQa8ruONZ/RA/kxyTf9kKM9UWdnI2S9FykH78MAEj98wLkdgo4enuY9XVJOsRYA9jUioBCpYRdIwfTzy3H9EZOHQ9Raf3YAFzfdPieDzklyynNKUTupXgEjK1o5Fo81B3qpKxqL8oU/9txNO4QYLoIkk//Dsi9nACg4txcmcIGQMV5sC2Gd6/xdtT7q9fhN6xrleWRX2/Hpo4vYHPXV7C56ysoScvBgccXVjqP6aaixAx492lriqH5kM7Iu+VKzGPCl8HBy73K4xx9GkOh+ud+0zIZ/Ef1RO7VxCrjACDhtxMIfnIwZPKKC2S1fLgX4nYeAwBsCH3GFOeZ+WuQcug8Djy+sEbvn6ih8B/ZA3lXEypd/OxWCb+fRIthXWHf2AUAEDx5COJ2HKs0hvMASUXoiyPQckwv7J/wYcUXqnfR+rEB1e6lvd38dDdDNs1D4w6BVZbnXIhFeZEGbm1aAKi4wjEAFP/zhTTVjlePMDydtrnSv4nXf8bI/Z8h9IUR3AMuEjynVgTsPV3Qf9VMyGzkkMmAooRMHH19OYCKQyqT9p9B0v4zkNspMPbEN7BRKmDr7IDxZ1cgdstfOLeo4nAbhaM9WjzUHTsGzhDy7VAdHJ+1Ar2Xvoq2Ux9BuVqDo1OXm9bdug0Up2Qjcvk2DN+1CEadHiXpuTg+8zsAQJNubdBx5gQY9QbIFDZIPxqJi8u2mJ5nwOq34dG24rYEY44uQ1FcOvaOrTjn1aNdK1z9YXet4741tqgf98K1tS9GH/4C+jIdNJl5ODFzBYCK2+wo3Zygza96k3DX4OboNHcSAEAmlyEnMg4R7/5gWj/qwGIcmLQImow8xG4+Ao8OgRhz7CsAwKVvd6AgJqXWcRM1VK2fGIiY9ZU/dN+aZ+rETJxfsgnDdy6ETC5H2tFIXLvlMGPOAyQVDt7u6PrB0yiMT8fQzR8AAPRlOvz+0GwAlfMGAJxbNIVHu1Y4OPnjSs9zp/kJAEbu/wyqJm6wc3HE+LMrkH78EsJfXw6ZXA73sBYovs0FEY+++Q16fv4SbJR20GvLcHjKEtN5vIPWzsH5xRuRcyEWzi2aYujWD6FQ2cHG3g7jz67Axa+2IvqnfXf93Cc1cduPIengGUAmg8rTFYHj+6Lr/Kfh0trH9HmDrJfMaLzL5UhJEOUlpVgXMEnoMGpsYuxa2DrYW/x1T58+XavxUVFRmDx5MtasWYOQkJAaPaZLly51Cc0sGuJ2YefqhH7fTcP+am6DUF/8hneDa5AvLi7dcvfBFiDU9k7Vq00dqEsNAISpAw0x3++GuSEMqc2FdWGufKrr/OTetiVCnnoQx9/6rt5jqq2Gkrfm+ht59QjD0K3zcfajtYj8ertpuUKlxOjwpXD09sAv7abU+QKrd2Op369Y58Ka4p5akhSlUong4GAolUqhQxGNsny1WRtaAEjcfQqJu0+Z9TVIGlgDiJgH9amu81NuZFyDaGilTKfRIvvcdTiN9ESjFk2RJaGrvYuxBrCpJUlp2bIlfv75Z6HDICKBsAYQMQ+IbnL2bwoAKM0rEjgSyxJjDWBTS0REREREomajUkLp7mw6pzZ48hB4tG2FrPMxKIpLFzo8uke83BdJSnR0NHr16oXo6GihQyEiAbAGEDEPSJo6TB+Pxy//iMcv/Q+jD3+BNs8MRcKeUzj01KdCh2ZxYqwB3FNLkmI0GlFeXg5eH41ImlgDiJgHJE3X1h1E3I5jkCts4Brsh7avj4aqsSt0pf/en7jf9zNgNBrx14tfmJbZuTph9J9f4syHa+A3tOsd19/YGm7R91RXYqwBbGobKIVKiYmxa4UOo8ZM9wkls7K27UKsuL2TJVhjvjM3qKGyxnyyFKnkbVF8OtLCIwEAKYf/RmbEVQzbuRA9PnkeR15dBgA48fYKPHz4S7Qc3Qtx2yvu4d190RRkRkThxtZwpBw+f8f1JBw2tQ2UTCZrEJdXp4aF2wWRdDDfieoP84n+K+tcDG5sCUfgo/1w9YfdyDoXA22eGsemf4s+y6ci/cQVNOkcDK+eYdjRfxoA3HU9CYfn1BIRERERkeRc+HIzDDo9Os56zLQs5dB5JOw6gQe+norunzyP4zP+D9o8dY3XkzDY1JKk+Pv7Y8OGDfD39xc6FCISAGsAEfOA6Kai+HTE7TiGZn3bo0m3Nqblpz/4Cc4tvZBy6DyS/zhX5XF3W9/QibEGsKklSbG3t0dAQADs7XkIEpEUsQYQMQ+IbnVx2RYY9Hp0nDnBtEyn0UKdkIm8q4nVPuZu6xs6MdYANrUkKWlpaVi4cCHS0tKEDoWIBMAaQMQ8IGlJP3EZq73HIfLr7dWuL4hJwRrfCdg37gOLxiUkMdYANrUkKQUFBdi5cycKCgqEDoWIBMAaQMQ8IJI6MdYANrVERERERERktXhLHyIiIiIiolvsHTvvntaTZXFPLREREREREVktNrUkKXK5HB07doRczk2fSIpYA4iYB0RSJ8YaIJ53QlQDBoMB58+fh8FgEDoUIhIAawAR84BI6sRYA9jUEhERERERkdViU0tERERERERWi00tERERERERWS02tSQpzs7OGDp0KJydnYUOhYgEwBpAxDwgkjox1gCZ0Wg0Ch0EUV2dPn3a7K/RpUsXs78GEdUd6wBJHXOAxKC8pBTrAiYJHUa9mxi7FrYO9mZ/HanXAYXQARBZklarRWZmJpo0aQKlUil0OERkYawBRMwDkh4bpS36fjcNLq19oddoocnKx4lZ30OdnFVpnJOvJ3ovew3u9/mjMC4du4a+bVrnGuKH7h9PgaqxCwzlOmSevYZTc3+AoUxn6bdzz8RYA3j4MUlKXFwcxo4di7i4OKFDISIBsAYQMQ9ImqJ/PoBtvadi5+CZSDpwFj2WvFRlTJlag3OfbsCRV5ZVWafXluHUnB+wrc8b2DloJuycHRD20ihLhF7vxFgD2NQSEREREZFo6bXlSDl03vRz1rkYOPs1qTKuLF+NzIgo6Eq0VdYVxaUj72oCAMBoMCD779hqn4OEwcOPGyij0QidpmpCNVQKlRIymUzoMKgecRsUlrX9/gHx/Q2EZjQCpXqho6gdexugvjYB5gAJjdugeLV5bjiSDpyt8+MVKiWCJg7EmYVr6zEquhdsahsonUZrVSfLW+okeLIcboPCsrbfPyC+v4HQSvVAn91CR1E74cMBVT19smAOkNC4DYpT26mPoFFLL+x7e0WdHi9T2KDviulI+fMCkvaZ/+JMVDM8/JiIiIiIiEQv7KVRaDG8Gw5O/Ah6TVmtHy9T2KDf99OhycxDxHv/M0OEVFfcU0uSEhISgoiICKHDICKBsAYQMQ9ImkJfHIGWY3ph/6MfoqywpNaPl9nI0fe7adDmqXH8re/MEKHliLEGsKklIiIiIiLRcvB2R9cPnkZhfDqGbv4AAKAv0+H3h2aj55KXkLT/DJL2n4HcToGxJ76BjVIBW2cHjD+7ArFb/sK5RevR8uFe8H+oO3Ivx2PUgcUAgIzT0Tg1Z5WA74xuYlNLkpKQkID58+dj3rx5aNGihdDhEJGFsQYQMQ9IekrScrHae1y1627d62oo0+HXTi9WO+7G1nDc2BpulvgsTYw1gOfUkqRoNBpcunQJGo1G6FCISACsAUTMAyKpE2MNYFNLREREREREVotNLREREREREVktnlMrIl49wjB06/xKy8qLNSi8kYbYzUdw9YfdMOoNAkVHUsBtkIiIpI5zIZHlsakVobjtx5B08Awgk0Hl6YrA8X3Rdf7TcGntgxMz63ajabHw9vbG/Pnz4e3tLXQoosZtkBoq1gAi5oGlcC6khkqMNYBNrQjlXo7DjS3/Xp0tevU+jA5fiqAnBuLcJxugzSkUMDphubi4YNiwYUKHIXrcBqmhYg0gYh5YCudCaqjEWAN4Tq0E6DRaZJ+7DplcjkYtmgodjqDy8vLw66+/Ii8vT+hQJIXbIDUUrAFEzAOhcC6khkKMNYBNrUQ4+1cUz9K8IoEjEVZGRgYWL16MjIwMoUORHG6D1BCwBhAxD4TEuZAaAjHWAEk0tdnZ2XjrrbcQEBAAe3t7NG/eHDNmzIBGo8GkSZMgk8mwatUqocOsNzYqJZTuzlB6NIJriB+6LZoCj7atkHU+BkVx6UKHRxLAbZCIiKSOcyGR5Yj+nNqLFy9iyJAhyMjIgKOjI0JDQ5GSkoIvvvgC8fHxSExMBAC0b99e4EjrT4fp49Fh+vhKyxL2nMLJt1cKFBFJDbdBy2k5uhfCXh4F19a+0Gm0SP3rIs4uWovi5GyhQ6MGpDw/Exnbl6Dg7G6UZSYAcjlsXZrAIaAT3Ho/CrcejwgdotncvBLt2Y/WIvLr7dWOmXj9Z+RcuIG9Y+dZNjgSNc6F9a/t62Pg0bYVPNq1gnOLplAnZWJz11eEDosaAFE3tdnZ2RgxYgQyMjIwa9YsvPfee3BycgIALFy4EO+99x4UCgXkcjnuu+8+gaOtP9fWHUTcjmOQK2zgGuyHtq+PhqqxK3SlZaYxcjsFRu77DDe2HUXkV1tNy3svfRX2nq44OPEjIUInkeA2aBkhzwxF90VTkBFxFac/+AlKd2eEPv8Qmnb/CLuGvQNNhnjOlaG6K8tKwtW3usCgKYJ734lQDX0ZAKBNu46Cs7/DUKoWdVNLJBTOhfWv05yJKM0tQm7kDdg1chA6HGpARN3UTp06FUlJSZg+fTo+/fTTSuvmzp2LdevWISoqCiEhIVCpVAJFWf+K4tORFh4JAEg5/DcyI65i2M6F6PHJ8zjy6jIAgKFMh/CpyzFs24dIPngWeVcS4De0C3wHd8aOAdOFDN+sHBwc0K1bNzg4sBCaE7dB81O6OaHTnInIvhiLvY/MM93zMOXw3xix+2N0nDkBx9/6TuAoGx4p1oD0bYuhy89AwNydcO06stK65lO+RFl2skCRkVCkmAdC4FxY/zZ3ewXqxEwAwMOHv4Cto73AEVknMdYA0Z5Te+XKFWzcuBFNmjTBggULqqyXyWTo1KkTgMqHHicnJ2Pq1Kno2bMnHBwcIJPJEB8fb6mwzSLrXAxubAlHq0f6wPP+1qbluZFxuPTNDvT56nU4eLujx2cv4tScVaLeu+Pn54fly5fDz89P6FAkhdtg/Wv+YFfYOqlwddVuU0MLADkXYpFx8ir8R/WE3FbU31vWiRRrgDb1GgDAuW3/atfbNfa1ZDjUAEgxDxoCzoX37mZDS/dGjDVAtE3tunXrYDAYMGnSpNt+C6FUKgFUbmqvX7+OjRs3wtXVFX379rVIrJZw4cvNMOj06DjrsUrLL361FYZyHUYdWIz045cRt+OYQBFahl6vh1qthl6vFzoUyeE2WL88OwYCADLPRFdZl3kmGnbODnAJ9LF0WA2eFGuA0qsVACB7/0oYjUaBoxGO6aI91fyTGinmQUPBuZAaAjHWANE2tX/88QcAoH//6r+ZBir2ygKVm9oHHngAGRkZ2L17NyZMmGDeIC2oKD4dcTuOoVnf9mjSrY1puVFvQEZEFOw9XHB942EBI7SMmJgYDBgwADExMUKHIjncBuuXg5c7AKAkLbfKuuLUnIox3u4WjckaSLEGNB39FuQOjZD8v+mInNICcZ9PRMbOpSi+flbo0Cyqw/TxePzyj9X+s3UUzylINSHFPGgoOBdSQyDGGiDapjYhIQEA4O/vX+16rVaLU6dOAajc1Mrlov2V4OKyLTDo9eg4899mvUmXYLR+fACu/rAbXT98Bjb2dgJGSGLHbbD+2KgqjjTRa8urrLu5TPHPGJI2pVcrhC69AM9hrwBGA3KPrEfyD9MQNaMzrkxtJ5nm9tq6g9j36Pxq/+k0WqHDIwnhXEhU/0R7wlVxcTEAoLS0tNr1a9euRUFBATw8PODjY95D9Dp37oz09Nrdj8zWKMc8dK3VY9JPXMZq73G3XV8Qk4I1vv8WUIVKid7LXsO5ResR9dM+DNv2ITrNnYiI936s1esCQFDrIJTLDHcfWM/Gjbv9+61OZmbFuRh79uzB2bM1+yA3ZsyYWsclBtwGhXW337/+nw/hNkpb6G+5kiYAKP75MGTpD+rWUAfqUgMAYeqAzE6Fpkvr51t0ZVN/+L30Dfxe+gbluWlQR59AzqGfUBCxE9cXjkDY8stQON/7nv2goNYwlmnqIeK61aA7ufWiPf9lNNTPdmsNOQBwLqypum6DnAvrpr5zvqGw1N9EDHOhl5cXzpw5U6fHirap9fLyQlFREc6cOYPOnTtXWpeUlITZs2cDANq1a2f2WNLT05GSklKrx9jJbICmZgroH13mPwV1UhaiVu8FABx942uMOrgEibsjkH7icq2eKzUtFWVGyx+Xf/PLi5rSaDSm/9b0sbX924kFt0Fh3e33X5Jecdixg7c7iuIqf2l287Dj6g5NNidrqAN1qQGAMHVArnQwSwraunvDrccjcOvxCG58/gTyjmxAwdnd8Og36Z6fOzU1FQZtST1EaZkaVN+sIQcAzoU1ZaltkHNhBWvM+Zqw1N9ErHNhTYm2qR00aBBiYmKwaNEiDBgwAEFBQQCAkydP4sknn0R+fj6Ayocem4uXl1etH2NrlANm/FLHp38HtBzVCzsGzjAtK0rIwNmP1qHX0lewY8AM6Iqr38tdnWbezQT5ZtDR0bFW428mrkqlqvFjzb0nv6HiNiisu/3+s/++juDJQ9Ckc3CVprZJ52CUqzUouG7Zycca6kBdagAgTB2Q2Zn/PE+n4B7IO7IB5Tn1s600a9asXvfUmrMGmYM15ADAubCmLLENci78lzXmfE1Y6m8ihrmwLj3TTaJtat955x388ssvSEpKQlhYGEJCQqDRaBAbG4uRI0eiefPmOHz4sEWa2rrsRi8vKcW6gHv/1vx2Ug7/jfUhT1VZHrV6r+mbwtq4FnMNtg6Wv1fY6dOnazVep9Ph6aefhrOzMxSKmm3+S5curUNk1o/boLDu9vtP3HcaXUtK0WbKcNzYGm66rY9H+wA07d4GMb8chqFcZ6lwAVhHHahLDQCEqQMaHdBn970/T9HFw3AM7g65snKTbDQYkH/6NwCAffPQe38hANeuxUBVT58szF2DzMEacgDgXFhTltgGORf+yxpzviYs9TcR61xYU6Jtav38/BAeHo5Zs2bhyJEjSEhIQFhYGGbPno1nn33W9E2DJZpaajgUCgXc3NyEDoPonmlzi3Du4/XotuBZDN06H7Gbj8De3RmhL4yAJqsAfy/eKHSIDZIUa0DGjs+hvnoMLl1GwCGgE2wcXFCen47841tQEnsWzm37w6XzQ0KHSRYkxTwgcWg17gE4+XoCAOw9GkFuq0C7N8cCAMoKihH1Y+2/CJAiMdYA0Ta1ABAWFobff/+9yvLc3FykpaVBoVAgNLR+vp0m65CcnIwvv/wS06ZNg6+vr9DhEN2Tq6t2Q5tbhLCXRqLr/Keh05Qh7chFnF20znTOLVUmxRrg9ei7yDv2K9SXj6Dw/H7o1LmwsXeEvW8b+D77OTyHvwqZiK/8T1VJMQ9IHIIeHwivnmGVlt3/9uMAAHVSJpvaGhJjDRB1U3s7kZEVVz8MCQmBUln1lhebN28G8O9hw3v27IGnpyf8/f2rXHSKrItarUZ4eDief/55oUMhqhc3tobjxtZwocOwGlKsAU7B3eEU3F3oMARztyvRAsC6wCctFE3DIMU8IHHYO3ae0CGIghhrgKSb2tsdejx+/PhKP7/yyisAgKeeegqrV682a2xERERERERUc2xqq2E0Gi0ZTr3w6d8BHd9+HHJbBfQaLY7PWoG8KwlVxrmG+KH7R8/B3tMFMpkMZz9ej8TdpwAAYS+NQuCEfjDq9NBry3Fy7g/IuRBr6bdCVsK5pRf6LHsdSndnlBUW4+ib36DgWnKlMV49wjBo3RwUxqaalv0+ci70pWVQONij/w9vwaNdKwDAL2HPmsa4BPmi7zdvmH62c3GErZMKG0KfMfO7IiIiqhsbpS36fjcNLq19oddoocnKx4lZ30OdnFVpnFev+9BpzkTYOtnDaDAicd9pnP9kQ5XnG7JpHtzDWlSaH4moepJuai1xj1pLsHNxRJ+vp2LP6PdQEJOCpj1C8cA3b2BH/+mVxtmo7DDgx1k4+sbXyIyIgsxGDjtXJwCAW2gLtHluGLb3nQZdSSlaje2D7h9Pwe/DZwvxlsgK9PzsRVxbewDXN/2JFg91R6/PX8bukXOrjMu/loxdQ9+ustyg0+HSNzugzSvCkE3vV1pXcC0ZOwfPNP3c7aPnACv8somIiKQl+ucDSDl0HgAQ8sxQ9FjyEg48tqDSmLKCYvz18pdQJ2bCRmmLIZveR8sxvRG37ahpTMizw6BOyoR7WAuLxk9krSR5ZYjjx4/DaDTiwQcfFDqUeuHs74XS7AIUxFTcZzDjxBU4+XrCvW3LSuNajemDrLPXkBkRBQAw6g3Q5hSa1ssVNlA4VJxjbNfIESVp4rvQjKenJ9544w14enoKHYpVs/doBI+2rRC75QgAIOH3k3D2a2K6ImFNGMp0SDsaibKCO9/0W26nQKsxvRGz4dA9xUwEsAYQAcwDc9Fry00NLQBknYuBs1+TKuNyL8VBnZhpekzupXg4t2hqWu/c0gstH+6FyK+3mT9okiQx1gBJ7qkVm8IbabD3aATP+1sj61wM/IZ1ha2TCk7NmyA3Ms40zjXIF/rSMgxcMxuO3u7IvZqI0/N/gjanEHlXEnB5xS6Mi/gW2jw19GXl2Dvm/Tu8qnXy8PDAxIkThQ7D6jn6NEZJRq7p3qgAoE7JhqNv4yqHWbkENMPI/Z/BqDcg5pfDiP5pX61eq8XwbihKykTu5fj6CJ0kjjWAiHlgKW2eG46kA2fvOEbl6Qr/Ed1xYNKiigUyGXoteRmn5qyCsVxvgShJisRYAyS5p1ZsyotKcHjKEnSaOwkj9n0Kr55hyItOgqFcV2mcXGEDn34dcGLWCuwcPBPFqdnovmgKAMDRtzGaP9gZW7q/il87vYgr3+/CA7ec0ygWhYWFOHjwIAoLC+8+mO7ov0cDy2SyKmNyIm9g0/0v4rchs3Do2c8QPHkI/Ef2qNXrtH5sAGLWcy8t1Q/WACLmgSW0nfoIGrX0wrlP1t92jK2TCgPXvIPIb3eYdkLc9/IopJ+8wi9yyazEWAPY1IpExsmr2Dt2HnY9+DbOLPgZDk3dTIcj36ROzkba8cum+1fe2BoOz46BAAD/ET2RH5UITWY+AOD6L4fRtHsb0d27MDU1FXPmzEFqaurdB9NtFadkw9HbHTKbf7cPx2YeKE7OrjSuXK1BeVEJAKAkLRdx24+iabc2NX4dR9/G8OwSjLhtvGUN1Q/WACLmgbmFvTQKLYZ3w8GJH0GvKat2jMLRHoPXv4vE/WdwZcUu0/Km3UMROKEfxkV8i2E7FsLOxRHjIr6FnYujpcInCRBjDRBXxyJhqiaupv9vP20c0o5dQlF8eqUx8b8dR+MOAbB1UgGouGJy7uWKKyQXJWagSdc2UDjYAwCaD+mM/JgUGA0GEP1XaU4hci/FI2DsAwCAFg91hzopq8qhx6omrsA/e3AVjvbwHdQJOZfia/w6rScMQOKeCJQVltRX6ERERGYT+uIItBzTC/snfHjbuUvhUNHQpvz5Ny5+ubnSuj8mf4zNnV/G5q6vYM/D76KsoBibu75y1+tPEEkdz6kViY6zHkPTbm0gs5Ej6+w1HJ/+LQCg55KXkLT/DJL2n0FxSjYil2/D8F2LYNTpUZKei+MzvwMAJO4+hcYdAjBy36fQa8tRXqxB+GtfCfmWqIE7PmsFei99FW2nPoJytQZHpy4HUHmba/FQdwQ/9SCMOj1kChsk/HYC13/591Dikfs/g6qJG+xcHDH+7AqkH7+E8NeXm9YHTuiHo29+Y/H3RkREVFsO3u7o+sHTKIxPx9DNHwAA9GU6/P7Q7EpzY+jzw+HZMRC2Dkq0GNYVABC/6wQuLtsqYPRE1o1NrUgcf+u7Gi2P/fUvxP76V7Vjzy1aj3OLbn/uB9GtCmNTq72Fz63bXNSPexH1497bPsdvQ2bd8TU2d32l7gESERFZUElaLlZ7j6t23a1z48VlW2vUwKqTs3iPWqIa4uHHJClKpRLBwcFQKpVCh0JEAmANIGIeEEmdGGsA99SSpLRs2RI///yz0GEQkUBYA4iYB0RSJ8YawD21REREREREZLXY1JKkREdHo1evXoiOjhY6FCISAGsAEfOASOrEWAPY1JKkGI1GlJeXw2g0Ch0KEQmANYCIeUAkdWKsAWxqiYiIiIiIyGrxQlENlEKlxMTYtUKHUWMKlXiunkYVuA0SCcveBggfLnQUtWNvU3/PZa4atLXn69Bk5EHV1A2PHF9+9wfUAuuQuFjbPAhY9zZojb/vmrDmv4k1YVPbQMlkMtg62AsdBkkYt0EiYclkgErCs7S5apBMLjP9lzWO7oTzoGXx9033QsLTJUmRv78/NmzYAB8fH6FDIRKMjdIWrScOgv+I7nAN9oOdswragmLkXopD/K6TiP31TxjKdAAAv6Fd4B7WEn9/vkngqOsHawAR84BI6sRYA9jUkqTY29sjICBA6DCIBOPk64mBa+fALbg5UsMjcemb7SjNKYTSzRlePcPQ49Pn0bhDAE7MXAEA8BvaFYET+oumqWUNIGIeEEmdGGsAm1qSlLS0NPzwww947rnn4O3tLXQ4RBZlo7TFwJ9nwyWgGQ4//zkSdp2otP7ydzvhGtwczfq1FyhC82MNIGIeEEmdGGsAr35MklJQUICdO3eioKBA6FCILK714wPhFuKHyyt+q9LQ3pQfnYQrK3YBAMZFfIvACf0BAE+nbTb98+oRZrGY6xtrABHzgEjqxFgDuKeWiEgi/Ef2AABc+/lAjcZHvP8jQl8cCa/uoTjy2jLT8vyYZLPER0RERFQXbGqJiCTCNcQPZUUlKErIqNH4xL2n4Te0K9A9FDe2hJs5OiIiIqK64eHHREQSYeesQnmRRugwiIiIiOoVm1qSFHd3dzz11FNwd3cXOhQiiysr0sDWSdr3AGQNIGIeEEmdGGsAm1qSFLlcDltbW8jl3PRJevKjEmHXyBHOLZoKHYpgWAOImAdEUifGGiCed0JUA9nZ2Vi1ahWys7OFDoXI4uL/ueJx0MRBNX6M0Wg0VziCYA0gYh4QSZ0YawCbWiIiiYjZcAj515IQ9tJI+A3vVu0Y1xA/hL040vSzrrgUAGDn4miRGImIiIhqi1c/JiKSCH1pGQ4++TEGrZ2DAT/MROpfF5Dy1wVoc4ugdHOGV49Q+AzsiJh1f5gek3U+Bm0AdF80BcmHz8NYrkfa0UiU5hQK90aIiIiIbsGmlohIQtSJmfjtwVkImjgI/iN6oN3rY2DrpIK2oBi5kTdw/K0VuLHliGl83LZj8GgXgJYP94L/wz0ht7HB3kfmIf3EZQHfBREREdG/2NSSpDg7O2Po0KFwdnYWOhQiweg1Zbi6ajeurtp917FGgwGn563G6XmrzR+YBbAGEDEPiKROjDWATS1Jio+PDz788EOhwyAigbAGEDEPiKROjDWAF4oiSdFqtUhKSoJWqxU6FCISAGsAEfOASOrEWAPY1JKkxMXFYezYsYiLixM6FCISAGsAEfOASOrEWAN4+HEDZTQaodNYz7cnCpUSMplM6DCoHhmNQKle6Chqzt4G4CZIYmJtOQgwD0lcmINE1oNNbQOl02ixLmCS0GHU2MTYtbB1sBc6DKpHpXqgz92vI9RghA8HVKxoJCLWloMA85DEhTlIZD14+DERERERERFZLTa1REREREREZLV4gAJJSkhICCIiIoQOg4gEwhpAxDwgkjox1gDuqSUiIiIiIiKrxaaWJCUhIQHPPvssEhIShA6FiATAGkDEPCCSOjHWADa1JCkajQaXLl2CRqMROhQiEgBrABHzgEjqxFgD2NQSERERERGR1eKFokTEq0cYhm6dX2lZebEGhTfSELv5CK7+sBtGvUGg6IiIiIikqSjyT1x7t3+lZXJ7R9j7BMO9/2Q0Gf4aZDY2wgRHJAJsakUobvsxJB08A8hkUHm6InB8X3Sd/zRcWvvgxMwVQodHREREJEluvSfApcsIwGhEeW4qcg6tRvKqN1GaeBktXv1e6PCIrBabWhHKvRyHG1vCTT9Hr96H0eFLEfTEQJz7ZAO0OYUCRicsb29vzJ8/H97e3kKHQkQCYA0gYh4IyaFlB3j0m2T62XPYy7j8ahtkH1iFZhMXwNa1qYDRkVSIsQbwnFoJ0Gm0yD53HTK5HI1aSLtYuri4YNiwYXBxcRE6FCISAGsAEfOgIbFxaATH4B6A0Qht+g2hwyGJEGMNYFMrEc7+Fc1saV6RwJEIKy8vD7/++ivy8vKEDoWIBMAaQMQ8aEiMRiO06dcBAIpGjQWOhqRCjDWATa0I2aiUULo7Q+nRCK4hfui2aAo82rZC1vkYFMWlCx2eoDIyMrB48WJkZGQIHQoRCYA1gIh5ICSDtgS6wmyUF2ShJP4iEr95AZq4C3Bo3QX2zVoLHR5JhBhrgCTOqc3OzsYnn3yCbdu2ISUlBZ6ennj00UexcOFCPP/881i3bh1WrlyJKVOmCB1qvegwfTw6TB9faVnCnlM4+fZKgSIiqSvPz0TG9iUoOLsbZZkJgFwOW5cmcAjoBLfej8KtxyNCh2iV2r4+Bh5tW8GjXSs4t2gKdVImNnd9ReiwqAE5+7CsxmPv+z4Oyqb+5gumASi8kQZ9aRkAQF+mg0Gvh5xXnCULStu0AGmbFvy7QCZDo07D0OLVVcIFRSQCom9qL168iCFDhiAjIwOOjo4IDQ1FSkoKvvjiC8THxyMxMREA0L59e4EjrT/X1h1E3I5jkCts4Brsh7avj4aqsSt0/0zkACC3U2Dkvs9wY9tRRH611bS899JXYe/pioMTPxIidBKhsqwkXH2rCwyaIrj3nQjV0JcBANq06yg4+zsMpWo2tXXUac5ElOYWITfyBuwaOQgdDjVA/tN+rvRzadJVpG9eBNfuY+D6n7xTuHhaMjSLMRqNSPj9JKJW70P6sUum5dqcQmzt8TqCJw9G0KTBULo6CRglSYXHoOfg3ucxQCaDXOkA+2ZBUDTyEDosIqsn6qY2OzsbI0aMQEZGBmbNmoX33nsPTk4Vk9bChQvx3nvvQaFQQC6X47777hM42vpTFJ+OtPBIAEDK4b+RGXEVw3YuRI9PnseRV5cBAAxlOoRPXY5h2z5E8sGzyLuSAL+hXeA7uDN2DJguZPgkMunbFkOXn4GAuTvh2nVkpXXNp3yJsuxkgSKzfpu7vQJ1YiYA4OHDX8DW0V7giKihufUqq0DFvTLTNy+Cyr9dlXViZCjX4ei0b3Fjy5Fq16uTMnH2o3WI2XAIg9e/C2eJX0yRzM/eOxCNOgwSOgwi0RH1ObVTp05FUlISpk+fjk8//dTU0ALA3LlzERISAp1Oh6CgIKhUKgEjNa+sczG4sSUcrR7pA8/7/z1fIzcyDpe+2YE+X70OB2939PjsRZyaswqaDPGcNP5fDg4O6NatGxwcuFfLUrSp1wAAzm37V7verrGvJcMRlZsNLdUca4B0GI1GnHj7+9s2tLcqvJGG/Y8tQGl2gQUiEx7zgEjaxFgDRNvUXrlyBRs3bkSTJk2wYMGCKutlMhk6deoEoPKhx5s3b8bYsWPRokULODg4ICQkBHPnzoVarbZY7OZw4cvNMOj06DjrsUrLL361FYZyHUYdWIz045cRt+OYQBFahp+fH5YvXw4/Pz+hQ5EMpVcrAED2/pUwGo0CR0NSxxogHVlnohGz4VCNxxfFpyPy6+3mC6gBYR4QSZsYa4Bom9p169bBYDBg0qRJt/0WQqlUAqjc1C5ZsgQ2NjZYtGgR9uzZg5deegnffvstHnzwQRgMBovEbg5F8emI23EMzfq2R5NubUzLjXoDMiKiYO/hgusbDwsYoWXo9Xqo1Wro9XqhQ5GMpqPfgtyhEZL/Nx2RU1og7vOJyNi5FMXXzwodGkkQa4B0RK3eV+vHxGw8DF2J1gzRNCzMAyJpE2MNEG1T+8cffwAA+vev/pBHAEhOrjiX79am9rfffsOmTZswceJE9O3bF2+++SaWL1+O48eP4+jRo+YN2swuLtsCg16PjjMnmJY16RKM1o8PwNUfdqPrh8/Axt5OwAjNLyYmBgMGDEBMTIzQoUiG0qsVQpdegOewVwCjAblH1iP5h2mImtEZV6a2Y3NLFsUaIA3lJaWI33Wi1o8ry1cj6cAZM0TUsDAPiKRNjDVAtBeKSkhIAAD4+/tXu16r1eLUqVMAKje1np5Vr/548zDllJSUOsXSuXNnpKfX7v6wtkY55qFrrR6TfuIyVnuPu+36gpgUrPH9t6FVqJTovew1nFu0HlE/7cOwbR+i09yJiHjvx1q9LgAEtQ5Cuczye7LHjbv9+61OZmbFOYh79uzB2bM1a6bGjBlT67jEQGanQtOl9VPslE394ffSN/B76RuU56ZBHX0COYd+QkHETlxfOAJhyy9D4ex+T68RFNQaxjJNvcTbENSlBgjNGupAXWoAIEwdqM8ctJSGkoduRiVmGDrW6bHvvPwmjr2aVs8RmRfnQvOozxx0btsPnXaY/xSchpKDZHlimAu9vLxw5kzdvlgUbVNbXFwMACgtLa12/dq1a1FQUAAPDw/4+Pjc8bmOHKm4yERISEidYklPT691Q2wnswHMfBHGLvOfgjopC1Gr9wIAjr7xNUYdXILE3RFIP3G5Vs+VmpaKMqPlD2G4+XeuKY1GY/pvTR9b1y8zrJ1c6WCWTdDW3RtuPR6BW49HcOPzJ5B3ZAMKzu6+5yuxpqamwqAtqacohWeJGlDfrKEO1KUGAMLUAXPloDk1lDwstXEA6niHooKCAqSUWFfd51xoHsxBsiZinQtrSrRNrZeXF4qKinDmzBl07ty50rqkpCTMnj0bANCuXbs7Pk9KSgref/99DBkyBB071u1bXy8vr1o/xtYoB8y4w8Onfwe0HNULOwbOMC0rSsjA2Y/WodfSV7BjwAzoiqv/QqA6zbybCbKHxtHRsVbjbyauSqWq8WPv9qWHWMnszH9FcKfgHsg7sgHlOfdeJJs1ayaqb6fNXQPMwRrqQF1qACBMHbBEDta3hpKHtkY5dAYDFHU5y8rFHj5u1lX3OReaB3OQrIkY5sK69Ew3ibapHTRoEGJiYrBo0SIMGDAAQUFBAICTJ0/iySefRH5+PoDKhx7/l1qtxsMPPwxbW1v8+GPtD8m9qS670ctLSrEuwHz3EEw5/DfWhzxVZXnU6r2mPbe1cS3mGmwdLH+PzNOnT9dqfFRUFDZs2IBhw4bVeM/70qVL6xCZ9dPogD677/15ii4ehmNwd8iVlT8cGA0G5J/+DQBg3zz0nl/n2rUYqERU0cxdA8zBGupAXWoAIEwdqK8ctKSGlIfhr3+F2M13v53PrZRuTth19jQUKqWZojIPzoXmwRwkayLWubCmRLvZv/POO/jll1+QlJSEsLAwhISEQKPRIDY2FiNHjkTz5s1x+PDh2za1paWlePjhhxEXF4fw8HA0a9bMwu+AzCEwMBD79u2Ds7Oz0KFIRsaOz6G+egwuXUbAIaATbBxcUJ6fjvzjW1ASexbObfvDpfNDQodplVqNewBOvhXHWNp7NILcVoF2b44FAJQVFCPqx9p/QSV2rAHSEfz00Fo3tYGPDbC6hrYumAdE0ibGGiDaptbPzw/h4eGYNWsWjhw5goSEBISFhWH27Nl49tlnTbvPq2tqy8vL8eijjyIiIgJ//PEHQkPvfS8SNQwKhQJubm5ChyEpXo++i7xjv0J9+QgKz++HTp0LG3tH2Pu2ge+zn8Nz+KuQyUV7IXazCnp8ILx6hlVadv/bjwMA1EmZbGqrwRogHZ73t0bQpMG4tvZAjcY3auWNtq9J42JIzAMiaRNjDRBtUwsAYWFh+P3336ssz83NRVpaGhQKRZWG1WAw4Mknn8SBAwewe/dudO1qXVcfpTtLTk7Gl19+iWnTpsHX11focCTBKbg7nIK7Cx2GKO0dO0/oEKyO1GuApa7A2hDIZDJ0/3gKDDodrv9y5/uwuwT6YPD6d2HvLp69Fnci9Twgkjox1gBJ7h6JjIwEUHE1Y6Wy8mFGr732GjZu3IgZM2ZApVLh5MmTpn8372tL1kutViM8PBxqtVroUIhIAKwB0iJX2KDXF69gwI+z0Kxv1SOznFt6ocu8p/DQ74vg1LyOl0u2QswDImkTYw0Q9Z7a27nZ1FZ36PHu3RVXBPjoo4/w0UcfVVo3b948fPDBB2aPj4iIiOqHTCaD39Cu8BvaFUWJGSiKS4dBp4e9RyN4tGvF0x+IiESATe1/xMfHWziae9d1wbPwe7AznJo3wfZ+05AfnVTtuNaPD6g4X0guQ9rRSJx8ZyWM+orbbzj6NEb3RVPQqJU3AODqj3sR9b89FnsPRERE5ubs1xTOftZ251GyJtfmDYEuLx2Qy2Gjckbz55fDoVWHascaykpxddr9kCsd0OaLijtlaDPicOPTcTAa9DAa9LD3CUGLV7+Hwklc5z8S1TdJN7V3u0ettUj4/QQufbsdw3csvO0Yp+ZN0HHWY9g5eCZKswswYPXbaP3EQFz7ueICGv3/NxORy7cjYdcJAIDK09USoRMRERGJRquZm6BwcgUA5J/cjvjlzyL0y3PVjk1ZOxeOIT2gibtgWmbr3gzBHx813QYvadWbSNu4AM2f+8LssRNZM0kec3P8+HEYjUY8+OCDQodSLzJOXkVJWu4dx7QY0R0JeyJQml0AAIhesx+tRvcGAHj3aQt9aZmpoQUATVa+2eIVkqenJ9544w14ekrn3Cki+hdrABHzwJxuNrQAoC8pgExW/Uftosvh0KbGwKPfk5WWy22VpobWqNdDr1HzEHmqd2KsAZLcUytFTj6NUZycZfpZnZwFR5/GAADXoOYozSlE3/+bhkYBzaBOzsTpD36COjFTqHDNxsPDAxMnThQ6DCISCGsAEfPA3OK+nIyiyIorbrf+oOqt1fSlxUj+4U0EzN0JbWpMlfWG8jJEzeyKsswEqFq2R/O5O80eM0mLGGsAv/qREKPx39s4yG5ZLlfYoFmfdrjw5a/4bchMpBw6j77fTbN8gBZQWFiIgwcPorCwUOhQiEgArAFEzANzazltDdr9Lwk+kxYiefXMKutTVs+E5/BXYefhU+3j5bZ2CF36N9r9lAF7n2Bk7f3O3CGTxIixBrCplQh1Sjacmjcx/ezo64nilOyKdclZyLkUh/xrFbcsit1yRLRXhExNTcWcOXOQmpoqdChEJADWACLmgaV4DHgKRZGHoSvMqbRcfeUo0jZ+iMjn/XFjyWPQJETi8mthVR4vt7WDx8BnkPPnz5YKmSRCjDVAfF0LVSvh95NoMawr7Bu7AACCJw9B3I5jAICUQ+fh4OUOBy93AIBP/47Ij0qC0WAQLF4iIiIia6IvKURZzr9NQt6JbVA4e8DG2b3SuNCvLqLtyni0XRmPVm/9AlWLtgj7+jIAoCwrEfrSYgCA0WBA3tFNcGghjgubEpkTz6kVgW6LpsDvwS5QNXHFg5veR3lxKbb2fB09l7yEpP1nkLT/DNSJmTi/ZBOG71wImVyOtKORuLb+DwCATqPFydkrMejn2YBMhrLCYhx5Zamwb4qIiIjIiuiLCxD76VgYyjSQyeRQNPJE4Lu7IJPJEL98Cly7joJrt1F3fA5NwiWkrHkHAGA0GuDQ6n40f/4rS4RPZNXY1IrAqTmrcGrOqirLj79V+RyMmHUHEbPuYLXPkfrnBez880K164iIiIjozuw8m6PNkohq1/m/XvVzGgA4t+1nukctALh0Hg6XzsPNEh+RmPHwY5IUpVKJ4OBgKJVKoUMhIgGwBhAxD4ikTow1gHtqSVJatmyJn3/mBReIpIo1gIh5QCR1YqwB3FNLREREREREVotNLUlKdHQ0evXqhejoaKFDISIBsAYQMQ+IpE6MNYBNLUmK0WhEeXk5jEaj0KEQkQBYA4iYB0RSJ8YawHNqGyiFSomJsWuFDqPGFCrxnGhOFextgHArugCjvY3QEdQva6sBAOtAfbO2HATEl4ckbcxBIuvBpraBkslksHWwFzoMkjCZDFCxQgiGNYCYg0TCYg4SWQ8efkxERERERERWi98/kaT4+/tjw4YN8PHxEToUIhIAawAR84BI6sRYA9jUkqTY29sjICBA6DCISCCsAUTMAyKpE2MN4OHHJClpaWlYuHAh0tLShA6FiATAGkDEPCCSOjHWADa1JCkFBQXYuXMnCgoKhA6FiATAGkDEPCCSOjHWADa1REREREREZLXY1BIREREREZHVYlNLREREREREVotNLUmKXC5Hx44dIZdz0yeSItYAIuYBkdSJsQaI550Q1YDBYMD58+dhMBiEDoWIBMAaQMQ8IJI6MdYANrVERERERERktdjUEhERERERkdViU0tERERERERWi00tSYqzszOGDh0KZ2dnoUMhIgGwBhAxD4ikTow1QGY0Go1CB0FUV6dPnzb7a3Tp0sXsr0FEdcc6QFLHHCAiqdcB7qklSdFqtUhKSoJWqxU6FCISAGsAEfOASOrEWAPY1JKkxMXFYezYsYiLixM6FCISAGsAEfOASOrEWAMUQgdA1TMajdBprOfbE4VKCZlMJnQYomdt24VYcXsnSzAagVK90FHUjr0NwNQgsWAOElkPNrUNlE6jxbqASUKHUWMTY9fC1sFe6DBEz9q2C7Hi9k6WUKoH+uwWOoraCR8OqPjJgkSCOUhkPXj4MREREREREVktNrVERERERERktXiAAklKSEgIIiIihA6DiATCGkDEPCCSOjHWAO6pJSIiIiIiIqvFppYkJSEhAc8++ywSEhKEDoWIBMAaQMQ8IJI6MdYANrUkKRqNBpcuXYJGoxE6FCISAGsAEfOASOrEWAPY1BIREREREZHV4oWiRMSrRxiGbp1faVl5sQaFN9IQu/kIrv6wG0a9QaDoSIoCH+2H3stew9E3vsb1TX9WWe/k64lxp/8P1zcextE3v7F8gEQiUxT5J66927/SMrm9I5TNguDR70k0GfE6ZDac+oks7Xa5ae8TDPf+k9Fk+GuQ2dgIExyRCHBmE6G47ceQdPAMIJNB5emKwPF90XX+03Bp7YMTM1cIHR4REZmZW+8JcOkyAjAaUZ6Xjpw/1yD5f9NRmnwVLV79XujwiCSrUm7mpiLn0Gokr3oTpYmXmZtE94BNrQjlXo7DjS3hpp+jV+/D6PClCHpiIM59sgHanEIBoxOWt7c35s+fD29vb6FDISIBSKUGOLTsAI9+k0w/ew5/BZdfbYPsA6vQbNJHsHXxFDA6EppU8qAhqpKbw17+NzcnLoCta1MBoyOpEGMN4Dm1EqDTaJF97jpkcjkatZB2sXRxccGwYcPg4uIidChEJACp1gAbe0c4BnUDjEZo02OFDocEJtU8aIhsHBrBMbjHP7l5Q+hwSCLEWAPY1EqEs39FM1uaVyRwJMLKy8vDr7/+iry8PKFDISIBSLkG3GxmFc4eAkdCQpNyHjQ0RqMR2vTrAABFo8YCR0NSIcYawKZWhGxUSijdnaH0aATXED90WzQFHm1bIet8DIri0oUOT1AZGRlYvHgxMjIyhA6FiAQglRpg0JZAV5iN8oIsaOIjkfjdq9DcOA+H1l1g36y10OGRwKSSBw3RrblZEn8Rid+8AE3cBeYmWZQYa4AkzqnNzs7GJ598gm3btiElJQWenp549NFHsXDhQjz//PNYt24dVq5ciSlTpggdar3oMH08OkwfX2lZwp5TOPn2SoEiIiIiS0rbtABpmxZUWubabTT8Xv4/gSIiIqCa3JTJ0KjTMLR4dZVwQRGJgOib2osXL2LIkCHIyMiAo6MjQkNDkZKSgi+++ALx8fFITEwEALRv317gSOvPtXUHEbfjGOQKG7gG+6Ht66OhauwKXWmZaYzcToGR+z7DjW1HEfnVVtPy3ktfhb2nKw5O/EiI0EmijEaj0CEQiYrHoOfg3ucxGA06aBIuIX3LJyjPz4DcTmUaU3Q5HNc/HFblsUZdGYwGPTpt01syZCJJuJmbkMkgVzrAvlkQFI14SgDRvRJ1U5udnY0RI0YgIyMDs2bNwnvvvQcnJycAwMKFC/Hee+9BoVBALpfjvvvuEzja+lMUn4608EgAQMrhv5EZcRXDdi5Ej0+ex5FXlwEADGU6hE9djmHbPkTywbPIu5IAv6Fd4Du4M3YMmC5k+CQiN79IsbG3q3a9jYMSAKC/5QsXIrp39t6BaNRhEADA5f6hcArtjeh3eiHhu5fRasZ6AIBzWB903Kiu9LiynFREzegMz4des3jMRFJwa24SUf0R9Tm1U6dORVJSEqZPn45PP/3U1NACwNy5cxESEgKdToegoCCoVKo7PJN1yzoXgxtbwtHqkT7wvP/f8zVyI+Nw6Zsd6PPV63DwdkePz17EqTmroMkQz0nj/+Xg4IBu3brBwcFB6FAkQZ2YCQBwDfKtdr1r64rlRf+MIzI3qdYAp+DucO87CXlHNkAdfbLaMYZyLW588gicQnvDe/wcC0dIliTVPCCiCmKsAaJtaq9cuYKNGzeiSZMmWLBgQZX1MpkMnTp1AlD50OPw8HAMHjwY3t7eUCqV8PX1xYQJE3D16lWLxW4OF77cDINOj46zHqu0/OJXW2Eo12HUgcVIP34ZcTuOCRShZfj5+WH58uXw8/MTOhRJyIm8AXVKFlqO7gVVU7dK6+S2CrR5dhiMBgOS9p8RKEKSGinXAO8J7wFyG6Sue6/a9YnfvgRDeSn831ht2cDI4qScB0Qkzhog2qZ23bp1MBgMmDRp0m2/hVAqKw59vLWpzcvLw3333YevvvoK+/fvx6efforLly+jR48eSE5Otkjs5lAUn464HcfQrG97NOnWxrTcqDcgIyIK9h4uuL7xsIARWoZer4darYZez3PFLMGoN+Dk2yth6+yAhw99gU5zJyFo0iC0mzYOI/d/Bq+eYYj8ejsKY1OFDpUkQso1wN47EO59HkPRhYMouhxeaV3mb1+h4MwuBMzeDrlSPN/cU/WknAdEJM4aINqm9o8//gAA9O/f/7Zjbjaptza1o0aNwpdffonx48ejb9++mDhxIrZs2YKCggJs2bLFvEGb2cVlW2DQ69Fx5gTTsiZdgtH68QG4+sNudP3wmdue+ygWMTExGDBgAGJiYoQORTKS/ziH3aPeRfqxSwh8tC+6L5qC+14cidKcQvz5wuc49/F6oUMkCZF6DfAaPxeQy5G6/n3TsqKLh5G85m20mrkJyqb+wgVHFiP1PCCSOjHWANFeKCohIQEA4O/vX+16rVaLU6dOAbj7lY8bN664GbZCUbdfV+fOnZGeXrv7w9oa5ZiHrrV6TPqJy1jtPe626wtiUrDG99+GVqFSovey13Bu0XpE/bQPw7Z9iE5zJyLivR9r9boAENQ6COUyQ60fd6/Gjbv9+61OZmbFuZt79uzB2bNna/SYMWPG1Douc6nLdtEQ5FyIxZ8vfC50GPVGqO2dqlebOlCXGgAIUwdkdio0XVq7DxzObfuh047bX01c1bxNpasaazPicWPxo/B9ejGc293+S+CaCgpqDWOZ5p6fh2pHanOhpdQlB2/nbrlZX5iD0iWGudDLywtnztTtlDTRNrXFxcUAgNLS0mrXr127FgUFBfDw8ICPj0+V9Xq9HgaDAUlJSZgzZw6aNWuGCRMmVPNMd5eeno6UlJRaPcZOZgM0rdPL1ViX+U9BnZSFqNV7AQBH3/gaow4uQeLuCKSfuFyr50pNS0WZ0fKHMNz8O9eURqMx/bemj63t386cLLFd0N0Jtb1T9WpTB+pSAwBh6oBc6WDWdDdoSxD78Wi4dB2FJvV0tePU1FQYtCX18lxUc1KbCy3F3DloDsxB6RLrXFhTom1qvby8UFRUhDNnzqBz586V1iUlJWH27NkAgHbt2lX7+L59++LYsYqLJgUGBuLQoUOmPbZ1iaW2bI1ywIw7gnz6d0DLUb2wY+AM07KihAyc/Wgdei19BTsGzICuuPovBKrTzLuZIHuuHB0dazX+ZuKqVKoaP7a6Lz2EYu7tgmpGqO2dqlebOlCXGgAIUwdkdua9Kn/e8S3QxF1Aaco15B3dWGV92NdXYOdZu4uINGvWjHuJBCC1udBSzJ2D5sAclC4xzIV16ZlukhmNRvMfCyGAV155Bf/3f/+H5s2b4+DBgwgKCgIAnDx5Ek8++SQSEhJQXl6ON998E19++WWVx0dHRyM/Px9xcXFYvHgxsrKycPToUYtdJay8pBTrAiZZ5LXqw8TYtbB1sLf4654+fbpW46OiojB58mSsWbMGISEhNXpMly5d6hKaWVjbdiFWQm3vVL3a1IG61ABAmDqg0QF9dlv8Ze9J+HBAJdqvyxsuqc2FlsIcJGsi1rmwpkR7oah33nkHbm5uSEpKQlhYGNq2bYvAwED06NEDbdq0Qe/evQHc/nza4OBgdOvWDY899hj++OMPFBYW4rPPPrPkWyAzCAwMxL59+xAYGCh0KEQkANYAIuYBkdSJsQaItqn18/NDeHg4hg8fDnt7eyQkJMDT0xOrVq3Cjh07EBUVBeDuF4kCAFdXVwQGBuL69evmDpvMTKFQwM3Nrc4X/SIi68YaQMQ8IJI6MdYA0Ta1ABAWFobff/8dRUVFKCwsxIkTJ/Dcc88hLy8PaWlpUCgUCA0NvevzZGZmIjo6GgEBARaImswpOTkZM2bMsOp7DhNR3bEGEDEPiKROjDVA1E3t7URGRgIAQkJCoFQqK62bNGkSPvjgA2zfvh1//vknVq5ciX79+kGhUGDatGlChEv1SK1WIzw8HGq1WuhQiEgArAFEzAMiqRNjDRDPPudauNnUVnfocffu3bFmzRosW7YMpaWlaN68Ofr37485c+agRYsWlg6ViIiIiIiI7oBN7X+89tpreO21+rlfn6XYKG3R97tpcGntC71GC01WPk7M+h7q5KxK47x6hGHQujkojE01Lft95FzoS8ugcLBH/x/egke7VgCAX8Keteh7oHsz+Jf3oPJ0BQwGlBeX4tTcH5B7Ob7KOCdfT/Re9hrc7/NHYVw6dg1927SurtuAQqXEg5s/gI3SFgCgycyrdvu7ObbboufQuH0A5LYKJO6JwNlF60zrXUP80P2j52Dv6QKZTIazH69H4u5Td43B0acxui+agkatvAEAV3/ci6j/7ak6UCZDtwXPwGfg/YDRiMsrdiH6p30AAL9hXdHhrQmAwQCZrQKJeyNw/pMNNfodEAlNp87HtXf7mX42aEugTb+B9msyoXB2Ny3XZsQj/qunUXLjPOy9W6PNF5Vvcp++bQlyDq2GzEYBua09mr+wHI6tG+7VLokaqsTvp6Lg9E6UZSYg9KtIqFrcV2XM3fKRiGpO0k3t7e5Ra42ifz6AlEPnAQAhzwxFjyUv4cBjC6qMy7+WXKmRucmg0+HSNzugzSvCkE3vmz1eql9/vfA5ygorbrbuN7QLen35Cn4bMqvKuDK1Buc+3QA7Zwd0mDmh0rq6bgO60jLse3S+6b7Goc8/hC7zn8bh5xZXGdv2jUcAADsGzIBMYYNBP89GixE9kLDrBGxUdhjw4ywcfeNrZEZEQWYjh52rU41i6P+/mYhcvh0Ju04AQEWDX42AcQ/AJag5tvWaCttGDhi1/zOkH4tEwfVUpB65iMS9pwGjEXJbBYbtWIDsczFI2s8PGdTwKZxcEbr0b9PP6duWQH35r0oNLQDYODSCz8SF0JcUIHX9vErrSuIuIOv35Qhdfhk2Kifk/LkWiSteRZslEZZ4C0Si4tZrHLwemYXo2b1vO+ZO+UhEtSPJc2qPHz8Oo9GIBx98UOhQ6oVeW25qaAEg61wMnP2a1Oo5DGU6pB2NRFlBcX2H16B4enrijTfegKenp9Ch1KubDS0A2Do7wGio/vbTZflqZEZEQVeirbKuztuA0WhqaAHA1kkFo8FQ7VD3UH/TtmrU6ZH65wUEjHsAANBqTB9knb2GzIiKK5Mb9QZocwrv+vLefdpCX1pmamgBQJOVX+1Y/1E9Eb1mP4wGA8ry1YjbeRwtH674wKErLgX+uW23jdIWNna2t/09kvUSaw34r5w/fkTjQc9VWa5wdodTaG/IlY7VPs6oK4dBW1ED9MX5sPPwNWucJAyp5IGQnMMegF3jO+fP3fKRyFzEWAMkuadW7No8NxxJB85Wu84loBlG7v8MRr0BMb8cNh16KRUeHh6YOHGi0GGYRe+vXod3zzAAwIEnFlr89YdsfB9ubfxQmlOI/dUcJQAA2X9fh/+onkjcexpyOwX8hneFrZMDAMA1yBf60jIMXDMbjt7uyL2aiNPzf7prY+sa1BylOYXo+3/T0CigGdTJmTj9wU9QJ2ZWGevk0xjFtxwWrU7KQuP2/17V3LNzMHp8+gJcWnkj6qd9SD5YfR6R9RJzDbhJHXUCuqIcuHQZUavHObRsjyYPT0fk8y2hcHaHzFaJ4EVHzBQlCUkKeUBEtyfGGiDJPbVi1nbqI2jU0gvnPllfZV1O5A1suv9F/DZkFg49+xmCJw+B/8geAkQpnMLCQhw8eBCFhXffA2htjk5djl87v4Rzn25A5/cnW/z190/4EBvbP4+4ncfR/s2x1Y659M12FKfmYMTujzHwp3eQeToaBp0OACBX2MCnXwecmLUCOwfPRHFqNrovmnLX15UrbNCsTztc+PJX/DZkJlIOnUff725/pXKj8d+9rzKZrNK6rDPR2DlwBjZ1fgmNOwSiafc2NXnrZEXEXANuyjn4P3j0nwyZTe2+t9ZmJqAgYifarohFu/8lo+moaYj7QlwfeqiCFPKAiG5PjDWATa2IhL00Ci2Gd8PBiR9Brymrsr5crUF5UcVhqiVpuYjbfhRNu0nrQ3tqairmzJmD1NTUuw+2UrG//gWvnmFQutXsfNR6ZTQiZu1BBIzrW+1qvbYcp+etxs7BM7Fv3Acoy1ej4FrFPdLUydlIO34ZJem5AIAbW8Ph2THwri+pTs5CzqU45P/zPLFbjsCjXSvI5FXLmzolG07N/z0039G3MYpTsquM0+YUIvngWfiPkNaXPlIg9hqgLy1G7tGNaDyo9hf7yzv2K1Qt7oOte8UF1zwGPgP15SMw6vX1HSYJTOx5QER3JsYawKZWJEJfHIGWY3ph/4QPK51feStVE1fgnz1TCkd7+A7qhJxL8ZYLkszC1kkFVVM3089+w7pCm6eGNq9+7z02ZNM8NO5Qtcm0b+xS6YJOLUf3Qu6VhNvGaqOyAwA4NW+C4KcexOXvfgMAxP92HI07BMDWSQUA8OnfAbmX/32eMeHL4ODlXuU5Uw6dh4OXu2mdT/+OyI9Kqva83oTfTiD4ycGQySsuQtXy4V6I23kMANAooFnl/BjcCblXE+/+iyFqQPKO/QqVfzvY+4bU+rFKr1ZQXzkKvaaidhRE/AZ73zaQ2djUd5hERET1iufUioCDtzu6fvA0CuPTMXTzBwAAfZkOvz80Gz2XvISk/WeQtP8MWjzUHcFPPQijTg+ZwgYJv53A9V8OmZ5n5P7PoGriBjsXR4w/uwLpxy8h/PXlAr0rqinbRg7ov+otKOztYDQYUZpTiD8mf2xaf+s2ILdTYOyJb2CjVMDW2QHjz65A7Ja/cG5RxeHqt9sGZHI53MNaoDgtp8rrOzbzQM8lL0FmYwOZDChKyED4a1+Z1g9aOwfnF29EzoVYOLdoir4rpsOo18OgMyBi3mrTrYeKU7IRuXwbhu9aBKNOj5L0XByf+R0AQOnRCEo3J2jzqzbqOo0WJ2evxKCfZwMyGcoKi3HklaWm9aMOLMaBSYugychD7OYj8OgQiDHHKuK79O0OFMSkAAD8R/ZAqzG9YSjXQ2YjR8Kuk4hZd/De/jhEFpZz8IcqF4iKXz4Frl1HwbXbKBjKtbj0YgCM5VroSwpw8VlfePR7Ej6TP4Zr9zEoiTmNqzM6Q26rhFzljJbT1wr0ToisW+J3ryI/YgfK89Jx7f1BsLF3wn0rrtc4H4modmTGW08wowajvKQU6wImCR1GjU2MXQtbB3uLv+7p06drNT4qKgqTJ0/GmjVrEBJSsz0ZXbo0nHs0CrVduLdtiZCnHsTxt76z+GsDgN/wbnAN8sXFpVsEef3/Emp7p+rVpg7UpQYAwtQBjQ7os9viL3tPwocDKn5dbnFSmwsthTlI1kSsc2FNcbMnSVEqlQgODoZSqRQ6FKuSGxknWEMLAIm7TyFx9ynBXp/EgzWAiHlAJHVirAFsaklSWrZsiZ9//lnoMIhIIKwBRMwDIqkTYw3ghaKIiIiIiIjIarGpJUmJjo5Gr169EB0dLXQoRCQA1gAi5gGR1ImxBrCpJUkxGo0oLy8Hr49GJE2sAUTMAyKpE2MN4Dm1DZRCpcTEWOu5lYJCJZ4TzRsya9suxIrbO1mCvU3FlUytiT1vaUsiwhwksh5sahsomUzGW4ZQFdwuiKRDJuOtOYiExBwksh48/JiIiIiIiIisFr9/Iknx9/fHhg0b4OPjI3QoRCQA1gAi5gGR1ImxBrCpJUmxt7dHQECA0GEQkUBYA4iYB0RSJ8YawMOPSVLS0tKwcOFCpKWlCR0KEQmANYCIeUAkdWKsAWxqSVIKCgqwc+dOFBQUCB0KEQmANYCIeUAkdWKsAWxqiYiIiIiIyGqxqSUiIiIiIiKrxaaWiIiIiIiIrBabWpIUd3d3PPXUU3B3dxc6FCISAGsAEfOASOrEWANkRqPRKHQQRERERERERHXBPbVERERERERktdjUEhERERERkdViU0tERERERERWi00tERERERERWS02tURERERERGS12NQSERERERGR1WJTS0RERERERFaLTS0RERERERFZLTa1REREREREZLXY1BIREREREZHVYlNLREREREREVotNLREREREREVktNrVERERERERktdjUEhERERERkdViU0tERERERERW6/8BZ+Gv/k3eRWQAAAAASUVORK5CYII="
+      "image/png": "iVBORw0KGgoAAAANSUhEUgAAA7UAAAEvCAYAAACaO+Y5AAAAOnRFWHRTb2Z0d2FyZQBNYXRwbG90bGliIHZlcnNpb24zLjEwLjAsIGh0dHBzOi8vbWF0cGxvdGxpYi5vcmcvlHJYcgAAAAlwSFlzAAAPYQAAD2EBqD+naQAAaShJREFUeJzt3XlcVPX+x/HXDDuKIIiCC6KioqRp7kuZVqZetU2zm7bcsn3frPRa1zLby9Zfi926pZmllmaWS1nivlXuioqICCiKLMo+8/uDpAhUQJjDnPN+Ph497uWc78x8Rs7n8+VzVpvT6XQiIiIiIiIi4obsRgcgIiIiIiIiUlVqakVERERERMRtqakVERERERERt6WmVkRERERERNyWmloRERERERFxW2pqRURERERExG2pqRURERERERG3paZWRERERERE3JaaWhEREREREXFbampFRERERETEbampFREREREREbelplZERERERETclppaERERERERcVtqakVERERERMRtqakVERERERERt6WmVkRERERERNyWmloRERERERFxW2pqRURERERExG2pqRURERERERG3paZWRERERERE3JaaWhEREREREXFbampFRERERETEbampFREREREREbelplZERERERETclppaERERERERcVtqakVERERERMRtqakVERERERERt6WmVkRERERERNyWp9EBiJyL9evXV2p8Wloac+fO5eqrr6ZBgwYVek23bt2qEpqIuEhl6kBVagCoDkjtprlQRKw+F+pIrVhKWloa06ZNIy0tzehQRMQAqgEiygMRqzNjDVBTKyIiIiIiIm5LTa2IiIiIiIi4LTW1IiIiIiIi4rbU1IqlBAQEMGjQIAICAowORUQMoBogojwQsToz1gCb0+l0Gh2ESFVV9o6PVVGb7/QmIqoDIsoBEbF6HdCRWrGUvLw8EhMTycvLMzoUETGAaoCI8kDE6sxYA9TUiqXEx8dzzTXXEB8fb3QoImIA1QAR5YGI1ZmxBngaHYCUz+l0UpjjPntPPP18sNlsRodheu62XZiVtndxBXfMd+WGuII75obZ1ESum/X3qrroGmpqa6nCnDxmtBpjdBgVNnrvdLz8fY0Ow/TcbbswK23v4grumO/KDXEFd8wNs6mJXDfr71V10TV0+rGIiIiIiIi4LTW1IiIiIiIi4rZ0+rFYSnR0NOvWrTM6DBExiGqAiPJAxOrMWAN0pFZERERERETclppasZSEhARuueUWEhISjA5FRAygGiCiPBCxOjPWADW1Yik5OTls3bqVnJwco0MREQOoBogoD0Sszow1QE2tiIiIiIiIuC3dKMpEwnrFMGjupFLLCk7kkLkvmb2zl7Pjo4U4ixwGRSdWFHXtxfR9415WPPA2e778ucz6uk1DGbH+/9gzaxkrHnzH9QGKmIzmAZHaR3OhSM1TU2tC8d+sJHHpBrDZ8AsNImpkP7pPupnA1k1Y/dj7RocnIiI1TPOAiIhYiZpaEzq2LZ59c2JLft71ySKujJ1Km+svYdMLM8k7mmlgdMYKDw9n0qRJhIeHGx2KiBjAKjVA84CciVXyQETKZ8YaoGtqLaAwJ4+0TXuw2e3Ua97I6HAMFRgYyODBgwkMDDQ6FBExgFVrgOYB+Sur5oGIFDNjDVBTaxEBkcV/xOSmZxkcibHS09P56quvSE9PNzoUETGAlWuA5gE5xcp5ICLmrAFqak3Iw88Hn+AAfELqERQdQY8pYwnp0JIjv8aRFZ9idHiGSk1N5eWXXyY1NdXoUETEAFapAZoH5EyskgciUj4z1gBLXFOblpbGCy+8wNdff01SUhKhoaFce+21TJ48mdtuu40ZM2bw4YcfMnbsWKNDrRadHh5Jp4dHllqW8P1a1jz+oUERiYiIK2keEBERKzF9U7t582YGDhxIamoqderUoX379iQlJfHaa6+xf/9+Dhw4AMD5559vcKTVZ/eMpcTPW4nd04OgthF0uO9K/BoEUZibXzLG7u3JsEUvse/rFWx5c27J8r5T78E3NIilo58zInSxKKfTaXQIIqaieUDE/WgurJgWV/Yh5q7hBLVuSmFOHod+2czGKdM5cTDN6NDEQKY+/TgtLY2hQ4eSmprKuHHjSElJYdOmTaSmpvLss88yd+5cfvvtN+x2O+edd57R4VabrP0pJMduIWnZb2x7bz4/3vA8DS6IotcLt5WMceQXEnv/W3S8/yrqt28OQMSgbjS9rCsrH37XqNDFZE79Ae3h613ueg9/HwCK/vKHtoicO80DIrWH5sLqE/2vQfT7v4coys1n/X/+x/YPv6Nxv44Mmf8cfo3qGx2eGMjUTe39999PYmIiDz/8MC+++CJ169YtWTdhwgSio6MpLCykTZs2+Pn5GRhpzTqyKY59c2JpefWFhF7QumT5sS3xbH1nHhe+eR/+4cH0eukO1o6fRk6qeS4a/zt/f3969OiBv7+/0aFYQvaBwwAEtWla7vqg1sXLs/4YJ1LTrFoDNA/IX1k1D4yiubB6+NSvS5fxo0nbvJcfrn6aXZ8uZvPUOSy5/jn8G9Wn82OjjA7RbZixBpi2qd2+fTuzZs2iYcOGPPvss2XW22w2unTpApQ99Tg+Pp7hw4cTEBBA/fr1ufHGGzl69KhL4q4pv78+G0dhEZ3HXVdq+eY35+IoKGT4kpdJWbWN+HkrDYrQNSIiInjrrbeIiIgwOhRLOLplH9lJR2hxZZ8ye1DtXp60u2UwToeDxMUbDIpQrMbKNUDzgJxi5TwwgubC6tHs8u541fVjx7SFOIscJcuP/r6X1DU7iBzeG7uX6a+srBZmrAGmbWpnzJiBw+FgzJgxp90L4eNTfLrHX5varKws+vfvz8GDB5k5cyYffPABsbGxDB06FIfDUe77uIOs/SnEz1tJ437n07BHu5LlziIHqet24hsSyJ5ZywyM0DWKiorIzs6mqKjI6FAswVnkYM3jH+IV4M8VP71GlwljaDPmUjo+NIJhi18irHcMW97+hsy9h4wOVSzCyjVA84CcYuU8MILmwuoR2jkKgMMbdpVZd3jDLrwD/AmMauLqsNySGWuAaZvaH3/8EYD+/fufdszBgweB0k3tBx98QFJSEt988w1Dhw5l5MiRzJgxgzVr1jB//vyaDbqGbX5jDo6iolKnZzTs1pbW/xzAjo8W0v2Zf532eg+ziIuLY8CAAcTFxRkdimUc/HETC4f/m5SVW4m6th89p4zlvDuGkXs0k59vf5VNz39udIhiIVavAZoHBJQHRtBceO78w4IBOJl8rMy6E4eKz6j0Dw92aUzuyow1wLTH6BMSEgCIjIwsd31eXh5r164FSje1CxYsoG/fvqUOx/fu3ZvIyEi+/fZbrrzyykrH0rVrV1JSKvdcQC+nnafpXqnXpKzexifhI067PiMuiU+b/vmHjKefD33fuJdNUz5n5/8WMfjrZ+gyYTTrJn5cqc8FaNO6DQU21x/JHjHi9N+3PIcPF1+v8v3337Nx48YKveaqq66qdFw1pSrbRW1w9Pe9/Hz7q0aHUW2M2t6lfJWpA1WpAWBMHXC3eQCUG0bRXOgezDQX1kSun+336uH3xw218grKrDu1zPOPMbWJq+qiGebCsLAwNmyo2mn4pm1qT5w4AUBubm6566dPn05GRgYhISE0afLnqQrbt29n5MiRZca3b9+e7du3VymWlJQUkpKSKvUab5sHNKrSx1VYt0k3kZ14hJ2f/ADAigfeZvjSVziwcB0pq7dV6r0OJR8i3+n6UxhO/Z4rKicnp+R/K/rayv7uapIrtgs5O6O2dylfZepAVWoAGFMH3G0eAOWGUTQXiqvVRK6f7fdalJMHgIePV5k7RXv+cYZJ4R9jahNX1UWzzoUVZdqmNiwsjKysLDZs2EDXrl1LrUtMTOTJJ58EoGPHjqXWpaenExQUVOb96tevz549e6ocS2V5Oe1Qgzt1mvTvRIvhfZh3ySMly7ISUtn43Az6TL2beQMeofBE+TsEytM4vLEhe+fr1KlTqfGnEtfPz6/Cr/3rTg+j1fR2IRVj1PYu5atMHahKDQBj6oC7zQOg3DCK5kJxtZrI9bP9Xk+mFJ927B8eTFZ86TMgT512XN6pyUZzVV00w1xYlZ7pFNM2tZdeeilxcXFMmTKFAQMG0KZNGwDWrFnDDTfcwPHjx4Gydz6G4jsj/925PBC7KofRC07mMqPVmCp/5tkkLfuNz6NvKrN85yc/lOyxr4zdcbvx8vetjtAqZf369ZUav3PnTmbOnMngwYOJjo6u0GumTp1ahchqRk1vF1IxRm3vUr7K1IGq1AAwpg642zwAyg2jaC4UV6uJXD/b7zXttz20vXEgDbu2LdPUNuzaloLsHDL21L4jia6qi2adCyvKtDeKeuKJJ6hfvz6JiYnExMTQoUMHoqKi6NWrF+3ataNv375A2aa2fv36pKeXfT7f8ePHCQ7WxefuLioqikWLFhEVFWV0KCJiANUAEeWBuKcDi9ZTcDKXdmOHYPP4s4UJOb8VjXq2I37+KhwFhQZG6D7MWANM29RGREQQGxvLkCFD8PX1JSEhgdDQUKZNm8a8efPYuXMnULapbdeuXbnXzm7fvp127dqVWS7uxdPTk/r16+PpadqTFETkDFQDRJQH4p7yjmWx6fnPadCxFYPmTqLNDZfR8YGruezzCeQcyeC3l2cZHaLbMGMNMG1TCxATE8N3331HVlYWmZmZrF69mltvvZX09HSSk5Px9PSkffv2pV4zdOhQVqxYQWJiYsmyNWvWsH//foYNG+bqryDV7ODBgzzyyCMlj3MSEWtRDRBRHoj72jFtIcvveQNPPx+6T7qZ9rcPI3n5FhYOm1Byza2cnRlrgKmb2tPZsmULANHR0fj4lL719+233054eDhXXnklCxYsYPbs2Vx//fV0796dK664wohwpRplZ2cTGxtLdna20aGIiAFUA0SUB+Le9s2N5duB45jecjRfxPyLX+56nezEw0aH5VbMWAMs3dSWd5OoevXq8dNPPxEeHs51113H2LFj6d27NwsWLMBut+Q/l4iIiIiISK1lnhOpK+FMTS1Aq1atWLBggStDOmeNLz6fC568HpvNjt3Lg63vzmPvV7+UGhPW5zy6jB+NV11fnA4nBxat59cXZgIQFB1Bz+fH4tcgEEdBIYc37mbthI9w5OuC+9rOw8eLfu89RGDrphTl5JFz5Dirx31A9sEj5Y4Pio6g53O34hsaiM1mY+Pzn3Ng4Vo8/X3p/9GjhHRsCcAXMbdUOIabk2dzbHsCOIpvWb/m3//l8NodZcZ1uPdKWlzRp+Tnus0bEff5j6z/z/8I7dKGXi/cBoDNy5PD63aw9t//rdQ2OPDLpwmOaX762G02ejz7L5pccgE4nWx7fwG7/reoZHWjXu3p9tSNePj5YPews+LBdziycXeFP1/ESJd9MRG/0CBwOCg4kcvaCR9xbNv+UmPqNg2l7xv3EnxeJJnxKSwY9Hip9TF3DSfq2oux2W1k7D3EygffIT/zpOu+hMg56P7sLURc3pW6zRryzcUPcXxX4mnHtv7nADrcexXYbSSv2MKaJz7EWVQ8h8XcOZyoURfjLCyiKK+ANRM+4ujve8/6+RWdC0POb0WPZ28h+LxIDv64iZ9ve7XMGA8fL4YtfonCnPwyeVqeyszhTS+5gE6PjaJ+dAQ7/vs9G575tNR6zYXijizd1P79GbXu7KJ3HmTRiP+QviOBuk1DuSr2DRIWri31jMH8jBPFp2gcOIyHjxcDv3yKFlf1Jf7rFRTl5bN2/Eek70jAZrdz0bsPEHPncLa8OdfAbyUVteuzJST99CsA0f8aRK9X7mTJdc+WGefh582Aj8ex4oG3ObxuJzYPO95BdQFwFBay9Z155KVnMfDLpyodw8JhEyg8eeZnWm55+xu2vP0NAHYvT6797QP2zY0F4Nj2/Xw7+AmchUVgs9F/2qO0vWEgOz5aWKHPj75lMNmJhwmOaX7aMa1GXERgm2Z83ed+vOr5M3zxS6Ss3ELGnkP4NarPhW/cy5LRz5ERl4SHjxcePl4V+/IitcAvt79a0oBGDOpGn9fv5tuB40qNyc/OYdOLM/EO8KfTY6NKrQu/qCNRI/vx3dDxFJ7I5fyHR9L5ietZO36ay76DyLlI+G41W9/9hiHzJp9xXN1mDek87jrmX/YYuWkZDPjkcVpffwm7P1tC/fbNaXfrYL7p9xCFJ3Npec2F9Hx+LN8NebJCMVRkLsw5nM66pz4muEMLGl9U/t+inZ/4J4c37CY4JrJCn1uZOTwzPpmVD79L5LBeePh4l1qnuVDclSXPp121ahVOp5PLL7/c6FCqlXegPwBeAf7kpmeVOcJ1bGs82QeKrzkoyivg2Nb9BDRvBEBWfArpOxIAcDocpP22l4CIhi6M3jVCQ0N54IEHCA0NNTqUalOUV1DS0AIc2RR32t9dy6su5MjG3RxeV3z3b2eRg7yjmQA48gtJXrGF/IwTNR80xX90nzx0lKOb9wFQlJNf3NACHt6eePh643RU7GHlAS3CaHFFH7a8/fUZx0UO782uTxfjdDjIP55N/PxVtLii+PFe0Tdfzt45y8mIK37GXVFegY5QmZAZa8Apf91evQL8cTrKPl89/3g2h9ftpPBkXpl1we0jSV27o2Rn6MGlG2k14qKaC1gMY9Y8SF2zg5PJZ79ZUPOhPUn4fh25aRkA7Pp0MS2v7Fuy3u7pgad/8T1XvOvVqdB7VsbJ5GOk/bYHR15Buesb9mhHvRbh7Ju9vMLvWZk5PHNfMunbE3AWlp1jNRdagxlrgCWP1JrRL3e8Rv+PHqPwZB7egXVYduvLZ3xWl19oEJFDe7JkzJQy6zz9fGgz+hI2TJ5ekyEbIiQkhNGjRxsdRo1qd+sQEpdsLHddUJumFOXmc8mnT1InPJhjOw6wftL/ShrbczFo7iTsnh4kr9jCry9+QWFO2T+a/6r19Zewe+ZPpZbVbRrKgI/HEdAijINLN7F7+tKzf7DNRp9X7mLt+Gk4C4rOOLRukwac+Mtp2dmJR2hwfisAgto0I+tAKgNnPYVvcACpa3ew4bnpFOXknz0GcRtmrwF937yP8N4xACy5/sxHq/7u6O97aXvDZfg2CCQ3LYOW11yEd4A/3kF1yT9unpuJiPnz4GzKzAUHj1CnSQMA0rcnsO39BYxY9y556dkU5Rfww1UVP3upsnPh33n6+dD9mZv56aYXqdcivFKvrQ6aC63BjDXAkkdqzcbmYafDfVfx080vMrvbXSwaOYm+b9xbclrp33nV9eOST59gy7vzOLYlvvR7eXrQ7/2HSfr5dxIXrXdF+C6VmZnJ0qVLycw89yauNupw/9XUaxHGphc+L3e93dODJhd3YvW495l/2WOcOJRGzyljz/lzv+p6JwsGPc7CYRPwDalH16duOON4/8YhNOweXXLq8SnZB48w/7LHmNXxNjx8vGg+pMdZP/u8u4aTsmZ7mWsHT8fp/PPolc1mK/n/dk8PwnrH8PPtr/LtoMfxqleHTo9cW6H3FPdh9hqw4v63+KrrnWx6cSZdn7qxUq9NWb2Nbe9/y6XTx/OPBVPISU0HOOMOUnFPZs+Diig1F/xleZ2mDWh2eVfm9LyHr7rcwfYPFnDROw9U6D0rOxeWp+tTN7Dz40WGPZ5Gc6E1mLEGqKk1geDzWuAfVp/D63cBxXvbc1LTCW5f9tpCzzq+XPb5vzmweAPb3y99MyybpwcXf/Bw8bUeE//rkthd7dChQ4wfP55Dhw4ZHUq1i7lzOM2H9GDp6OdOu0c1+2Aayau2lUyW++bGEto56pw/+0RSGgCFOXns/GQRjXq0O+P41qP6k7how2mP/hSezCX+m5W0vPrCs352o57tiRp1MSPWvcvgeZPxDqzDiHXv4h1Yp8zY7KQ06jb789TsOk0blMSeffAIB5duIj/jBM4iB/HfrCC0c+uzfr64FzPXgL/a+9UvhPWOwad++Ts3T2fXp4tZMOhxvhs6ntQ12zmRlFbq3gxiDlbJg9MpOxeElswFkUN7c3znAXIOHwdgzxfLaNSzHbYKPAGjsnNheRp2b0enh0cwYt279HvvQepHR3DFz69X+n2qSnOhNZixBqipNYETSWn4h4dQr1VjAAIiwwho3ojMfcmlxnn6Fze0ST//xubXZ5daZ/Ow0++9h8hLz2bVo++5LHapHu3vGEqLq/qweNQzZ7z2Zf+3q2jQqRVedf0AaNK/E8e2JVToMwZ++TQNOpVtgL0D6+Dh98eNJmw2WlzRm6Nb95/xvaJG9Sdu5o+llgU0b4TN0wMovolUxJAeHNvxZ2xXxb6Bf1hwmff68cbnmd31LmZ3v5vvr/g3+RknmN397nKvK0r4djVtb7gMm734BlktruhD/PyVAOz7egXhfc7D7l18VUaT/p05tv3M30OktvCq64dfo/olP0cM7k5eejZ56ZU7bdivYRBQfFO5TuOuY8u786ozTJFaIeG7NTQf3B3fBoEAtL1xIPHziueCrAOpNOzeDk9/XwCaDezK8bikkns8VOdcWJ75lzzC7O53M7v73fxy51TSdx5g3sUPlaw/3VxYXcw4F4b1iuHm5Nml/hu95zOGLX6J9rcPxeahdsgMdE2tCeSmZbB63Pv0//CR4huD2GyseXIaJ1OO0fuVO0lcvIHExRtof9sQQjtH4eXvQ/PB3QHYv2A1m9+YS4sr+hD5j54c27af4UteBiB1/S7d9dIN+IcH0/0/N5O5P4VBs/8DQFF+Id/9o/hOjX/dBk4kpbHlra8ZsmAKzsIiTqYcY9Vjf+7EGLb4Jfwa1sc7sA4jN75PyqqtxN73Fja7neCY5pxIPlrm8wOjmtDrpTvA6cTm6cGxLftYO/HjkvWXTh/Pry/PKnkcQnjfDmCD5Ngtpd4nrHcM7W8firPIgc3Tg5QVW0p2vviE1MOnfl3yqnBd3/AlL7NkTPGplHtnLyekUxRXrXwTgK3vziu5GcaRDbtIXLyB4UtfwVlYRPrORFY//kGlP0/ECF71/Ok/7VE8fb1xOpzkHs3kxxufB0rXALu3J9esfgcPH0+8AvwZufF99s75hU1Tii9ZGPjFRLDb8fDyZO/sX9j53++N/FoildJjylgiLu+GX8MgLv/yKQpO5DK3931A6TzIPnCYX1/5kiHzJ2Oz20lesYXdnxfvaD2wcC0NOrVi2KIXKcoroOBEDrH3Fs8Z1TUXBjRvxKC5z+Dp542HrzcjN77P5jfnlnrEXHnONheebg6H0nNhWK8YLnz7frwC/LDZbLS4og9rnvyQxMUbTD0Xxn+zksSlG8Bmwy80iKiR/eg+6WYCWzdh9WPvGx2enCOb868XFUitUXAylxmtxhgdRoWN3jsdrz/2arrS+vWVu+53586d3HjjjXz66adER0dX6DXdunWrSmg1wqjtIrhDC6Jvutywo/gRQ3oQ1KYpm6fOMeTz/86o7V3KV5k6UJUaAMbUAXebB0C5YRTNha6hufBPNZHrNfV7DesVw6C5k9j43PSSxwpC8U25roydSp3wEL7oOLZabppZHlfVRbPOhRWlI7ViKT4+PrRt2xYfHx+jQ3Erx7bEG3pa+oGFazmwcK1hny/moRogojyoKs2F5lKYk0fapj3UHRZKveaNOFJDTW1tZMYaoKZWLKVFixZ89tlnRochIgZRDRBRHoicEhDZCIDc9CyDI3EtM9YANbUiIiIiImJqHn4++AQHlFxT2/bGgYR0aMmRX+PIik8xOjw5R7rdl1jKrl276NOnD7t27TI6FBExgGqAiPJArKnTwyP557aP+efW/3Llstdo969BJHy/lp9uetHo0FzOjDVAR2rFUpxOJwUFBej+aCLWpBogojwQa9o9Yynx81Zi9/QgqG0EHe67Er8GQRTm5peMufiDR3A6nfxyx2sly7yD6nLlz6+z4ZlPiRjU/Yzr982Ndel3qioz1gA1tbWUp58Po/dONzqMCvP0M8+F5rWZu20XZqXtXVzBHfNduSGu4I65YTbumOtZ+1NKHieYtOw3Dq/bweD5k+n1wm0sv+cNAFY//j5XLHudFlf2If6b4mcX95wylsPrdrJvbixJy34943oxjpraWspms+mxCFKGtgsR61C+i5RPuSHV4cimOPbNiSXq2ovZ8dFCjmyKIy89m5UPv8uFb91PyurtNOzalrDeMczr/xDAWdeLcXRNrYiIiIiIWM7vr8/GUVhE53HXlSxL+ulXEhas5qK376fnC7ex6pH/Iy89u8LrxRhqasVSIiMjmTlzJpGRkUaHIiIGUA0QUR6InJK1P4X4eStp3O98GvZoV7J8/X/+R0CLMJJ++pWDP24q87qzra/tzFgD1NSKpfj6+tKqVSt8fXXakogVqQaIKA9E/mrzG3NwFBXR+bFRJcsKc/LITjhM+o4D5b7mbOtrOzPWADW1YinJyclMnjyZ5ORko0MREQOoBogoD8RaUlZv45PwEWx5+5ty12fEJfFp01EsGvEfl8ZlJDPWADW1YikZGRnMnz+fjIwMo0MREQOoBogoD0Sszow1QE2tiIiIiIiIuC090kdEREREROQvfrjm6XNaL66lI7UiIiIiIiLittTUiqXY7XY6d+6M3a5NX8SKVANElAciVmfGGmCebyJSAQ6Hg19//RWHw2F0KCJiANUAEeWBiNWZsQaoqRURERERERG3paZWRERERERE3JaaWhEREREREXFbamrFUgICAhg0aBABAQFGhyIiBlANEFEeiFidGWuAzel0Oo0OQqSq1q9fX+Of0a1btxr/DBGpOtUBsTrlgJhBwclcZrQaY3QY1W703ul4+fvW+OdYvQ54Gh2AiCvl5eVx+PBhGjZsiI+Pj9HhiIiLqQaIKA/Eerzr+TNozqSSnz38fAho3ogvOtxK/vHskuWtRvYj5vahJT/7Nw4hdc0Olt36MgAxdw0n6tqLsdltZOw9xMoH3yE/86Trvkg1MWMN0OnHYinx8fFcc801xMfHGx2KiBhANUBEeSDWk595kvmXPVby3+7pS0n66ddSDS3A3q9+KTUuJzWdfXOXAxB+UUeiRvbju6Hj+abfQxzbup/OT1xvxNc5Z2asAWpqRURERETEMqKu60/czB/POKZBpyh8Q4M4sGgDAMHtI0ldu4PCE7kAHFy6kVYjLqrxWKVidPpxLeV0OinMyTM6jArz9PPBZrMZHYZUI22DxnK3f38w3+/AaE4n5BYZHUXl+HpAdW0CygExmrZBcwrt0gbf+nVJXLLxjONaXz+AvbN/wVlYXIiP/r6Xtjdchm+DQHLTMmh5zUV4B/jjHVS3zBFfcT01tbVUYU6eW10s76qL4MV1tA0ay93+/cF8vwOj5RbBhQuNjqJyYoeAXzX9ZaEcEKNpGzSn1v8cwJ6vfsFZ5DjtGA8/b1oM78N3wyeULEtZvY1t73/LpdPH4ywsImHhWgAcBYU1HrOcnZpaERERERExPU8/HyKH9+a7IU+ccVzk0F4cjztIxu6DpZbv+nQxuz5dDEDoBa05kZRWcjqyGEtNrVhKdHQ069atMzoMETGIaoCI8kCsK3JYL9J3JJCx59AZx7W+bgBxM38qs9yvYRA5h4/j4edNp3HXseXdeTUVao0yYw1QUysiIiIiIqbX+vpLiPu8dLPa+5U7SVy8gcTFxTeECmjeiJCOLVl64/NlXj/wi4lgt+Ph5cne2b+w87/fuyRuOTvd/VgsJSEhgVtuuYWEhASjQxERA6gGiCgPxLq+v3Iie2YtK7Vs1aPvlTS0AFkJqcxofUO5pxXPG/AI8y5+iLl97uP312fXeLw1xYw1QE2tWEpOTg5bt24lJyfH6FBExACqASLKAxGrM2MNUFMrIiIiIiIibktNrYiIiIiIiLgt3SjKRMJ6xTBo7qRSywpO5JC5L5m9s5ez46OFZ3wml8i50jYoIiJWp7lQxPXU1JpQ/DcrSVy6AWw2/EKDiBrZj+6TbiawdRNWP/a+0eEZKjw8nEmTJhEeHm50KKambVBqK9UAEeWBq2gulNrKjDVATa0JHdsWz745sSU/7/pkEVfGTqXN9Zew6YWZ5B3NNDA6YwUGBjJ48GCjwzA9bYNSW6kGiCgPXEVzodRWZqwBuqbWAgpz8kjbtAeb3U695o2MDsdQ6enpfPXVV6SnpxsdiqVoG5TaQjVARHlgFM2FUluYsQaoqbWIgMji4pmbnmVwJMZKTU3l5ZdfJjU11ehQLEfboNQGqgEiygMjaS6U2sCMNcASTW1aWhqPPvoorVq1wtfXl2bNmvHII4+Qk5PDmDFjsNlsTJs2zegwq42Hnw8+wQH4hNQjKDqCHlPGEtKhJUd+jSMrPsXo8MQCtA2KiIjVaS4UcR3TX1O7efNmBg4cSGpqKnXq1KF9+/YkJSXx2muvsX//fg4cOADA+eefb3Ck1afTwyPp9PDIUssSvl/Lmsc/NCgisRptg67T4so+xNw1nKDWTSnMyePQL5vZOGU6Jw6mGR2a1CIFxw+T+s0rZGxcSP7hBLDb8QpsiH+rLtTvey31e11tdIg15tSdaDc+N50tb39T7pjRez7j6O/7+OGap10bnJia5sLq1+G+qwjp0JKQji0JaN6I7MTDzO5+t9FhSS1g6qY2LS2NoUOHkpqayrhx45g4cSJ169YFYPLkyUycOBFPT0/sdjvnnXeewdFWn90zlhI/byV2Tw+C2kbQ4b4r8WsQRGFufskYu7cnwxa9xL6vV7Dlzbkly/tOvQff0CCWjn7OiNDFJLQNukb0vwbRc8pYUtftYP1//odPcADtb/sHjXo+x4LBT5CTap5rZaTq8o8ksuPRbjhysgjuNxq/QXcBkJe8h4yN3+HIzTZ1UytiFM2F1a/L+NHkHsvi2JZ9eNfzNzocqUVM3dTef//9JCYm8vDDD/Piiy+WWjdhwgRmzJjBzp07iY6Oxs/Pz6Aoq1/W/hSSY7cAkLTsNw6v28Hg+ZPp9cJtLL/nDQAc+YXE3v8Wg79+hoNLN5K+PYGIQd1oellX5g142Mjwa5S/vz89evTA31+FsCZpG6x5PvXr0mX8aNI27+WHq58ueeZh0rLfGLrweTo/NopVj75ncJS1jxVrQMrXL1N4PJVWE+YT1H1YqXXNxr5OftpBgyITo1gxD4ygubD6ze5xN9kHDgNwxbLX8Krja3BE7smMNcC019Ru376dWbNm0bBhQ5599tky6202G126dAFKn3p88OBB7r//fnr37o2/vz82m439+/e7KuwacWRTHPvmxNLy6gsJvaB1yfJjW+LZ+s48LnzzPvzDg+n10h2sHT/N1Ed3IiIieOutt4iIiDA6FEvRNlj9ml3eHa+6fuyYtrCkoQU4+vteUtfsIHJ4b+xept5vWSVWrAF5h3YDENChf7nrvRs0dWU4UgtYMQ9qA82F5+5UQyvnxow1wLRN7YwZM3A4HIwZM+a0eyF8fHyA0k3tnj17mDVrFkFBQfTr188lsbrC76/PxlFYROdx15VavvnNuTgKChm+5GVSVm0jft5KgyJ0jaKiIrKzsykqKjI6FMvRNli9QjtHAXB4w64y6w5v2IV3gD+BUU1cHVatZ8Ua4BPWEoC0xR/idDoNjsY4JTftKec/q7FiHtQWmgulNjBjDTBtU/vjjz8C0L9/+XumofioLJRuai+66CJSU1NZuHAho0aNqtkgXShrfwrx81bSuN/5NOzRrmS5s8hB6rqd+IYEsmfWMgMjdI24uDgGDBhAXFyc0aFYjrbB6uUfFgzAyeRjZdadOHS0eEx4sEtjcgdWrAGNrnwUu389Dv73YbaMbU78q6NJnT+VE3s2Gh2aS3V6eCT/3PZxuf951THPJUgVYcU8qC00F0ptYMYaYNqmNiEhAYDIyMhy1+fl5bF27VqgdFNrt5v2n4TNb8zBUVRE58f+bNYbdmtL638OYMdHC+n+zL/w8PU2MEIxO22D1cfDr/hMk6K8gjLrTi3z/GOMWJtPWEvaT/2d0MF3g9PBseWfc/Cjh9j5SFe239/RMs3t7hlLWXTtpHL/K8zJMzo8sRDNhSLVz7QXXJ04cQKA3NzcctdPnz6djIwMQkJCaNKkZk/R69q1KykplXsemZfTztN0r9RrUlZv45PwEaddnxGXxKdN/yygnn4+9H3jXjZN+Zyd/1vE4K+focuE0ayb+HGlPhegTes2FNgcZx9YzUaMOP33Lc/hw8XXYnz//fds3FixP+SuuuqqSsdlBtoGjXW2f/+iP/4I9/Dxougvd9IE8PzjjyFX/6HuDnWgKjUAjKkDNm8/Gk2tnr3oPo0iibjzHSLufIeCY8lk71rN0Z/+R8a6+eyZPJSYt7bhGXDuR/bbtGmNMz+nGiKuWg06k7/etOfvnI7q2W7dIQdAc2FFVXUb1FxYNdWd87WFq34nZpgLw8LC2LBhQ5Vea9qmNiwsjKysLDZs2EDXrl1LrUtMTOTJJ58EoGPHjjUeS0pKCklJSZV6jbfNAxrVUEB/6DbpJrITj7Dzkx8AWPHA2wxf+goHFq4jZfW2Sr3XoeRD5Dtdf17+qZ0XFZWTk1PyvxV9bWV/d2ahbdBYZ/v3P5lSfNqxf3gwWfGld5qdOu24vFOTa5I71IGq1AAwpg7YffxrJAW9gsOp3+tq6ve6mn2vXk/68plkbFxIyMVjzvm9Dx06hCPvZDVE6ZoaVN3cIQdAc2FFuWob1FxYzB1zviJc9Tsx61xYUaZtai+99FLi4uKYMmUKAwYMoE2bNgCsWbOGG264gePHjwOlTz2uKWFhYZV+jZfTDjW4U6dJ/060GN6HeZc8UrIsKyGVjc/NoM/Uu5k34BEKT5R/lLs8jcMbG7JnsE6dOpUafypx/fz8Kvzamj6SX1tpGzTW2f79037bQ9sbB9Kwa9syTW3Drm0pyM4hY49rJx93qANVqQFgTB2wedf8dZ512/YifflMCo5Wz7bSuHHjaj1SW5M1qCa4Qw6A5sKKcsU2qLnwT+6Y8xXhqt+JGebCqvRMp5i2qX3iiSf44osvSExMJCYmhujoaHJycti7dy/Dhg2jWbNmLFu2zCVNbVUOoxeczGVGq3Pfa346Sct+4/Pom8os3/nJDyV7Citjd9xuvPxd/6yw9evXV2p8YWEhN998MwEBAXh6Vmzznzp1ahUic3/aBo11tn//A4vW0/1kLu3GDmHf3NiSx/qEnN+KRj3bEffFMhwFha4KF3CPOlCVGgDG1IGcQrhw4bm/T9bmZdRp2xO7T+km2elwcHz9twD4Nmt/7h8E7N4dh181/WVR0zWoJrhDDoDmwopyxTaoufBP7pjzFeGq34lZ58KKMm1TGxERQWxsLOPGjWP58uUkJCQQExPDk08+yS233FKyp8EVTa3UHp6entSvX9/oMETOWd6xLDY9/zk9nr2FQXMnsXf2cnyDA2h/+1ByjmTw28uzjA6xVrJiDUid9yrZO1YS2G0o/q264OEfSMHxFI6vmsPJvRsJ6NCfwK7/MDpMcSEr5oGYQ8sRF1G3aSgAviH1sHt50vHBawDIzzjBzo8rvyPAisxYA0zb1ALExMTw3XfflVl+7NgxkpOT8fT0pH376tk7Le7h4MGDvP766zz00EM0bdrU6HBEzsmOaQvJO5ZFzJ3D6D7pZgpz8klevpmNU2aUXHMrpVmxBoRd+2/SV35F9rblZP66mMLsY3j41sG3aTua3vIqoUPuwWbiO/9LWVbMAzGHNv+8hLDeMaWWXfD4PwHITjyspraCzFgDTN3Uns6WLcV3P4yOjsbHp+wjL2bPng38edrw999/T2hoKJGRkWVuOiXuJTs7m9jYWG677TajQxGpFvvmxrJvbqzRYbgNK9aAum17UrdtT6PDMMzZ7kQLMCPqBhdFUztYMQ/EHH645mmjQzAFM9YASze1pzv1eOTIkaV+vvvuuwG46aab+OSTT2o0NhEREREREak4NbXlcDqdrgynWp3/8Eg6PzaKby5+iOO7EkutC+sVw6UzxpO591DJsu+GTSh5xmXMXcOJuvZibHYbGXsPsfLBd8jPrJ5HM4j5BLQI48I37sMnOID8zBOsePAdMnYfLDWm1ch+xNw+tORn/8YhpK7ZwbJbX6Zus4ZcPO0R7HY7Ng87GXsOseqx98jPKL4j383Jszm2PQH+eH7kmn//l8Nrd7juC4qIiFRC92dvIeLyrtRt1rDU32G+IfW48K37CGgeRlF+Aasf/4DD63aW+x4xdw4natTFOAuLKMorYM2Ejzj6+15Xfg0Rt2TpptYVz6h1peAOLQjt0obsg0dOO+b47oMsGPR4meXhF3UkamQ/vhs6nsITucXN8RPXs3b8tJoMWdxY75fuYPf0Jez58mea/6MnfV69i4XDJpQas/erX9j71S8lP1/x06vsm7scgJOpx/j+ioklO1W6P3Mz5z80gvX/+V/J+IXDJlB4suKPMhARETFKwner2fruNwyZN7nU8i4TxnBkYxxLrn+OkPNb0X/ao8zpeU/JXetPqd++Oe1uHcw3/R6i8GQuLa+5kJ7Pj+W7IU+68muIuCVL3hli1apVOJ1OLr/8cqNDqTZ2b096ThnLmic+hCocaQ5uH0nq2h0lz0I7uHQjrUZcVN1hGi40NJQHHniA0NBQo0Nxa74h9Qjp0JK9c4ob1ITv1hAQ0bDkjoTladApCt/QIA4sKr5W3ZFfWNLQ2ux2PP19cTrc9ywJcQ+qASLKg5qSumYHJ5PL3qQvcngvdvzxeJ6jv+8l58hxGnVvV+572D098PQvvt+Ld7065b6fyLkyYw2w5JFaM+r82HXsmxNLduLhM44LbNWYYYtfwlnkIO6LZez63yKguMi2veEyfBsEkpuWQctrLsI7wB/voLrkH892xVdwiZCQEEaPHm10GG6vTpMGnEw9Vmovc3ZSGnWaNjjtmQKtrx/A3tm/4CwsKllm9/Jk6MLnqdM0lPTtCfx40wulXjNo7iTsnh4kr9jCry9+QWFOXs18IbEM1QAR5YEr+dSvi81uJ+9oZsmy7MQj1GnaoMzY9O0JbHt/ASPWvUteejZF+QX8cNVTrgxXLMKMNcCSR2rNJrRLGxp0anXWh3Qf3bKPLy+4g28HjuOnW16i7Y0DiRzWCyi+O+S297/l0unj+ceCKeSkpgPgKCis8fhdKTMzk6VLl5KZmXn2wXJGfz8hwGaznXash583LYb3IW7mT6WWOwoKmX/ZY8zqOJaMvUm0vXFgybqvut7JgkGPs3DYBHxD6tH1KWvdnVRqhmqAiPLA1f5+r5bTzZd1mjag2eVdmdPzHr7qcgfbP1jARe884IoQxWLMWAPU1JpAWK/2BEY1YcS6dxmx7l38w0O4bOa/aTKgc6lxBdk5FGQV3/jpZPIx4r9ZQaMef57+suvTxSwY9DjfDR1P6prtnEhKKzkd2SwOHTrE+PHjOXTo0NkHy2mdSEqjTngwNo8/S0idxiGcOJhW7vjIob04HnewzI2kTnEUFLLni2WlTnk/kVT8XoU5eez8ZFGpbVWkqlQDRJQHrpSXXny2m09IvZJldZo2KHe+jBzam+M7D5Bz+DgAe75YRqOe7fQcaal2ZqwByhIT2PL2N3zZ+XZmd7+b2d3v5mTyUZb8czJJP/1aapxfwyD4Y++gZx1fml7ahaNb95deT/FRtU7jrmPLu/Nc9A3E3eQezeTY1v20uqa4CW3+j55kJx45/anH1w0oc5S2TpMGePr98Zxom43I4b05tuMAAN6BdfDw8y5Z1+KK3qW2VREREXeR8O1q2t08CICQ81vh1zCI1HVl7+afdSCVht3b4envC0CzgV05HpeE0+EoM1ZEStM1tSbX+5U7SVy8gcTFG2j+j560velynIVF2Dw9SPh2NXu++LPRGPjFRLDb8fDyZO/sX9j53+8NjFxqu1Xj3qfv1HvocP/VFGTnsOL+t4DS2xxAQPNGhHRsydIbny/1+qC2zegyYQwANruNo1viWffvjwAIjGpCr5fuAKcTm6cHx7bsY+3Ej1347URERCqnx5SxRFzeDb+GQVz+5VMUnMhlbu/72DB5Ohe+fT9Xr3yLooJCYu97q+SeFJ0eG0VOajq7Pl3MgYVradCpFcMWvUhRXgEFJ3KIvfdNg7+ViHtQU2tCs7vfXfL/Vz36Xsn/3/nxD+z8+PTX3c4b8EiNxiXmkrn3UJlH+EDpbQ4gKyGVGa3LXg+b9NOvZc4mOOXIxt3Mv0Tbo4iIuI+146eV+yjE3LQMllz3bLmv+e3lWaV+3jTlczZN+bxG4hMxM51+LJbi4+ND27Zt8fHxMToUETGAaoCI8kDE6sxYA3SkViylRYsWfPbZZ0aHISIGUQ0QUR6IWJ0Za4CO1IqIiIiIiIjbUlMrlrJr1y769OnDrl27jA5FRAygGiCiPBCxOjPWADW1YilOp5OCgoIyD0IXEWtQDRBRHohYnRlrgJpaERERERERcVu6UVQt5ennw+i9040Oo8I8/cxz9zQppm1QxFi+HhA7xOgoKsfXo/req6Zq0Nze95GTmo5fo/pcveqtan1v1SFzcbd5ENx7G3THf++KcOffiTtRU1tL2Ww2vPx9jQ5DLEzboIixbDbws/AsXVM1yGa3lfyvapycieZB19K/t5wLC0+XYkWRkZHMnDmTJk2aGB2KiGE8fLxoPfpSIof2JKhtBN4BfuRlnODY1nj2L1jD3q9+xpFfCEDEoG4Ex7Tgt1e/NDjq6qEaIKI8ELE6M9YANbViKb6+vrRq1croMEQMU7dpKJdMH0/9ts04FLuFre98Q+7RTHzqBxDWO4ZeL95Gg06tWP3Y+wBEDOpO1Kj+pmlqVQNElAciVmfGGqCmViwlOTmZjz76iFtvvZXw8HCjwxFxKQ8fLy757EkCWzVm2W2vkrBgdan1296bT1DbZjS++HyDIqx5qgEiygMRqzNjDdDdj8VSMjIymD9/PhkZGUaHIuJyrf95CfWjI9j2/rdlGtpTju9KZPv7CwAYse5dokb1B+Dm5Nkl/4X1inFZzNVNNUBEeSBidWasATpSKyJiEZHDegGw+7MlFRq/7qmPaX/HMMJ6tmf5vW+ULD8ed7BG4hMRERGpCjW1IiIWERQdQX7WSbISUis0/sAP64kY1B16tmffnNgajk5ERESkanT6sYiIRXgH+FGQlWN0GCIiIiLVSk2tWEpwcDA33XQTwcHBRoci4nL5WTl41bX2MwBVA0SUByJWZ8YaoKZWLMVut+Pl5YXdrk1frOf4zgN416tDQPNGRodiGNUAEeWBiNWZsQaY55uIVEBaWhrTpk0jLS3N6FBEXG7/H3c8bjP60gq/xul01lQ4hlANEFEeiFidGWuAmloREYuIm/kTx3cnEnPnMCKG9Ch3TFB0BDF3DCv5ufBELgDegXVcEqOIiIhIZenuxyIiFlGUm8/SG57n0unjGfDRYxz65XeSfvmdvGNZ+NQPIKxXe5pc0pm4GT+WvObIr3G0A3pOGcvBZb/iLCgiecUWco9mGvdFRERERP5CTa2IiIVkHzjMt5ePo83oS4kc2ouO912FV10/8jJOcGzLPlY9+j775iwvGR//9UpCOraixRV9iLyiN3YPD364+mlSVm8z8FuIiIiI/ElNrVhKQEAAgwYNIiAgwOhQRAxTlJPPjmkL2TFt4VnHOh0O1j/9Ceuf/qTmA3MB1QAR5YGI1ZmxBqipFUtp0qQJzzzzjNFhiIhBVANElAciVmfGGqAbRYml5OXlkZiYSF5entGhiIgBVANElAciVmfGGqCmViwlPj6ea665hvj4eKNDEREDqAaIKA9ErM6MNUCnH9dSTqeTwhz32Xvi6eeDzWYzOgypRk4n5BYZHUXF+XqANkExE3fLQVAeirkoB0Xch5raWqowJ48ZrcYYHUaFjd47HS9/X6PDkGqUWwQXnv0+QrVG7BDwU0UTE3G3HATloZiLclDEfej0YxEREREREXFbampFRERERETEbekEBbGU6Oho1q1bZ3QYImIQ1QAR5YGI1ZmxBuhIrYiIiIiIiLgtNbViKQkJCdxyyy0kJCQYHYqIGEA1QER5IGJ1ZqwBamrFUnJycti6dSs5OTlGhyIiBlANEFEeiFidGWuAmloRERERERFxW7pRlImE9Yph0NxJpZYVnMghc18ye2cvZ8dHC3EWOQyKTkRERMSasrb8zO5/9y+1zO5bB98mbQnufyMNh9yLzcPDmOBETEBNrQnFf7OSxKUbwGbDLzSIqJH96D7pZgJbN2H1Y+8bHZ6IiIiIJdXvO4rAbkPB6aTg2CGO/vQJB6c9SO6BbTS/5wOjwxNxW2pqTejYtnj2zYkt+XnXJ4u4MnYqba6/hE0vzCTvaKaB0RkrPDycSZMmER4ebnQoImIA1QAR5YGR/Ft0IuTiMSU/hw6+i233tCNtyTQaj34Wr6BGBkYnVmHGGqBrai2gMCePtE17sNnt1Gtu7WIZGBjI4MGDCQwMNDoUETGAaoCI8qA28fCvR522vcDpJC9ln9HhiEWYsQaoqbWIgMjiZjY3PcvgSIyVnp7OV199RXp6utGhiIgBVANElAe1idPpJC9lDwCe9RoYHI1YhRlrgJpaE/Lw88EnOACfkHoERUfQY8pYQjq05MivcWTFpxgdnqFSU1N5+eWXSU1NNToUETGAaoCI8sBIjryTFGamUZBxhJP7N3PgndvJif8d/9bd8G3c2ujwxCLMWAMscU1tWloaL7zwAl9//TVJSUmEhoZy7bXXMnnyZG677TZmzJjBhx9+yNixY40OtVp0engknR4eWWpZwvdrWfP4hwZFJFZXcPwwqd+8QsbGheQfTgC7Ha/Ahvi36kL9vtdSv9fVRofoljrcdxUhHVoS0rElAc0bkZ14mNnd7zY6LKlFNl5hq/DY8z6Ix6dRZM0FUwtk7kumKDcfgKL8QhxFRdh1x1lxoeQvnyX5y2f/XGCzUa/LYJrfM824oERMwPRN7ebNmxk4cCCpqanUqVOH9u3bk5SUxGuvvcb+/fs5cOAAAOeff77BkVaf3TOWEj9vJXZPD4LaRtDhvivxaxBE4R8TOYDd25Nhi15i39cr2PLm3JLlfafeg29oEEtHP2dE6GJC+UcS2fFoNxw5WQT3G43foLsAyEveQ8bG73DkZqupraIu40eTeyyLY1v24V3P3+hwpBaKfOizUj/nJu4gZfYUgnpeRdDf8s4zMNSVobmM0+kk4bs17PxkESkrt5Yszzuaydxe99H2xstoM+YyfILqGhilWEXIpbcSfOF1YLNh9/HHt3EbPOuFGB2WiNszdVOblpbG0KFDSU1NZdy4cUycOJG6dYsnrcmTJzNx4kQ8PT2x2+2cd955BkdbfbL2p5AcuwWApGW/cXjdDgbPn0yvF25j+T1vAODILyT2/rcY/PUzHFy6kfTtCUQM6kbTy7oyb8DDRoYvJpPy9csUHk+l1YT5BHUfVmpds7Gvk5920KDI3N/sHneTfeAwAFcsew2vOr4GRyS1zV/vsgrFz8pMmT0Fv8iOZdaZkaOgkBUPvcu+OcvLXZ+deJiNz80gbuZPXPb5vwmw+M0Upeb5hkdRr9OlRochYjqmvqb2/vvvJzExkYcffpgXX3yxpKEFmDBhAtHR0RQWFtKmTRv8/PwMjLRmHdkUx745sbS8+kJCL/jzeo1jW+LZ+s48LnzzPvzDg+n10h2sHT+NnFTzXDT+d/7+/vTo0QN/fx3VcpW8Q7sBCOjQv9z13g2aujIcUznV0ErFqQZYh9PpZPXjH5y2of2rzH3JLL7uWXLTMlwQmfGUByLWZsYaYNqmdvv27cyaNYuGDRvy7LPPlllvs9no0qULUPrU49mzZ3PNNdfQvHlz/P39iY6OZsKECWRnZ7ss9prw++uzcRQW0XncdaWWb35zLo6CQoYveZmUVduIn7fSoAhdIyIigrfeeouIiAijQ7EMn7CWAKQt/hCn02lwNGJ1qgHWcWTDLuJm/lTh8Vn7U9jy9jc1F1AtojwQsTYz1gDTNrUzZszA4XAwZsyY0+6F8PHxAUo3ta+88goeHh5MmTKF77//njvvvJN3332Xyy+/HIfD4ZLYa0LW/hTi562kcb/zadijXclyZ5GD1HU78Q0JZM+sZQZG6BpFRUVkZ2dTVFRkdCiW0ejKR7H71+Pgfx9my9jmxL86mtT5UzmxZ6PRoYkFqQZYx85PFlX6NXGzllF4Mq8GoqldlAci1mbGGmDapvbHH38EoH//8k95BDh4sPhavr82td9++y1ffvklo0ePpl+/fjz44IO89dZbrFq1ihUrVtRs0DVs8xtzcBQV0fmxUSXLGnZrS+t/DmDHRwvp/sy/8PD1NjDCmhcXF8eAAQOIi4szOhTL8AlrSfupvxM6+G5wOji2/HMOfvQQOx/pyvb7O6q5FZdSDbCGgpO57F+wutKvyz+eTeKSDTUQUe2iPBCxNjPWANPeKCohIQGAyMjIctfn5eWxdu1aoHRTGxpa9u6Pp05TTkpKqlIsXbt2JSWlcs+H9XLaeZrulXpNyuptfBI+4rTrM+KS+LTpnw2tp58Pfd+4l01TPmfn/xYx+Otn6DJhNOsmflypzwVo07oNBTbXH8keMeL037c8hw8XX4P4/fffs3FjxZqpq666qtJxmYHN249GU6un2Pk0iiTizneIuPMdCo4lk71rNUd/+h8Z6+azZ/JQYt7ahmdA8Dl9Rps2rXHm51RLvLVBVWqA0dyhDlSlBoAxdaA6c9BVakse1nf68Iijc5Ve+8RdD7LynuRqjqhmaS6sGdWZgwEdLqbLvJq/BKe25KC4nhnmwrCwMDZsqNqORdM2tSdOnAAgNze33PXTp08nIyODkJAQmjRpcsb3Wr68+CYT0dHRVYolJSWl0g2xt80DavgmjN0m3UR24hF2fvIDACseeJvhS1/hwMJ1pKzeVqn3OpR8iHyn609hOPV7rqicnJyS/63oa6u6M8Pd2X38a2QT9AoOp36vq6nf62r2vXo96ctnkrFx4TnfifXQoUM48k5WU5TGc0UNqG7uUAeqUgPAmDpQUzlYk2pLHuZ6+EMVn1CUkZFB0kn3qvuaC2uGclDciVnnwooybVMbFhZGVlYWGzZsoGvXrqXWJSYm8uSTTwLQsWPHM75PUlISTz31FAMHDqRz56rt9Q0LC6v0a7ycdqjBAx5N+neixfA+zLvkkZJlWQmpbHxuBn2m3s28AY9QeKL8HQLlaRze2JAjNHXq1KnU+FOJ6+fnV+HXnm2nh1nZvGv+juB12/YifflMCo6ee5Fs3LixqfZO13QNqAnuUAeqUgPAmDrgihysbrUlD72cdgodDjyrcpVVoC9N6rtX3ddcWDOUg+JOzDAXVqVnOsW0Te2ll15KXFwcU6ZMYcCAAbRp0waANWvWcMMNN3D8+HGg9KnHf5ednc0VV1yBl5cXH39c+VNyT6nKYfSCk7nMaFVzzxBMWvYbn0ffVGb5zk9+KDlyWxm743bj5e/6Z2SuX7++UuN37tzJzJkzGTx4cIWPvE+dOrUKkbm/nEK4cOG5v0/W5mXUadsTu0/pPw6cDgfH138LgG+z9uf8Obt3x+FnoopW0zWgJrhDHahKDQBj6kB15aAr1aY8jL3vTfbOPvvjfP7Kp35dFmxcj6efTw1FVTM0F9YM5aC4E7POhRVl2s3+iSee4IsvviAxMZGYmBiio6PJyclh7969DBs2jGbNmrFs2bLTNrW5ublcccUVxMfHExsbS+PGjV38DaQmREVFsWjRIgICAowOxTJS571K9o6VBHYbin+rLnj4B1JwPIXjq+Zwcu9GAjr0J7DrP4wO0y21HHERdZsWn2PpG1IPu5cnHR+8BoD8jBPs/LjyO6jMTjXAOtrePKjSTW3UdQPcrqGtCuWBiLWZsQaYtqmNiIggNjaWcePGsXz5chISEoiJieHJJ5/klltuKTl8Xl5TW1BQwLXXXsu6dev48ccfad/+3I8iSe3g6elJ/fr1jQ7DUsKu/TfpK78ie9tyMn9dTGH2MTx86+DbtB1Nb3mV0CH3YLOb9kbsNarNPy8hrHdMqWUXPP5PALITD6upLYdqgHWEXtCaNmMuY/f0JRUaX69lOB3utcbNkJQHItZmxhpg2qYWICYmhu+++67M8mPHjpGcnIynp2eZhtXhcHDDDTewZMkSFi5cSPfu7nX3UTmzgwcP8vrrr/PQQw/RtGlTo8OxhLpte1K3bU+jwzClH6552ugQ3I7Va4Cr7sBaG9hsNno+PxZHYSF7vjjzc9gDo5pw2ef/xjfYPEctzsTqeSBidWasAZY8PLJlyxag+G7GPj6lTzO69957mTVrFo888gh+fn6sWbOm5L9Tz7UV95WdnU1sbCzZ2dlGhyIiBlANsBa7pwd9XrubAR+Po3G/smdmBbQIo9vTN/GP76ZQt1kVb5fshpQHItZmxhpg6iO1p3OqqS3v1OOFC4vvCPDcc8/x3HPPlVr39NNP85///KfG4xMREZHqYbPZiBjUnYhB3ck6kEpWfAqOwiJ8Q+oR0rGlLn8QETEBNbV/s3//fhdHc+48fLzo995DBLZuSlFOHjlHjrN63AdkHzxSZmzXiTfQ5JILcBY5yEvPYtWj75G1P4XwCzvQ7akbS8b5Nggk58hxvh04zpVfRUREpMYERDQiIMLdnjwq7mLLbZHYvHyxexffBT7smicJvnBUqTHZO1dz4L27AHAWFlC3fV+a3fYmdi8f8lL3s/XOKPyan1cyvtXjc/AJb+W6LyHipizd1J7tGbXuZNdnS0j66VcAov81iF6v3MmS654tNabZ5d1o1LM98y99FGdhER0fvIYLnryeX+54jeTYLcy/7LGSsZd8+iQpK7e69DuIiIiIuLNWj88u1ZT+nX+L82n3ynpsnl44HQ72vTiCtB/ep+Gw+wHwqBNE+6m/uShaEfOw5Dk3q1atwul0cvnllxsdSrUoyisoaWgBjmyKIyCiYbljPbw98fDxAsCrrh8nk4+WGePXqD7hfc9j7+xfaiZgA4WGhvLAAw8QGmqda6dE5E+qASLKAyPZffyxeRb/HeYszMeRnwM6BV5czIw1wJJHas2u3a1DSFyysczyxMUbCOsdw6jN0yjMzuFEyjF+uKrs3VOjrr2Ygz/9Su7RTFeE61IhISGMHj3a6DBExCCqASLKg5oU/9ponE4HdVr3oMmNz+MVWLZpyEvdz97nryQveQ+BXf5Bg4G3l6xz5GSy45FuOB1FBPW4kvCRE7B5eLjyK4gFmLEGaNeQyXS4/2rqtQhj0wufl1kX0rElgVFN+Krz7czqdDvJsVvoOeXWMuOiRvUn7vMfXRGuy2VmZrJ06VIyM83XsIvI2akGiCgPakrbKctp/8bvtH9tE571Qtj/xk3ljvNpFEn7qb/R8ZMUHIV5HF89FwCv4HA6fHSQdq+up80zS8neHkvqvFdd+RXEIsxYA9TUmkjMncNpPqQHS0c/R1FOfpn1UddeTMrKreRnngSnk71f/kxYn9LXfTTq2Q5PPx8O/fy7q8J2qUOHDjF+/HgOHTpkdCgiYgDVABHlQU3xDo0AwObpRaNhD5K9LfaM4z386hLc9zqOLZ8BgN3LB6+g4svHPAOCaXDpLWRvP/N7iFSFGWuAmlqTaH/HUFpc1YfFo54pblrLkXUglfALO2DzLD6NpdnArqTvPFBqTOvrBrDny2U4HY4aj1lERETEDIpyT1CYfbzk52OxM/Fv2bnMuLzkvTgLCwBwFORzfM1c/JoX37i04Pjhv6zLI331XPxblH0PESlL19SagH94MN3/czOZ+1MYNPs/ABTlF/LdP56k9yt3krh4A4mLN7Dz4x8Iat2UK5e9RlF+ITmH01n92Psl7+NZx5fm/+jJvEseMeibiIiIiLifwuOp7H3hGnAU4cSJT6OWRD74KQD73xpLUPfhBPUYTtbWn0md/zo2uwfOokICOg4gfNREALJ3rODQ50+VWhd27QQjv5aI21BTawInk4/xSfiIctetevS9kv/vyC8s9fPfFZ7IZUbrG6o9PhEREREz8wlrSfupv5a7LvK+aSX/v8Flt9LgsrL3MwGo3+tq6ve6ukbiEzE7nX4sluLj40Pbtm3x8fExOhQRMYBqgIjyQMTqzFgDdKRWLKVFixZ89tlnRochIgZRDRBRHohYnRlrgI7UioiIiIiIiNtSUyuWsmvXLvr06cOuXbuMDkVEDKAaIKI8ELE6M9YANbViKU6nk4KCApxOp9GhiIgBVANElAciVmfGGqBramspTz8fRu+dbnQYFebpZ54LzaWYrwfEDjE6iorz9TA6gurlbjUAVAeqm7vlIJgvD8XalIMi7kNNbS1ls9nw8vc1OgyxMJsN/FQhDKMaIMpBEWMpB0Xch04/FhEREREREbel/U9iKZGRkcycOZMmTZoYHYqIGEA1QER5IGJ1ZqwBamrFUnx9fWnVqpXRYYiIQVQDRJQHIlZnxhqg04/FUpKTk5k8eTLJyclGhyIiBlANEFEeiFidGWuAmlqxlIyMDObPn09GRobRoYiIAVQDRJQHIlZnxhqgplZERERERETclppaERERERERcVtqakVERERERMRtqakVS7Hb7XTu3Bm7XZu+iBWpBogoD0Sszow1wDzfRKQCHA4Hv/76Kw6Hw+hQRMQAqgEiygMRqzNjDVBTKyIiIiIiIm5LTa2IiIiIiIi4LTW1IiIiIiIi4rbU1IqlBAQEMGjQIAICAowORUQMoBogojwQsToz1gCb0+l0Gh2ESFWtX7++xj+jW7duNf4ZIlJ1qgNidcoBEbF6HdCRWrGUvLw8EhMTycvLMzoUETGAaoCI8kDE6sxYA9TUiqXEx8dzzTXXEB8fb3QoImIA1QAR5YGI1ZmxBngaHYCUz+l0UpjjPntPPP18sNlsRodhOu62HViVtn+pCU4n5BYZHUXl+HqAUkHMQjko4j7U1NZShTl5zGg1xugwKmz03ul4+fsaHYbpuNt2YFXa/qUm5BbBhQuNjqJyYoeAn/6yEJNQDoq4D51+LCIiIiIiIm5LTa2IiIiIiIi4LZ2gIJYSHR3NunXrjA5DRAyiGiCiPBCxOjPWAB2pFREREREREbelplYsJSEhgVtuuYWEhASjQxERA6gGiCgPRKzOjDVATa1YSk5ODlu3biUnJ8foUETEAKoBIsoDEaszYw1QUysiIiIiIiJuSzeKMpGwXjEMmjup1LKCEzlk7ktm7+zl7PhoIc4ih0HRiRVFXXsxfd+4lxUPvM2eL38us75u01BGrP8/9sxaxooH33F9gCImk7XlZ3b/u3+pZXbfOvg0bkPIxTfQcOh92Dw09Yu42uly07dJW4L730jDIfdi8/AwJjgRE9DMZkLx36wkcekGsNnwCw0iamQ/uk+6mcDWTVj92PtGhyciIjWsft9RBHYbCk4nBekpHP35Uw7+92FyD+6g+T0fGB2eiGWVys1jhzj60yccnPYguQe2KTdFzoGaWhM6ti2efXNiS37e9ckiroydSpvrL2HTCzPJO5ppYHTGCg8PZ9KkSYSHhxsdiogYwCo1wL9FJ0IuHlPyc+iQu9l2TzvSlkyj8Zjn8AoMNTA6MZpV8qA2KpObg+/6MzdHP4tXUCMDoxOrMGMN0DW1FlCYk0fapj3Y7HbqNbd2sQwMDGTw4MEEBgYaHYqIGMCqNcDDtw512vQAp5O8lL1GhyMGs2oe1EYe/vWo07bXH7m5z+hwxCLMWAPU1FpEQGRxM5ubnmVwJMZKT0/nq6++Ij093ehQRMQAVq4Bp5pZz4AQgyMRo1k5D2obp9NJXsoeADzrNTA4GrEKM9YANbUm5OHng09wAD4h9QiKjqDHlLGEdGjJkV/jyIpPMTo8Q6WmpvLyyy+TmppqdCgiYgCr1ABH3kkKM9MoyDhCzv4tHHjvHnL2/Yp/6274Nm5tdHhiMKvkQW3019w8uX8zB965nZz435Wb4lJmrAGWuKY2LS2NF154ga+//pqkpCRCQ0O59tprmTx5MrfddhszZszgww8/ZOzYsUaHWi06PTySTg+PLLUs4fu1rHn8Q4MiEhERV0r+8lmSv3y21LKgHlcScdf/GRSRiEA5uWmzUa/LYJrfM824oERMwPRN7ebNmxk4cCCpqanUqVOH9u3bk5SUxGuvvcb+/fs5cOAAAOeff77BkVaf3TOWEj9vJXZPD4LaRtDhvivxaxBEYW5+yRi7tyfDFr3Evq9XsOXNuSXL+069B9/QIJaOfs6I0MWinE6n0SGImErIpbcSfOF1OB2F5CRsJWXOCxQcT8Xu7VcyJmtbLHueGVzmtc7CfJyOIrp8XeTKkEUs4VRuYrNh9/HHt3EbPOvpkgCRc2XqpjYtLY2hQ4eSmprKuHHjmDhxInXr1gVg8uTJTJw4EU9PT+x2O+edd57B0VafrP0pJMduASBp2W8cXreDwfMn0+uF21h+zxsAOPILib3/LQZ//QwHl24kfXsCEYO60fSyrswb8LCR4YuJnNqR4uHrXe56D38fAIr+ssNFRM6db3gU9TpdCkDgBYOo274vu57oQ8J7d9Hykc8BCIi5kM6zsku9Lv/oIXY+0pXQf9zr8phFrOCvuSki1cfU19Tef//9JCYm8vDDD/Piiy+WNLQAEyZMIDo6msLCQtq0aYOfn98Z3sm9HdkUx745sbS8+kJCL/jzeo1jW+LZ+s48LnzzPvzDg+n10h2sHT+NnFTzXDT+d/7+/vTo0QN/f3+jQ7GE7AOHAQhq07Tc9UGti5dn/TFOpKZZtQbUbduT4H5jSF8+k+xda8od4yjIY98LV1O3fV/CR453cYTiSlbNAxEpZsYaYNqmdvv27cyaNYuGDRvy7LPPlllvs9no0qULUPrU49jYWC677DLCw8Px8fGhadOmjBo1ih07drgs9prw++uzcRQW0XncdaWWb35zLo6CQoYveZmUVduIn7fSoAhdIyIigrfeeouIiAijQ7GEo1v2kZ10hBZX9sGvUf1S6+xenrS7ZTBOh4PExRsMilCsxso1IHzURLB7cGjGxHLXH3j3ThwFuUQ+8IlrAxOXs3IeiIg5a4Bpm9oZM2bgcDgYM2bMafdC+PgUn/r416Y2PT2d8847jzfffJPFixfz4osvsm3bNnr16sXBgwddEntNyNqfQvy8lTTudz4Ne7QrWe4scpC6bie+IYHsmbXMwAhdo6ioiOzsbIqKdK2YKziLHKx5/EO8Avy54qfX6DJhDG3GXErHh0YwbPFLhPWOYcvb35C595DRoYpFWLkG+IZHEXzhdWT9vpSsbbGl1h3+9k0yNiyg1ZPfYPcxz557KZ+V80BEzFkDTNvU/vjjjwD079//tGNONal/bWqHDx/O66+/zsiRI+nXrx+jR49mzpw5ZGRkMGfOnJoNuoZtfmMOjqIiOj82qmRZw25taf3PAez4aCHdn/nXaa99NIu4uDgGDBhAXFyc0aFYxsEfN7Fw+L9JWbmVqGv70XPKWM67Yxi5RzP5+fZX2fT850aHKBZi9RoQNnIC2O0c+vypkmVZm5dx8NPHafnYl/g0ijQuOHEZq+eBiNWZsQaY9kZRCQkJAERGRpa7Pi8vj7Vr1wJnv/NxgwbFD8P29KzaP1fXrl1JSanc82G9nHaepnulXpOyehufhI847fqMuCQ+bfpnQ+vp50PfN+5l05TP2fm/RQz++hm6TBjNuokfV+pzAdq0bkOBzVHp152rESNO/33Lc/hw8bWb33//PRs3bqzQa6666qpKx1VdqrId1EZHf9/Lz7e/anQYNcao7V+KVaYOVKUGgDF1wObtR6OplfuDI6DDxXSZd/q7ifs1a1fqrsZ5qfvZ9/K1NL35ZQI6nn4ncEW1adMaZ37OOb+PVI7Z50KjVCUHT+dsuVldlIPWZYa5MCwsjA0bqnZJmmmb2hMnTgCQm5tb7vrp06eTkZFBSEgITZo0KbO+qKgIh8NBYmIi48ePp3HjxowaNaqcdzq7lJQUkpKSKvUab5sHNKrSx1VYt0k3kZ14hJ2f/ADAigfeZvjSVziwcB0pq7dV6r0OJR8i3+n6UxhO/Z4rKicnp+R/K/rayv7uqpMrtgM5d0Zt/1KsMnWgKjUAjKkDdh//Gk1/R95J9j5/JYHdh9Owmu52fOjQIRx5J6vlvaTizD4XGqWmc7AmKAety6xzYUWZtqkNCwsjKyuLDRs20LVr11LrEhMTefLJJwHo2LFjua/v168fK1cW3zQpKiqKn376qeSIbVViqSwvpx1q8MBPk/6daDG8D/MueaRkWVZCKhufm0GfqXczb8AjFJ4of4dAeRqHNzbkSFWdOnUqNf5U4vr5+VX4teXt9HCVmt4OpHoYtf1LscrUgarUADCmDti8a/au/Omr5pAT/zu5SbtJXzGrzPqYt7fjHVq5m4g0btxYR4kMYPa50Cg1nYM1QTloXWaYC6vSM51iczqdNX8uhAHuvvtu/u///o9mzZqxdOlS2rRpA8CaNWu44YYbSEhIoKCggAcffJDXX3+9zOt37drF8ePHiY+P5+WXX+bIkSOsWLHCZXcJKziZy4xWY1zyWdVh9N7pePn7uvxz169fX6nxO3fu5MYbb+TTTz8lOjq6Qq/p1q1bVUKrFu62HViVUdu/FKtMHahKDQBj6kBOIVy40OUfe05ih4CfaXeX115mnwuNohwUd2LWubCiTHujqCeeeIL69euTmJhITEwMHTp0ICoqil69etGuXTv69u0LnP562rZt29KjRw+uu+46fvzxRzIzM3nppZdc+RWkBkRFRbFo0SKioqKMDkVEDKAaIKI8ELE6M9YA0za1ERERxMbGMmTIEHx9fUlISCA0NJRp06Yxb948du7cCZz9JlEAQUFBREVFsWfPnpoOW2qYp6cn9evXr/JNv0TEvakGiCgPRKzOjDXAtE0tQExMDN999x1ZWVlkZmayevVqbr31VtLT00lOTsbT05P27duf9X0OHz7Mrl27aNWqlQuilpp08OBBHnnkEbd+5rCIVJ1qgIjyQMTqzFgDTN3Uns6WLVsAiI6OxsfHp9S6MWPG8J///IdvvvmGn3/+mQ8//JCLL74YT09PHnroISPClWqUnZ1NbGws2dnZRociIgZQDRBRHohYnRlrgHmOOVfCqaa2vFOPe/bsyaeffsobb7xBbm4uzZo1o3///owfP57mzZu7OlQRERERERE5AzW1f3Pvvfdy773V87w+V/Gu58+gOZNKfvbw8yGgeSO+6HAr+cf/3AMT1uc8uowfjVddX5wOJwcWrefXF2YCEH5hB7o9dWPJWN8GgeQcOc63A8e57otIpQS0COPCN+7DJziA/MwTrHjwHTJ2lz2NpG7TUPq+cS/B50WSGZ/CgkGPl1p39eq3Ob7zQMmyZWNfISshFb9G9ek79R7qNg2lKL+QjD1JrH78g1Lb1Olc9sVE/EKDwOGg4EQuayd8xLFt+8sd2/qfA+hw71Vgt5G8YgtrnvgQZ1Hx43Fi7hpO1LUXY7PbyNh7iJUPvkN+ZsWfv9fntbtp/c8BTG81hsKTZR9RFTm8Nx3uvQq7lwc4neyavpSd//0egNAubej1wm0A2Lw8ObxuB2v//V8c+YUV/nwRV8rY9AOHpv8bZ2E+dh9/Iu5+H/8WZee5tCUfkTLnBZxOB/U6XkLEne9i8/AkZ/8WDrx/DwUZh7F5elG3bS+a3f4Wdi+fcj5NRP7OkZ/LvleuIzdxO3Yff7yCwoi46z18GkWWGpe5+SeSPn0SR04W2O0Edb+CxmMmY7PZyEvdz9Y7o/Brfl7J+FaPz8EnXJfAiZyJpZva0z2j1t3kZ55k/mWPlfwcc+dwwnq1L9N85Gec4Je7Xif7wGE8fLwY+OVTtLiqL/FfryA5dkup97jk0ydJWbnVZd9BKq/3S3ewe/oS9nz5M83/0ZM+r97FwmETyozLz85h04sz8Q7wp9Njo8quzzxR6nd/irPIwe+vz+bwuuKbqnWdeANdJoxm9WPvnzW2X25/taT5jBjUjT6v313uDpK6zRrSedx1zL/sMXLTMhjwyeO0vv4Sdn+2hPCLOhI1sh/fDR1P4Ylczn94JJ2fuJ6146ed9fMBml7WhbM9sexk8lGWjn6OnCPH8QrwZ9iiFzm2ZR+H1+/i2Pb9fDv4CZyFRWCz0X/ao7S9YSA7PnKz5zuIJRRmpxP/2hjaPh+LX7N2ZG39hfjXRhPzVuk6npcaz6EZE2k39Vc8Axuy97krSFvyEaGD7sDm7UuzO97GP7IjzqIi4l+9ntRvXiV85HiDvpWI+wkdeDv1ugzGZrNx+Lu3SXj3dtpMWlxqjGfd+rR8dCY+YS1x5Oey+6lLSV8+k+B+1wPgUSeI9lN/MyB6EfdlyWtqV61ahdPp5PLLLzc6lBoRdV1/4mb+WGb5sa3xZB84DEBRXgHHtu4noHmjMuP8GtUnvO957J39S43H6mqhoaE88MADhIaGGh3KOfENqUdIh5bsnbMcgITv1hAQ0ZC6Tct+r/zj2Rxet5PCk3mV+ozctIyShhbgyKY4AiLKbi/l+evRVK8Af5yO8pvL5kN7kvD9OnLTMgDY9eliWl5Z/Lit4PaRpK7dQeGJ4iOsB5dupNWIiyr0+T7169LpkWtZ/5//nXHc4fW7yDlyHICCrJNk7DlE3YiGABTl5Bc3tICHtycevt44HY4Kfb7UXmapAX+Xl7wXr8CG+DVrB0DAef3IP5zAyb2bSo1LXzmboJ5X4RXUCJvNRuigOzkWW3zGjm/j1vhHFu/stXl44N+6G3kp+1z7RcQlzJoHRrN7+xLYdQg2mw2AOm16kl9ODvm37IxPWMuS1/i36EReqnJNXMeMNcCSTa2ZhXZpg2/9uiQu2XjGcX6hQUQO7cnBHzeVWRd17cUc/OlXco9m1lSYhgkJCWH06NGEhIQYHco5qdOkASdTj5WcpguQnZRGnaYNKv1eXnX9GPr9Cwxb/BLnPzQCm71sWbDZ7UT/axCJSzZU+H37vnkfIze8xwWP/5MV979V7pi6TRpw4uCRP7/DwSPUaVL8HY7+vpfGF52Pb4NAAFpecxHeAf54B9U962f3nHIbv73yJQVZFT9VObBNU0K7tCF5xZ9Htuo2DWX4kpe5btt/Kcg6ye7pSyv8flI7maUG/J1v49YUZh4he9caANJXf40jN5u8w/tLjctPO4B3wz/vD+HdMJL8Iwf4u6LcE6QtmUZQ92E1GrcYw6x5UNscXvAmgd3OnEMF6Smkr5pNYJchJcscOZnseKQb2x+6gENfPIOzqKimQxWLMWMNUFNrMq3/OYA9X/1Sqtn5O6+6flzy6RNseXcex7bEl1kfNao/cZ+XPdJrBpmZmSxdupTMTPdv2P9+Zu2pPcOVcfJwOl9dcAcLBj/BomufoVGPdsTcWXYC7vnCWPIzT7Djo+8r/N4r7n+Lr7reyaYXZ9L1L9dr/91fTxH+6zdIWb2Nbe9/y6XTx/OPBVPISU0HwFFw5mtamw/tRVFBIQeXnnnHzl/5hwdzycePs/rxD0o+B4qb7PmXPcasjrfh4eNF8yE9KvyeUjuZqQb8lUedQFo+MYekT59gx8NdyN76M77N2mPz8Cpn9F8zrexZFM7CAuJfHkW9TgMJ6nFFjcUsxjFrHtQmyV9NIS85jiY3PHfaMUUnM9kzeRhhV4/Dv9UFAHgFh9Pho4O0e3U9bZ5ZSvb2WFLnveqqsMUizFgD1NSaiKefD5HDe7Pni59OP6aOL5d9/m8OLN7A9vcXlFnfqGc7PP18OPTz7zUZqmEOHTrE+PHjOXTokNGhnJMTSWnUCQ/G5vFnCtdpHMKJg2mVeh9HfmHJEfn849nEffETjXq0KzWmx+RbqNO4Ab/c8XrZTroC9n71C2G9Y/CpX/YIa3ZSGnWbNfzzOzQN5UTSn99h16eLWTDocb4bOp7UNds5kZRWcjry6YT3iSG8z3mMWPcuI9a9C8CVv7xOUHREueP9GtXn8i+f5veps0lYsLrcMYUnc4n/ZiUtr77wrN9Xajez1IDyBMRcRNvnfqbdaxtpcvNLFBw7hG+z0vns3SCC/L8cvc0/nIB36J+54SwsYN9L1+JVP5xmt73hqtDFxcycB7VBytevcHz1XKKe+h67j3+5Y4pOZhH3n0EEdR9OoyseLllu9/LBK6h4XvQMCKbBpbeQvT3WJXGLdZixBqipNZHIYb1I35FAxp7yN1BP/+KGNunn39j8+uxyx7S+bgB7vlymawdrudyjmRzbup9W1xRfY9r8Hz3JTjxC9l9O5a0I35B62Dw9ALB7e9J8SE+Obv3z6H33Z28hoEU4P93yUpkjpH3fvI+Iwd3LvKdXXT/8GtUv+TlicHfy0rPJSy971+SE79bQfHD3klOM2944kPh5K0vW+zUMAsDDz5tO465jy7vzStYN/PJpGnSKKvOea56cxldd7mB297uZ3f1uAL7p91CpOzz/9f0v/+pptrz9DXu/Kn0NeUDzRn/+23h5EjGkB8d2JJR5D5HaouBYcsn/T571LAEdB+AbXjpH6ve+huNrvqbgeCpOp5MjP7xHcN/rAHAWFbLvlevwCAgm4p4PqnT2h4jVpc57jfTYmbR+ZgmedYPKHVOUk03cpEHUu+BywkdNLLWu4PhhnIUFADgK8khfPRf/Fp1rOmwRt2fJux+bVevrLyHu89JHaXu/cieJizeQuHgD7W8bQmjnKLz8fWj+RzOyf8FqNr8xFyg+itv8Hz2Zd8kjLo9dKm/VuPfpO/UeOtx/NQXZOaWuW/3r793u7ck1q9/Bw8cTrwB/Rm58n71zfmHTlM9p2KMdnR8bhbPIgc3Tg5QVW9j8xhwAGnZrS/uxQzged5Ch3z0PQFbiYZbd8jIAIR1blnsnYK96/vSf9iievt44HU5yj2by443Plxtb9oHD/PrKlwyZPxmb3U7yii3s/sup7wO/mAh2Ox5enuyd/UvJ43ZsdjvBMc05kXy00v9ul04fz68vz+Lo73vpPO466jRpQPuxQ2g/tvh6pu3TFrJn1jLCesfQ/vahpf9tTrMzSKQ2SJoxkewdK6CokDrRvWh+70cA7H9rLEHdhxPUYzg+YS0Jv34Sux7vg9PpIKDDABpcdisAx2JncXz1XPwiO7LjoeI/outG9yHizncM+04i7iQ/7SAH//sI3mEt2f3v/gDYPH1o98raUnl4+Ns3OBG3DkfeCY6v+RqA+r1HEn7tBLJ3rODQ509hs3vgLCokoOMAwq4t+2QDESnN5jzbMy/EEAUnc5nRaozRYVTY6L3T8fL3dfnnrl+/vlLjd+7cyY033sinn35KdHR0hV7TrVu3qoRWLWrrduAdVJeL33uIxdc9a8jnB3doQfRNl7Pq0fcM+fy/M2r7l2KVqQNVqQFgTB3IKYQL3ewJUrFDwE+7y13O7HOhUZSD4k7MOhdWlE4/Fkvx8fGhbdu2+Pj4GB2KW8s/nm1YQwtwbEt8rWloxb2oBogoD0Sszow1QPtyxFJatGjBZ599ZnQYImIQ1QAR5YGI1ZmxBuhIrYiIiIiIiLgtNbViKbt27aJPnz7s2rXL6FBExACqASLKAxGrM2MNUFMrluJ0OikoKED3RxOxJtUAEeWBiNWZsQbomtpaytPPh9F7pxsdRoV5+pnnQvPaxN22A6vS9i81wdej+E6m7sTXw+gIRKqPclDEfaipraVsNpseESLaDkQszGbTozlEjKQcFHEfOv1YRERERERE3Jb2P4mlREZGMnPmTJo0aWJ0KCJiANUAEeWBiNWZsQaoqRVL8fX1pVWrVkaHISIGUQ0QUR6IWJ0Za4BOPxZLSU5OZvLkySQnJxsdiogYQDVARHkgYnVmrAFqasVSMjIymD9/PhkZGUaHIiIGUA0QUR6IWJ0Za4CaWhEREREREXFbampFRERERETEbampFREREREREbelplYsJTg4mJtuuong4GCjQxERA6gGiCgPRKzOjDXA5nQ6nUYHISIiIiIiIlIVOlIrIiIiIiIibktNrYiIiIiIiLgtNbUiIiIiIiLittTUioiIiIiIiNtSUysiIiIiIiJuS02tiIiIiIiIuC01tSIiIiIiIuK21NSKiIiIiIiI21JTKyIiIiIiIm5LTa2IiIiIiIi4LTW1IiIiIiIi4rbU1IqIiIiIiIjbUlMrIiIiIiIibktNrYiIiIiIiLgtNbUiIiIiIiLitv4fK6N2BhbL28kAAAAASUVORK5CYII="
      },
-     "execution_count": 6,
+     "execution_count": 21,
      "metadata": {},
      "output_type": "execute_result"
     }
    ],
-   "execution_count": 6
+   "execution_count": 21
   },
   {
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2025-06-25T06:49:48.391855Z",
-     "start_time": "2025-06-25T06:49:48.387742Z"
+     "end_time": "2025-06-26T01:24:15.451836Z",
+     "start_time": "2025-06-26T01:24:15.447333Z"
     }
    },
    "cell_type": "code",
@@ -356,44 +354,54 @@
     "    def __init__(self):\n",
     "        super(RandomQCCNN, self).__init__()\n",
     "        self.conv = nn.Sequential(\n",
-    "            RandomQuantumConvolutionalLayer(nqubit=4, num_circuits=3, seed=1024),   # num_circuits=3代表我们在quanv1层只用了3个量子卷积核\n",
+    "            RandomQuantumConvolutionalLayer(nqubit=4, num_circuits=3),   # num_circuits=3代表我们在quanv1层只用了3个量子卷积核\n",
     "            nn.BatchNorm2d(3),  # 添加批量归一化\n",
     "            nn.ReLU(),\n",
     "            nn.MaxPool2d(kernel_size=2, stride=1),\n",
+    "            # 添加形状检查层以确保尺寸正确\n",
     "            nn.Conv2d(3, 6, kernel_size=2, stride=1),\n",
     "            nn.BatchNorm2d(6),  # 添加批量归一化\n",
     "            nn.ReLU(),\n",
-    "            nn.MaxPool2d(kernel_size=2, stride=1)\n",
+    "            # 添加自适应池化层确保固定输出尺寸\n",
+    "            nn.AdaptiveMaxPool2d((9, 9))  # 确保输出固定尺寸\n",
     "        )\n",
     "        self.fc = nn.Sequential(\n",
-    "            nn.Linear(6 * 6 * 6, 1024),\n",
-    "            nn.BatchNorm1d(1024),  # 添加批量归一化\n",
-    "            nn.Dropout(0.5),  # 增加dropout比例\n",
+    "            # 根据自适应池化后的固定尺寸计算输入维度\n",
+    "            nn.Linear(6 * 9 * 9, 1024),  # 确保与自适应池化输出匹配\n",
+    "            nn.BatchNorm1d(1024),\n",
+    "            nn.Dropout(0.5),\n",
     "            nn.ReLU(),\n",
-    "            nn.Linear(1024, 10)\n",
+    "            nn.Linear(1024, 512),\n",
+    "            nn.BatchNorm1d(512),\n",
+    "            nn.ReLU(),\n",
+    "            nn.Linear(512, 10)\n",
     "        )\n",
     "\n",
     "    def forward(self, x):\n",
+    "        # 添加详细的形状检查输出\n",
+    "        print(f\"Input shape: {x.shape}\")\n",
     "        x = self.conv(x)\n",
+    "        print(f\"After conv layers: {x.shape}\")  # 添加中间形状检查\n",
     "        x = x.reshape(x.size(0), -1)\n",
+    "        print(f\"After flatten: {x.shape}\")  # 添加展平后形状检查\n",
     "        x = self.fc(x)\n",
     "        return x"
    ],
-   "id": "82be55cb39abc20a",
+   "id": "37c556ec625ae040",
    "outputs": [],
-   "execution_count": 7
+   "execution_count": 22
   },
   {
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2025-06-25T06:54:06.782447Z",
-     "start_time": "2025-06-25T06:49:48.398218Z"
+     "end_time": "2025-06-26T02:23:23.162303Z",
+     "start_time": "2025-06-26T01:24:15.471352Z"
     }
    },
    "cell_type": "code",
    "source": [
     "# 修改RandomQCCNN模型的训练参数\n",
-    "num_epochs = 300\n",
+    "num_epochs = 100\n",
     "device = torch.device(\"cuda\" if torch.cuda.is_available() else \"cpu\")\n",
     "print(device)\n",
     "seed_torch(42)   # 使用相同的随机种子值\n",
@@ -405,7 +413,7 @@
     "torch.save(optim_model.state_dict(), './data/notebook2/random_qccnn_weights.pt')   # 保存训练好的模型参数，用于后续的推理或测试\n",
     "pd.DataFrame(metrics).to_csv('./data/notebook2/random_qccnn_metrics.csv', index='None')   # 保存模型训练过程，用于后续图标展示"
    ],
-   "id": "2087a1b2f259ad1",
+   "id": "15cd5b8c97c677d0",
    "outputs": [
     {
      "name": "stdout",
@@ -418,14 +426,220442 @@
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "Train loss: 0.556  Valid Acc: 0.760:  11%|█         | 33/300 [04:18<34:49,  7.83s/it]"
+      "  0%|          | 0/100 [00:00<?, ?it/s]"
      ]
     },
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Early stopping at epoch 34 due to no improvement in validation accuracy.\n"
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 1.077  Valid Acc: 0.709:   1%|          | 1/100 [00:45<1:15:29, 45.75s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.747  Valid Acc: 0.727:   2%|▏         | 2/100 [01:31<1:14:33, 45.65s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.668  Valid Acc: 0.754:   2%|▏         | 2/100 [02:16<1:14:33, 45.65s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.668  Valid Acc: 0.754:   3%|▎         | 3/100 [02:16<1:13:32, 45.49s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.613  Valid Acc: 0.768:   4%|▍         | 4/100 [03:01<1:12:32, 45.34s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.591  Valid Acc: 0.781:   5%|▌         | 5/100 [03:45<1:10:51, 44.75s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.569  Valid Acc: 0.779:   6%|▌         | 6/100 [04:30<1:10:27, 44.97s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.549  Valid Acc: 0.790:   7%|▋         | 7/100 [05:14<1:09:17, 44.70s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.531  Valid Acc: 0.797:   8%|▊         | 8/100 [05:59<1:08:40, 44.79s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.523  Valid Acc: 0.794:   9%|▉         | 9/100 [06:45<1:08:19, 45.05s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.507  Valid Acc: 0.799:  10%|█         | 10/100 [07:30<1:07:23, 44.93s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.499  Valid Acc: 0.804:  10%|█         | 10/100 [08:14<1:07:23, 44.93s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.499  Valid Acc: 0.804:  11%|█         | 11/100 [08:14<1:06:18, 44.70s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.488  Valid Acc: 0.800:  11%|█         | 11/100 [08:58<1:06:18, 44.70s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.488  Valid Acc: 0.800:  12%|█▏        | 12/100 [08:58<1:05:21, 44.56s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.472  Valid Acc: 0.806:  13%|█▎        | 13/100 [09:41<1:03:40, 43.91s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.475  Valid Acc: 0.805:  14%|█▍        | 14/100 [10:26<1:03:37, 44.39s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.462  Valid Acc: 0.810:  15%|█▌        | 15/100 [11:10<1:02:46, 44.31s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.454  Valid Acc: 0.817:  15%|█▌        | 15/100 [11:56<1:02:46, 44.31s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.454  Valid Acc: 0.817:  16%|█▌        | 16/100 [11:56<1:02:34, 44.69s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.450  Valid Acc: 0.809:  16%|█▌        | 16/100 [12:40<1:02:34, 44.69s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.450  Valid Acc: 0.809:  17%|█▋        | 17/100 [12:40<1:01:29, 44.45s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.445  Valid Acc: 0.813:  18%|█▊        | 18/100 [13:25<1:01:15, 44.83s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.437  Valid Acc: 0.819:  19%|█▉        | 19/100 [14:11<1:00:42, 44.97s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.428  Valid Acc: 0.823:  20%|██        | 20/100 [14:57<1:00:33, 45.42s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.427  Valid Acc: 0.826:  21%|██        | 21/100 [15:43<59:49, 45.43s/it]  "
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.424  Valid Acc: 0.824:  22%|██▏       | 22/100 [16:28<59:07, 45.48s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.419  Valid Acc: 0.824:  22%|██▏       | 22/100 [17:12<59:07, 45.48s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.419  Valid Acc: 0.824:  23%|██▎       | 23/100 [17:12<57:45, 45.01s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.416  Valid Acc: 0.822:  24%|██▍       | 24/100 [17:56<56:38, 44.72s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.413  Valid Acc: 0.823:  25%|██▌       | 25/100 [18:42<56:29, 45.20s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.406  Valid Acc: 0.823:  26%|██▌       | 26/100 [19:29<56:21, 45.69s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.406  Valid Acc: 0.827:  27%|██▋       | 27/100 [20:17<56:12, 46.20s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.403  Valid Acc: 0.822:  28%|██▊       | 28/100 [21:03<55:38, 46.37s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.392  Valid Acc: 0.825:  29%|██▉       | 29/100 [21:49<54:27, 46.02s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.391  Valid Acc: 0.821:  30%|███       | 30/100 [22:36<54:10, 46.44s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.388  Valid Acc: 0.820:  31%|███       | 31/100 [23:20<52:35, 45.74s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.385  Valid Acc: 0.828:  32%|███▏      | 32/100 [24:06<51:43, 45.64s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.384  Valid Acc: 0.828:  33%|███▎      | 33/100 [24:53<51:24, 46.04s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.380  Valid Acc: 0.828:  34%|███▍      | 34/100 [25:39<50:41, 46.08s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.377  Valid Acc: 0.829:  35%|███▌      | 35/100 [26:23<49:27, 45.66s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.374  Valid Acc: 0.828:  36%|███▌      | 36/100 [27:07<48:11, 45.17s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.371  Valid Acc: 0.833:  36%|███▌      | 36/100 [27:54<48:11, 45.17s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.371  Valid Acc: 0.833:  37%|███▋      | 37/100 [27:54<47:43, 45.45s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.366  Valid Acc: 0.832:  38%|███▊      | 38/100 [28:39<46:58, 45.46s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.363  Valid Acc: 0.823:  39%|███▉      | 39/100 [29:25<46:14, 45.49s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.363  Valid Acc: 0.833:  40%|████      | 40/100 [30:10<45:19, 45.33s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.359  Valid Acc: 0.834:  41%|████      | 41/100 [30:55<44:43, 45.49s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.355  Valid Acc: 0.834:  42%|████▏     | 42/100 [31:41<44:07, 45.65s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.357  Valid Acc: 0.833:  43%|████▎     | 43/100 [32:27<43:17, 45.58s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.356  Valid Acc: 0.836:  44%|████▍     | 44/100 [33:12<42:25, 45.46s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.351  Valid Acc: 0.830:  45%|████▌     | 45/100 [33:58<41:41, 45.49s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.347  Valid Acc: 0.832:  46%|████▌     | 46/100 [34:44<41:13, 45.80s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.346  Valid Acc: 0.830:  46%|████▌     | 46/100 [35:28<41:13, 45.80s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.346  Valid Acc: 0.830:  47%|████▋     | 47/100 [35:28<39:52, 45.14s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.346  Valid Acc: 0.833:  48%|████▊     | 48/100 [36:12<38:56, 44.94s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.342  Valid Acc: 0.832:  49%|████▉     | 49/100 [36:59<38:38, 45.47s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.338  Valid Acc: 0.831:  50%|█████     | 50/100 [37:46<38:22, 46.06s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.314  Valid Acc: 0.841:  50%|█████     | 50/100 [38:33<38:22, 46.06s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.314  Valid Acc: 0.841:  51%|█████     | 51/100 [38:33<37:42, 46.18s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.311  Valid Acc: 0.839:  52%|█████▏    | 52/100 [39:18<36:49, 46.03s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.305  Valid Acc: 0.842:  53%|█████▎    | 53/100 [40:05<36:05, 46.07s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.304  Valid Acc: 0.842:  53%|█████▎    | 53/100 [40:50<36:05, 46.07s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.304  Valid Acc: 0.842:  54%|█████▍    | 54/100 [40:50<35:05, 45.76s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.298  Valid Acc: 0.839:  55%|█████▌    | 55/100 [41:35<34:07, 45.49s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.300  Valid Acc: 0.840:  55%|█████▌    | 55/100 [42:19<34:07, 45.49s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.300  Valid Acc: 0.840:  56%|█████▌    | 56/100 [42:19<33:08, 45.19s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.297  Valid Acc: 0.841:  57%|█████▋    | 57/100 [43:04<32:22, 45.16s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.294  Valid Acc: 0.843:  58%|█████▊    | 58/100 [43:49<31:30, 45.01s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.293  Valid Acc: 0.839:  58%|█████▊    | 58/100 [44:34<31:30, 45.01s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.293  Valid Acc: 0.839:  59%|█████▉    | 59/100 [44:34<30:43, 44.96s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.288  Valid Acc: 0.843:  60%|██████    | 60/100 [45:18<29:52, 44.82s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.288  Valid Acc: 0.842:  61%|██████    | 61/100 [46:04<29:17, 45.05s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.284  Valid Acc: 0.843:  62%|██████▏   | 62/100 [46:51<28:56, 45.69s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.287  Valid Acc: 0.844:  63%|██████▎   | 63/100 [47:36<28:09, 45.66s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.286  Valid Acc: 0.838:  64%|██████▍   | 64/100 [48:22<27:26, 45.75s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.282  Valid Acc: 0.838:  65%|██████▌   | 65/100 [49:08<26:36, 45.61s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.279  Valid Acc: 0.841:  66%|██████▌   | 66/100 [49:53<25:49, 45.59s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.277  Valid Acc: 0.844:  67%|██████▋   | 67/100 [50:40<25:14, 45.89s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.281  Valid Acc: 0.846:  68%|██████▊   | 68/100 [51:26<24:26, 45.83s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.274  Valid Acc: 0.840:  69%|██████▉   | 69/100 [52:12<23:42, 45.89s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.280  Valid Acc: 0.840:  70%|███████   | 70/100 [52:58<23:00, 46.01s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.273  Valid Acc: 0.841:  71%|███████   | 71/100 [53:42<21:58, 45.46s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.272  Valid Acc: 0.841:  72%|███████▏  | 72/100 [54:28<21:13, 45.48s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.268  Valid Acc: 0.842:  73%|███████▎  | 73/100 [55:13<20:29, 45.52s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.270  Valid Acc: 0.842:  74%|███████▍  | 74/100 [56:01<19:57, 46.07s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.257  Valid Acc: 0.845:  75%|███████▌  | 75/100 [56:47<19:15, 46.21s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.252  Valid Acc: 0.845:  76%|███████▌  | 76/100 [57:33<18:27, 46.16s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.250  Valid Acc: 0.843:  77%|███████▋  | 77/100 [58:22<17:57, 46.85s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.250  Valid Acc: 0.843:  77%|███████▋  | 77/100 [59:07<17:39, 46.07s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Early stopping at epoch 78 due to no improvement in validation accuracy.\n"
      ]
     },
     {
@@ -436,13 +220872,13 @@
      ]
     }
    ],
-   "execution_count": 8
+   "execution_count": 23
   },
   {
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2025-06-25T06:54:11.647094Z",
-     "start_time": "2025-06-25T06:54:06.811569Z"
+     "end_time": "2025-06-26T02:23:28.265197Z",
+     "start_time": "2025-06-26T02:23:23.249369Z"
     }
    },
    "cell_type": "code",
@@ -454,23 +220890,491 @@
     "\n",
     "test_acc = test_model(random_qccnn_model, test_loader, device)"
    ],
-   "id": "b19364a05a067c50",
+   "id": "d17bfde5bcabc2b3",
    "outputs": [
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Test Acc: 0.797\n"
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "After conv layers: torch.Size([64, 6, 9, 9])\n",
+      "After flatten: torch.Size([64, 486])\n",
+      "Test Acc: 0.838\n"
      ]
     }
    ],
-   "execution_count": 9
+   "execution_count": 24
   },
   {
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2025-06-25T06:54:11.888135Z",
-     "start_time": "2025-06-25T06:54:11.759275Z"
+     "end_time": "2025-06-26T02:23:28.601077Z",
+     "start_time": "2025-06-26T02:23:28.458769Z"
     }
    },
    "cell_type": "code",
@@ -503,26 +221407,26 @@
     "\n",
     "plt.show()"
    ],
-   "id": "752fc968a05aed09",
+   "id": "7cd464ba7b2e535e",
    "outputs": [
     {
      "data": {
       "text/plain": [
        "<Figure size 1200x500 with 2 Axes>"
       ],
-      "image/png": "iVBORw0KGgoAAAANSUhEUgAAA+kAAAHUCAYAAABGRmklAAAAOnRFWHRTb2Z0d2FyZQBNYXRwbG90bGliIHZlcnNpb24zLjEwLjAsIGh0dHBzOi8vbWF0cGxvdGxpYi5vcmcvlHJYcgAAAAlwSFlzAAAPYQAAD2EBqD+naQAA3+ZJREFUeJzs3Xd4FNXXwPHvphfSCQklBaQESAih9yIdQaoUpUkRRFRA3x9SBKwoiIAgIFIiKlVpShEQqQklhFCkQ0IoCS0kIb3N+8eQ1ZhCyiabcj7Ps09mZ+/MnE1Yds/ee8/VKIqiIIQQQgghhBBCCL0z0HcAQgghhBBCCCGEUEmSLoQQQgghhBBCFBOSpAshhBBCCCGEEMWEJOlCCCGEEEIIIUQxIUm6EEIIIYQQQghRTEiSLoQQQgghhBBCFBOSpAshhBBCCCGEEMWEJOlCCCGEEEIIIUQxIUm6EEIIIYQQQghRTEiSLso0jUaTq9vBgwcLdJ3Zs2ej0WjydezBgwd1EkNBrv3LL78U+bXz49y5c7z++utUrVoVMzMzypUrR4MGDZg7dy4RERH6Dk8IIUQuyHtz7u3YsQONRoODgwOJiYl6jaUkio6O5rPPPqNRo0ZYW1tjamqKu7s7I0eOJDAwUN/hiTLMSN8BCKFP/v7+Ge5/8skn/PXXXxw4cCDD/jp16hToOqNHj6Zr1675OrZBgwb4+/sXOIbS7vvvv2f8+PHUqlWL//u//6NOnTokJycTEBDA8uXL8ff3Z+vWrfoOUwghxHPIe3PurVq1CoCIiAi2bdvGwIED9RpPSXLjxg06d+7MgwcPGDduHB999BHlypUjJCSETZs20bBhQyIjI7GxsdF3qKIM0iiKoug7CCGKixEjRvDLL78QExOTY7u4uDgsLCyKKCr9OXjwIO3bt2fz5s30799f3+Fky9/fn9atW9OpUye2bduGqalphseTkpLYs2cPL7/8coGvFR8fj5mZWb57X4QQQuSNvDdnLTw8HBcXF9q0aYOfnx+tW7dm7969+g4rS8Xtb5OamoqPjw+3bt3i2LFjeHp6Zmqze/du2rZtW+C4FUUhISEBc3PzAp1HlC0y3F2I52jXrh2enp4cPnyYFi1aYGFhwciRIwHYuHEjnTt3pmLFipibm1O7dm0++OADYmNjM5wjqyF17u7u9OjRgz179tCgQQPMzc3x8PBg9erVGdplNaRuxIgRlCtXjuvXr9O9e3fKlSuHi4sL7733Xqbhbnfu3KF///5YWVlha2vLa6+9xqlTp9BoNPj6+urkd3ThwgV69eqFnZ0dZmZm1K9fnx9++CFDm7S0ND799FNq1aqFubk5tra21KtXj0WLFmnbPHz4kDfeeAMXFxdMTU1xdHSkZcuW7N+/P8frf/7552g0GlasWJEpQQcwMTHJkKBrNBpmz56dqZ27uzsjRozQ3vf19UWj0bB3715GjhyJo6MjFhYWbNy4EY1Gw59//pnpHMuWLUOj0XDu3DntvoCAAF5++WXs7e0xMzPDx8eHTZs25fichBBCZE/em+GHH34gJSWFSZMm0bdvX/78809u3bqVqV1kZCTvvfce1apVw9TUlAoVKtC9e3cuX76sbZOYmMjHH39M7dq1MTMzw8HBgfbt2+Pn5wdASEhItrH99z01/fcaGBhI//79sbOz44UXXgDU98NBgwbh7u6Oubk57u7uDB48OMu47969q/1MYGJiQqVKlejfvz/3798nJiYGW1tbxo4dm+m4kJAQDA0NmTdvXra/u23btnH+/HmmTp2aZYIO0K1bN22CPmLECNzd3TO1yerfkEajYcKECSxfvpzatWtjamrKypUrqVChAkOHDs10jsjISMzNzZk8ebJ2X3R0NO+//z5Vq1bFxMSEypUrM3HixEz/hkXpJcPdhciFsLAwhgwZwv/+9z8+//xzDAzU77euXbtG9+7dmThxIpaWlly+fJkvv/ySkydPZhqWl5WzZ8/y3nvv8cEHH+Dk5MTKlSsZNWoU1atXp02bNjkem5yczMsvv8yoUaN47733OHz4MJ988gk2NjbMnDkTgNjYWNq3b09ERARffvkl1atXZ8+ePTodDnflyhVatGhBhQoV+Oabb3BwcOCnn35ixIgR3L9/n//9738AzJ07l9mzZzNjxgzatGlDcnIyly9fJjIyUnuuoUOHEhgYyGeffUbNmjWJjIwkMDCQx48fZ3v91NRUDhw4QMOGDXFxcdHZ8/q3kSNH8tJLL/Hjjz8SGxtLjx49qFChAmvWrKFDhw4Z2vr6+tKgQQPq1asHwF9//UXXrl1p2rQpy5cvx8bGhg0bNjBw4EDi4uIyfCkghBAi98r6e/Pq1aupWLEi3bp1w9zcnHXr1uHr68usWbO0bZ4+fUqrVq0ICQlhypQpNG3alJiYGA4fPkxYWBgeHh6kpKTQrVs3jhw5wsSJE3nxxRdJSUnh+PHjhIaG0qJFizzFla5v374MGjSIcePGaZPLkJAQatWqxaBBg7C3tycsLIxly5bRuHFjLl68SPny5QE1QW/cuDHJyclMmzaNevXq8fjxY/744w+ePHmCk5MTI0eOZMWKFcydOzfDkPSlS5diYmKi/dImK+kjDnr37p2v5/Y827Zt48iRI8ycORNnZ2cqVKhAcHAwy5cv59tvv8Xa2lrbdv369SQkJPD6668D6qiDtm3bcufOHe1z//vvv5k5cybnz59n//79MpqvLFCEEFrDhw9XLC0tM+xr27atAih//vlnjsempaUpycnJyqFDhxRAOXv2rPaxWbNmKf99ubm5uSlmZmbKrVu3tPvi4+MVe3t7ZezYsdp9f/31lwIof/31V4Y4AWXTpk0Zztm9e3elVq1a2vvffvutAii7d+/O0G7s2LEKoKxZsybH55R+7c2bN2fbZtCgQYqpqakSGhqaYX+3bt0UCwsLJTIyUlEURenRo4dSv379HK9Xrlw5ZeLEiTm2+a/w8HAFUAYNGpTrYwBl1qxZmfa7ubkpw4cP195fs2aNAijDhg3L1Hby5MmKubm59vkpiqJcvHhRAZTFixdr93l4eCg+Pj5KcnJyhuN79OihVKxYUUlNTc113EIIURbJe3Nmhw8fVgDlgw8+0D7PqlWrKm5ubkpaWpq23ccff6wAyr59+7I919q1axVA+f7777NtExwcnG1s/31PTf+9zpw587nPIyUlRYmJiVEsLS2VRYsWafePHDlSMTY2Vi5evJjtsTdu3FAMDAyUBQsWaPfFx8crDg4Oyuuvv57jdbt27aoASkJCwnNjVBT1b+vm5pZpf1b/hgDFxsZGiYiIyLD/3LlzCqCsWLEiw/4mTZooDRs21N6fM2eOYmBgoJw6dSpDu19++UUBlF27duUqZlGyyXB3IXLBzs6OF198MdP+mzdv8uqrr+Ls7IyhoSHGxsa0bdsWgEuXLj33vPXr18fV1VV738zMjJo1a2Y57Ou/NBoNPXv2zLCvXr16GY49dOgQVlZWmQrjDB48+Lnnz60DBw7QoUOHTL3YI0aMIC4uTlsAqEmTJpw9e5bx48fzxx9/EB0dnelcTZo0wdfXl08//ZTjx4+TnJysszgLol+/fpn2jRw5kvj4eDZu3Kjdt2bNGkxNTXn11VcBuH79OpcvX+a1114DICUlRXvr3r07YWFhXLlypWiehBBClDJl+b05vWBcem+xRqNhxIgR3Lp1K8NUrN27d1OzZk06duyY7bl2796NmZlZjj3P+ZHVe2dMTAxTpkyhevXqGBkZYWRkRLly5YiNjc3wt9m9ezft27endu3a2Z6/WrVq9OjRg6VLl6I8K7G1bt06Hj9+zIQJE3T6XPLqxRdfxM7OLsM+Ly8vGjZsyJo1a7T7Ll26xMmTJzP87n///Xc8PT2pX79+hs8NXbp0KRYrCoiiIUm6ELlQsWLFTPtiYmJo3bo1J06c4NNPP+XgwYOcOnWKLVu2AGqBsedxcHDItM/U1DRXx1pYWGBmZpbp2ISEBO39x48f4+TklOnYrPbl1+PHj7P8/VSqVEn7OMDUqVP56quvOH78ON26dcPBwYEOHToQEBCgPWbjxo0MHz6clStX0rx5c+zt7Rk2bBjh4eHZXr98+fJYWFgQHByss+f0X1k9v7p169K4cWPtm21qaio//fQTvXr1wt7eHoD79+8D8P7772NsbJzhNn78eAAePXpUaHELIURpVlbfm58+fcrmzZtp0qQJjo6OREZGEhkZSZ8+fdBoNNoEHtRaL1WqVMnxfA8fPqRSpUra6QK6ktXf59VXX2XJkiWMHj2aP/74g5MnT3Lq1CkcHR0z/H5zEzfAu+++y7Vr19i3bx8A3377Lc2bN6dBgwY5Hpf+JUxhfXbI6rmD+qWKv7+/th5A+pf7//6C5v79+5w7dy7T5wYrKysURZHPDWWEzEkXIheymvtz4MAB7t27x8GDB7Xf0AMZ5ljrm4ODAydPnsy0P6ekNz/XCAsLy7T/3r17ANr5ZUZGRkyePJnJkycTGRnJ/v37mTZtGl26dOH27dtYWFhQvnx5Fi5cyMKFCwkNDWXHjh188MEHPHjwgD179mR5fUNDQzp06MDu3bu5c+dOrt7UTU1Ns1xPNru579nN/Xr99dcZP348ly5d4ubNm4SFhWnnlP37uU+dOpW+fftmeY5atWo9N14hhBCZldX35vXr1xMXF8fJkycz9dYCbN26lSdPnmBnZ4ejoyN37tzJ8XyOjo4cPXqUtLS0bBP19C8e/vvemVPNmP/+faKiovj999+ZNWsWH3zwgXZ/YmIiERERmWJ6Xtyg9lh7enqyZMkSypUrR2BgID/99NNzj+vSpQsrVqxg27ZtGWLJjpmZWZafG7JLmLP73DB48GAmT56Mr68vn332GT/++CO9e/fO8HcsX7485ubmmYoV/vtxUfpJT7oQ+ZT+H/B/q4l/9913+ggnS23btuXp06fs3r07w/4NGzbo7BodOnTQfij6t7Vr12JhYUGzZs0yHWNra0v//v156623iIiIICQkJFMbV1dXJkyYQKdOnQgMDMwxhqlTp6IoCmPGjCEpKSnT48nJyfz222/a++7u7hmqr4P6we55y/v81+DBgzEzM8PX1xdfX18qV65M586dtY/XqlWLGjVqcPbsWRo1apTlzcrKKk/XFEIIkb2y8N68atUqrKys+PPPP/nrr78y3ObNm0diYiI///wzoFYov3r1ao4F87p160ZCQkKOVeWdnJwwMzPL9N65ffv2XMUM6t9GUZRMf5uVK1eSmpqaKaa//vorV1PC3nnnHXbu3MnUqVNxcnLilVdeee4xvXr1wsvLizlz5nDhwoUs2/zxxx/ExcUB6ueGBw8eaEfIgbq86x9//PHca/2bnZ0dvXv3Zu3atfz++++Eh4dnmmbQo0cPbty4gYODQ5afG7KqMi9KH+lJFyKfWrRogZ2dHePGjWPWrFkYGxvz888/c/bsWX2HpjV8+HAWLFjAkCFD+PTTT6levTq7d+/Wvqnkdmjb8ePHs9zftm1bZs2axe+//0779u2ZOXMm9vb2/Pzzz+zcuTNDxdWePXvi6elJo0aNcHR05NatWyxcuBA3Nzdq1KhBVFQU7du359VXX8XDwwMrKytOnTrFnj17su2FTte8eXOWLVvG+PHjadiwIW+++SZ169YlOTmZM2fOsGLFCjw9PbXzBIcOHcqHH37IzJkzadu2LRcvXmTJkiUZqsPmhq2tLX369MHX15fIyEjef//9TL/T7777jm7dutGlSxdGjBhB5cqViYiI4NKlSwQGBrJ58+Y8XVMIIUT2Svt784ULFzh58iRvvvlmlvPxW7Zsyfz581m1ahUTJkxg4sSJbNy4kV69evHBBx/QpEkT4uPjOXToED169KB9+/YMHjyYNWvWMG7cOK5cuUL79u1JS0vjxIkT1K5dm0GDBqHRaBgyZAirV6/mhRdewNvbm5MnT7Ju3bpcP29ra2vatGnDvHnzKF++PO7u7hw6dIhVq1Zha2uboe3HH3/M7t27adOmDdOmTcPLy4vIyEj27NnD5MmT8fDw0LYdMmQIU6dO5fDhw8yYMQMTE5PnxmJoaMjWrVvp3LkzzZs3580336R9+/ZYWlpy69YtfvnlF3777TeePHkCwMCBA5k5cyaDBg3i//7v/0hISOCbb77J9OVCbowcOZKNGzcyYcIEqlSpkqlewMSJE/n1119p06YNkyZNol69eqSlpREaGsrevXt57733aNq0aZ6vK0oY/datE6J4ya6CbN26dbNs7+fnpzRv3lyxsLBQHB0dldGjRyuBgYGZKqBmV0H2pZdeynTOtm3bKm3bttXez66C7H/jzO46oaGhSt++fZVy5copVlZWSr9+/ZRdu3YpgLJ9+/bsfhUZrp3dLT2m8+fPKz179lRsbGwUExMTxdvbO1MF2Pnz5ystWrRQypcvr5iYmCiurq7KqFGjlJCQEEVRFCUhIUEZN26cUq9ePcXa2loxNzdXatWqpcyaNUuJjY3NMc50QUFByvDhwxVXV1fFxMREsbS0VHx8fJSZM2cqDx480LZLTExU/ve//ykuLi6Kubm50rZtWyUoKCjb6u7/rbD6b3v37tX+Pq5evZplm7NnzyoDBgxQKlSooBgbGyvOzs7Kiy++qCxfvjxXz0sIIcoyeW/+x8SJExVACQoKyrbNBx98oADK6dOnFUVRlCdPnijvvvuu4urqqhgbGysVKlRQXnrpJeXy5cvaY+Lj45WZM2cqNWrUUExMTBQHBwflxRdfVPz8/LRtoqKilNGjRytOTk6KpaWl0rNnTyUkJCTb6u4PHz7MFNudO3eUfv36KXZ2doqVlZXStWtX5cKFC5nefxVFUW7fvq2MHDlScXZ2VoyNjZVKlSopAwYMUO7fv5/pvCNGjFCMjIyUO3fuZPt7yUpkZKTyySefKA0aNFDKlSunGBsbK66ursqQIUOUY8eOZWi7a9cupX79+oq5ublSrVo1ZcmSJdlWd3/rrbeyvWZqaqri4uKiAMr06dOzbBMTE6PMmDFDqVWrlmJiYqLY2NgoXl5eyqRJk5Tw8PA8PUdRMmkU5Vk5RCFEmfH5558zY8YMQkNDczWHWwghhBCFS96b8ycpKQl3d3datWrFpk2b9B2OEDohw92FKOWWLFkCgIeHB8nJyRw4cIBvvvmGIUOGyIcAIYQQQg/kvbngHj58yJUrV1izZg3379/PVQE4IUoKSdKFKOUsLCxYsGABISEhJCYm4urqypQpU5gxY4a+QxNCCCHKJHlvLridO3fy+uuvU7FiRZYuXfrcZdeEKElkuLsQQgghhBBCCFFMyBJsQgghhBBCCCFEMSFJuhBCCCGEEEIIUUxIki6EEEIIIYQQQhQTZa5wXFpaGvfu3cPKygqNRqPvcIQQQggUReHp06dUqlQJAwP5/lwX5P1eCCFEcZKX93q9JumHDx9m3rx5nD59mrCwMLZu3Urv3r2zbX/06FGmTJnC5cuXiYuLw83NjbFjxzJp0qRcX/PevXu4uLjoIHohhBBCt27fvi3LL+mIvN8LIYQojnLzXq/XJD02NhZvb29ef/11+vXr99z2lpaWTJgwgXr16mFpacnRo0cZO3YslpaWvPHGG7m6ppWVFaD+cqytrQsUvxBCCKEL0dHRuLi4aN+jRMHJ+70QQojiJC/v9XpN0rt160a3bt1y3d7HxwcfHx/tfXd3d7Zs2cKRI0dynaSnD3mztraWN20hhBDFigzL1h15vxdCCFEc5ea9vkRPfDtz5gx+fn60bds22zaJiYlER0dnuAkhhBBCCCGEEMVRiUzSq1SpgqmpKY0aNeKtt95i9OjR2badM2cONjY22pvMTxNCCCGEEEIIUVyVyCT9yJEjBAQEsHz5chYuXMj69euzbTt16lSioqK0t9u3bxdhpEIIIYQQQgghRO6VyCXYqlatCoCXlxf3799n9uzZDB48OMu2pqammJqaFmV4QgihE4qikJKSQmpqqr5DEQVkaGiIkZGRzDkvZlJTU0lOTtZ3GKKUkNe5EEJXSmSS/m+KopCYmKjvMIQQQqeSkpIICwsjLi5O36EIHbGwsKBixYqYmJjoOxSdW7p0KfPmzSMsLIy6deuycOFCWrdunW37n3/+mblz53Lt2jVsbGzo2rUrX331FQ4ODto2v/76Kx9++CE3btzghRde4LPPPqNPnz46izkmJoY7d+6gKIrOzilEaX6dCyGKjl6T9JiYGK5fv669HxwcTFBQEPb29ri6ujJ16lTu3r3L2rVrAfj2229xdXXFw8MDUNdN/+qrr3j77bf1Er8QQhSGtLQ0goODMTQ0pFKlSpiYmEjPTAmmKApJSUk8fPiQ4OBgatSogYFBiZxtlqWNGzcyceJEli5dSsuWLfnuu+/o1q0bFy9exNXVNVP7o0ePMmzYMBYsWEDPnj25e/cu48aNY/To0WzduhUAf39/Bg4cyCeffEKfPn3YunUrAwYM4OjRozRt2rTAMaempnLnzh0sLCxwdHSU15cosNL+OhdCFC2NosevkA8ePEj79u0z7R8+fDi+vr6MGDGCkJAQDh48CMDixYv57rvvCA4OxsjIiBdeeIExY8YwduzYXP9HGB0djY2NDVFRUbIkixCiWEpISCA4OBg3NzcsLCz0HY7Qkbi4OG7dukXVqlUxMzPL8FhJfm9q2rQpDRo0YNmyZdp9tWvXpnfv3syZMydT+6+++oply5Zx48YN7b7Fixczd+5cbd2YgQMHEh0dze7du7Vtunbtip2dXY51aP4tp99p+mvM3d0dc3PzPD1fIXKS0+tcCFG25eW9Xq896e3atctxmJmvr2+G+2+//bb0mgshygzphSldSuPfMykpidOnT/PBBx9k2N+5c2f8/PyyPKZFixZMnz6dXbt20a1bNx48eMAvv/zCSy+9pG3j7+/PpEmTMhzXpUsXFi5cmG0siYmJGaa/5WbJVelBF7pWGl/nQoiiJ/+TCCGEECJfHj16RGpqKk5OThn2Ozk5ER4enuUxLVq04Oeff2bgwIGYmJjg7OyMra0tixcv1rYJDw/P0zlBllwVQghRekiSLoQQQogC+W+PtKIo2fZSX7x4kXfeeYeZM2dy+vRp9uzZQ3BwMOPGjcv3OUGWXBVCCFF6lPjq7kIIIUqvdu3aUb9+/RyHOQv9KV++PIaGhpl6uB88eJCpJzzdnDlzaNmyJf/3f/8HQL169bC0tKR169Z8+umnVKxYEWdn5zydE2TJ1fyS15gQQhQ/0pMuhBCiwDQaTY63ESNG5Ou8W7Zs4ZNPPilQbCNGjKB3794FOofImomJCQ0bNmTfvn0Z9u/bt48WLVpkeUxcXFymebuGhoYA2jo1zZs3z3TOvXv3ZnvOsqA4v8bS+fn5YWhoSNeuXXVyPiGEKKukJ10IIUSBhYWFabc3btzIzJkzuXLlinbffytoJycnY2xs/Nzz2tvb6y5IUSgmT57M0KFDadSoEc2bN2fFihWEhoZqh6//dznVnj17MmbMGJYtW0aXLl0ICwtj4sSJNGnShEqVKgHw7rvv0qZNG7788kt69erF9u3b2b9/P0ePHtXb89S3kvAaW716NW+//TYrV64kNDQ0yyX4ikpun78QQhRH0pNeAFvP3KHrwsPM2XVJ36EIIUoxRVGIS0rRyy23q3Q6OztrbzY2Nmg0Gu39hIQEbG1t2bRpE+3atcPMzIyffvqJx48fM3jwYKpUqYKFhQVeXl6Zltdq164dEydO1N53d3fn888/Z+TIkVhZWeHq6sqKFSsK9Ps9dOgQTZo0wdTUlIoVK/LBBx+QkpKiffyXX37By8sLc3NzHBwc6NixI7GxsYC6lGiTJk2wtLTE1taWli1bcuvWrQLFU9IMHDiQhQsX8vHHH1O/fn0OHz7Mrl27cHNzA9TkMjQ0VNt+xIgRfP311yxZsgRPT09eeeUVatWqxZYtW7RtWrRowYYNG1izZg316tXD19eXjRs36mSN9KzIa2yi9n5+X2OxsbFs2rSJN998kx49emRaoQdgx44dNGrUCDMzM8qXL0/fvn21jyUmJvK///0PFxcXTE1NqVGjBqtWrQLU1X5sbW0znGvbtm0ZahTMnj2b+vXrs3r1aqpVq4apqSmKorBnzx5atWqFra0tDg4O9OjRI8PyfwB37txh0KBB2NvbY2lpSaNGjThx4gQhISEYGBgQEBCQof3ixYtxc3PL9d9OCFE8KIrC4j+vMWZtALcex+o7nBxJT3oBxCamcjn8KW4Oso6xEKLwxCenUmfmH3q59sWPu2Bhopu3iilTpjB//nzWrFmDqakpCQkJNGzYkClTpmBtbc3OnTsZOnQo1apVyzEZmz9/Pp988gnTpk3jl19+4c0336RNmzZ4eHjkOaa7d+/SvXt3RowYwdq1a7l8+TJjxozBzMyM2bNnExYWxuDBg5k7dy59+vTh6dOnHDlyBEVRSElJoXfv3owZM4b169eTlJTEyZMny+SyXuPHj2f8+PFZPpZVspabJVX79+9P//79dRHec8lrLKP8vMY2btxIrVq1qFWrFkOGDOHtt9/mww8/1L4edu7cSd++fZk+fTo//vgjSUlJ7Ny5U3v8sGHD8Pf355tvvsHb25vg4GAePXqUp+d//fp1Nm3axK+//qqdQhEbG8vkyZPx8vIiNjaWmTNn0qdPH4KCgjAwMCAmJoa2bdtSuXJlduzYgbOzM4GBgaSlpeHu7k7Hjh1Zs2YNjRo10l5nzZo1jBgxoky+1oUoqRRFYc7uy6w4fBOA4zcfs2BAfTrWyb7WiT5Jkl4AthbqMKrIuGQ9RyKEEMXfxIkTM/ScAbz//vva7bfffps9e/awefPmHBOI7t27axPCKVOmsGDBAg4ePJivJH3p0qW4uLiwZMkSNBoNHh4e3Lt3jylTpjBz5kzCwsJISUmhb9++2p5hLy8vACIiIoiKiqJHjx688MILANSuXTvPMQihK/p8ja1atYohQ4YA0LVrV2JiYvjzzz/p2LEjAJ999hmDBg3io48+0h7j7e0NwNWrV9m0aRP79u3Ttq9WrVpenjoASUlJ/Pjjjzg6Omr39evXL1OcFSpU4OLFi3h6erJu3ToePnzIqVOntEP/q1evrm0/evRoxo0bx9dff42pqSlnz54lKCgow8gPIUTxt3D/NW2CXs3RkpsPYxm9NoC32r/A5E61MDQoXl+6SZJeADbmapIeFS9JuhCi8JgbG3Lx4y56u7au/LsnCiA1NZUvvviCjRs3cvfuXRITE0lMTMTS0jLH89SrV0+7nT7k98GDB/mK6dKlSzRv3jxDj1jLli2JiYnhzp07eHt706FDB7y8vOjSpQudO3emf//+2NnZYW9vz4gRI+jSpQudOnWiY8eODBgwgIoVK+YrFqE/8hrLKK+vsStXrnDy5Elt4mpkZMTAgQNZvXq1NukOCgpizJgxWR4fFBSEoaEhbdu2zdXzzI6bm1uGBB3gxo0bfPjhhxw/fpxHjx6RlpYGQGhoKJ6engQFBeHj45Pt3PzevXszYcIEtm7dyqBBg1i9ejXt27fH3d29QLEKIYrOsoM3WPTnNQA+7FGHoc3c+HzXJXz9Qvj2rxucvR3FokH1cShXfFYIkSS9AGzNTQDpSRdCFC6NRqOz4bD69N/EYP78+SxYsICFCxfi5eWFpaUlEydOJCkpKcfz/LcYlEaj0X7wzqus1t5On2eq0WgwNDRk3759+Pn5sXfvXhYvXsz06dM5ceIEVatWZc2aNbzzzjvs2bOHjRs3MmPGDPbt20ezZs3yFY/QD3mNZZTX19iqVatISUmhcuXK2n2KomBsbMyTJ0+ws7PLVNju33J6DMDAwCDT/O/k5MyfvbL68qFnz564uLjw/fffU6lSJdLS0vD09NT+Dp53bRMTE4YOHcqaNWvo27cv69atk+XqhChBfI8F8+WeywD8X5dajGpVFYDZL9fFx9WWD349z9Hrj+ix+ChLX2uAj6udPsPVksJxBaAd7h6f85udEEKIzI4cOUKvXr0YMmQI3t7eVKtWjWvXrhVpDHXq1MHPzy9DAuDn54eVlZU24dBoNLRs2ZKPPvqIM2fOYGJiwtatW7XtfXx8mDp1Kn5+ftrhs0IUB0XxGktJSWHt2rXMnz+foKAg7e3s2bO4ubnx888/A2rv/J9//pnlOby8vEhLS+PQoUNZPu7o6MjTp0+1BRtB7X1/nsePH3Pp0iVmzJhBhw4dqF27Nk+ePMnQpl69egQFBREREZHteUaPHs3+/ftZunQpycnJmaYUCCGKpw0nQ5n920UA3n6xOm+1r57h8V71K7N9QkuqlbckLCqBAd/586N/SLEoCilJegHYPEvSE5LTSEhO1XM0QghRslSvXl3bS33p0iXGjh1LeHh4oVwrKioqQwIRFBREaGgo48eP5/bt27z99ttcvnyZ7du3M2vWLCZPnoyBgQEnTpzg888/JyAggNDQULZs2cLDhw+pXbs2wcHBTJ06FX9/f27dusXevXu5evWqzEsXxUZRvMZ+//13njx5wqhRo/D09Mxw69+/v7ZC+6xZs1i/fj2zZs3i0qVLnD9/nrlz5wJqRfnhw4czcuRItm3bRnBwMAcPHmTTpk0ANG3aFAsLC6ZNm8b169dZt25dlgUJ/8vOzg4HBwdWrFjB9evXOXDgAJMnT87QZvDgwTg7O9O7d2+OHTvGzZs3+fXXX/H399e2qV27Ns2aNWPKlCkMHjz4ub3vQgj923bmLlO3ngdgdKuqTO5UM8t2NZ2s2D6hJV3rOpOcqvDh9r+ZvOks8Un6ze0kSS8AK1MjbZEBmZcuhBB58+GHH9KgQQO6dOlCu3bttB+UC8PBgwfx8fHJcJs5cyaVK1dm165dnDx5Em9vb8aNG8eoUaOYMWMGANbW1hw+fJju3btTs2ZNZsyYwfz58+nWrRsWFhZcvnyZfv36UbNmTd544w0mTJjA2LFjC+U5CJFXRfEaW7VqFR07dsTGxibTY/369SMoKIjAwEDatWvH5s2b2bFjB/Xr1+fFF1/kxIkT2rbLli2jf//+jB8/Hg8PD8aMGaPtObe3t+enn35i165d2mXkZs+e/dzYDAwM2LBhA6dPn8bT05NJkyYxb968DG1MTEzYu3cvFSpUoHv37nh5efHFF19oq8OnGzVqFElJSYwcOTIfvyUhRG5FxSXzOCaxQOfYfT6M9zafRVHgtaauTH+pdo6rMViZGbNsSAOmd6+NoYGGrWfu0mfpMYIf6W+ZNo1SHPrzi1B0dDQ2NjZERUVhbW1d4PM1+GQfEbFJ7J3UhppOVjqIUAhR1iUkJBAcHEzVqlUxMzPTdzhCR3L6u+r6vUnk/DuV15jIq88++4wNGzZw/vz5HNvJvy0hci8qPpm/70Zx/m4U5+5Gcf5OFKERcWg00LqGI681daWDRwWMDHPfr3zg8n3G/nia5FSFfg2qMK9/PQzyULn9+M3HTFh3hkcxiViZGjF/gDed6zrn5+llkpf3+pJfJUXPbMyNiYhNkuJxQgghhBClTExMDJcuXWLx4sV88skn+g5HiBLraUIyf9+L5vwdNSG/cDcq255qRYHDVx9y+OpDnKxNGdjIhYFNXKlsm/NUk2PXHzHup0CSUxV61KvI3Dwm6ADNqjmw851WvPVzIAG3nvDGj6cZ1/YF3u9cM09fFhSUJOkFlL4MW2ScFI8TQgghhChNJkyYwPr16+ndu7cMdRfiOdLSFMKjEwh5HEvo4zhCHsdx63EsV+4/JfhRLFmN33axN8ersg1elW2pV8UGz0o2RMYnsf7kbTYH3OZ+dCLfHLjOkr+u075WBV5t6kq7WhUyrWt+KiSC0T8EkJSSRqc6TiwYWD/fa587WZux/o1mzNl1mdXHgll+6AZV7MwZ0swtX+fLD0nSC+ifCu/Sky6EEEIIUZr4+vrmqkidEGVFSmoadyPjufUsAU9PxG89juNWRBxJKdkv11jZ9llCXsXmWWJug52lSaZ2NhbGfNDNg8mdavLH3+GsOxGK/83H/Hn5AX9efkAlGzMGNXFlYGMXnKzNOHs7ktfXnCI+OZU2NR1Z8qoPxgXs9TY2NGBmzzo0cLNla+BdBjV2KdD58kqS9AKyfdaTHiXD3YUQQgghhBAlXGJKKrcj4rVJeOi/kvE7T+JJScu+pJmRgQYXewvcHCxws7fAzcGSao6WeFW2waGcaZ7iMDEyoKd3JXp6V+LmwxjWnwxl8+k73ItK4Ot9V1n05zVe9KjAyeAIYhJTaFbNnu+GNMTUyPD5J8+lHvUq8ZJXxRwLzxUGSdILyNZC/fZH1koXQgghhBBClCTX7j/lwOUHGXrE70XFZzk0PZ2pkYGahDtYqol4eUvcHSxws7ekkq1ZoczdruZYjukv1eG9zrXYc0HtXT8ZEsG+i/cBaOBqy8rhjTE30V2Cnq6oE3SQJL3A0uekyxJsQgghhBBCiJJAURR+OhHKx7/9TXJq5ozc0sQQNwdL3Murybh7elLuYIGTlVmeC7LpipmxIb19KtPbpzJX7z9lw8nbRMUnM7NnHcqZlp7UtvQ8Ez3RzkmX4e5CCCGEEEIIHYlPSmX6tvPcjohjWvfa+Lja6e68W8+z5cxdAJpXc6Cxu12GpNzB0kQvPch5UdPJipk96+g7jEIhSXoBSU+6EEIIIYQQZY+iKCw+cJ07T+KY2q12lkXQ8isqLpnRa09xKuQJAP2W+TGyZVXe61yrQEO6Qx7FMu6n01wOf4qhgYYpXWsxpnW1Yp+QlzVFt9hbKSU96UIIIYQQQpQ9Kw7f5Ot9V9kUcIeXvz3K1ftPdXLe+9EJDFzhz6mQJ1iZGdG1rjNpCqw8GkzXRYfxv/E4X+fd+3c4PRcf5XL4U8qXM+Xn0U15o80LkqAXQ5KkF5CNuRSOE0IIXWnXrh0TJ07U3nd3d2fhwoU5HqPRaNi2bVuhxiVEaSGvMSF0489L9/liz2UA7CyMuR0RT59vj7H/WSGz/Ap+FEu/ZX5cDn+Ko5Upm8Y2Z/nQhqwZ0ZiKNmbcehzH4O+PM23reZ4m5K6TMCU1jS/3XOaNH0/zNDGFRm527HynFc2qORQoVlF4JEkvIOlJF0II6NmzJx07dszyMX9/fzQaDYGBgXk+76lTp3jjjTcKFNuIESPo3bt3gc4hhL4V59dYOj8/PwwNDenatatOzidEcXX1/lPe3RCEosCrTV358712NKtmT2xSKmN+DGDpwesoOZVHz8aFu1G8styPO0/icXOwYMubLahd0RqA9h4V2DupDa81dQVg3YlQOi84zF+XH+R4zkcxiQxbfZJlB28AMLJlVda/0Qwna7M8xyeKjiTpBZS+TvrThBRSc1gzUAghSrNRo0Zx4MABbt26lemx1atXU79+fRo0aJDn8zo6OmJhYaGLEIUo0UrCa2z16tW8/fbbHD16lNDQUJ2cM7+Sk6XzRBSOJ7FJjP4hQLsu90cv18Xe0oQfRzXltaauKArM3XOFiRuDSEhOzfV5/W88ZtCK4zyKSaJORWt+GdcCF/uMr00rM2M+6+PF+jHNcHOwICwqgdd9TzFpYxBPYjOP6g0MfUKPb47id+MxFiaGLB7sw8yedTAuhCXShG7JX6iA0gvHAURL8TghRGFQFEiK1c8tlz0BPXr0oEKFCvj6+mbYHxcXx8aNGxk1ahSPHz9m8ODBVKlSBQsLC7y8vFi/fn2O5/3vUNxr167Rpk0bzMzMqFOnDvv27cvrbzOTQ4cO0aRJE0xNTalYsSIffPABKSkp2sd/+eUXvLy8MDc3x8HBgY4dOxIbGwvAwYMHadKkCZaWltja2tKyZcsskyhRzMlrTHs/v6+x2NhYNm3axJtvvkmPHj0yxQmwY8cOGjVqhJmZGeXLl6dv377axxITE/nf//6Hi4sLpqam1KhRg1WrVgHg6+uLra1thnNt27Ytwzza2bNnU79+fVavXk21atUwNTVFURT27NlDq1atsLW1xcHBgR49enDjxo0M57pz5w6DBg3C3t4eS0tLGjVqxIkTJwgJCcHAwICAgIAM7RcvXoybm1u+ekpFyZacmsabP58mNCIOF3tzlr7WUJvwGhsa8FkfLz7p7YmhgYbtQfcY+J0/4VEJzz3vngvhDF99kpjEFJpWtWfD2GY4Wplm2775Cw7sebcNY1pXxUADW8/cpdOCQ+w8F4aiKCiKwg9+Ier1oxN4wdGS7W+1pKd3JZ39LkThkuruBWRkaEA5UyNiElOIjE/WaVVHIYQAIDkOPtfTG+u0e2Bi+dxmRkZGDBs2DF9fX2bOnKn98Lx582aSkpJ47bXXiIuLo2HDhkyZMgVra2t27tzJ0KFDqVatGk2bNn3uNdLS0ujbty/ly5fn+PHjREdHZ5hbmx93796le/fujBgxgrVr13L58mXGjBmDmZkZs2fPJiwsjMGDBzN37lz69OnD06dPOXLkCIqikJKSQu/evRkzZgzr168nKSmJkydPSgGekkheY0DBXmMbN26kVq1a1KpViyFDhvD222/z4YcfauPcuXMnffv2Zfr06fz4448kJSWxc+dO7fHDhg3D39+fb775Bm9vb4KDg3n06FGurp3u+vXrbNq0iV9//RVDQ7X6dWxsLJMnT8bLy4vY2FhmzpxJnz59CAoKwsDAgJiYGNq2bUvlypXZsWMHzs7OBAYGkpaWhru7Ox07dmTNmjU0atRIe501a9YwYsQIea2XEIqikJqmYKSD3uOPfvub4zcjsDQxZNXwxthn8bl/aDM3XnC0ZPzPgZy9E8XLS46yYlgj6rvYZnnODSdDmbb1PGkKdK7jxDeDfTAzfn71dnMTQ6a/VIfuXhWZ8us5rt6P4a11gXSp64SZsSHbg+4B8JJXRb7sX69UrSFeFshfSwdszI3VJD0uCXj+G60QQpRGI0eOZN68eRw8eJD27dsD6vDXvn37Ymdnh52dHe+//762/dtvv82ePXvYvHlzrhKI/fv3c+nSJUJCQqhSpQoAn3/+Od26dct3zEuXLsXFxYUlS5ag0Wjw8PDg3r17TJkyhZkzZxIWFkZKSgp9+/bFzc0NAC8vLwAiIiKIioqiR48evPDCCwDUrl0737EI8TzF+TW2atUqhgwZAkDXrl2JiYnhzz//1M6j/+yzzxg0aBAfffSR9hhvb28Arl69yqZNm9i3b5+2fbVq1XLzK8kgKSmJH3/8EUdHR+2+fv36ZYqzQoUKXLx4EU9PT9atW8fDhw85deoU9vb2AFSvXl3bfvTo0YwbN46vv/4aU1NTzp49S1BQEFu2bMlzfKLoHb32iBnbzhOTmMKHPerwsnelfH+58qN/CD8dD0WjgUWDfKjpZJVt2xYvlGf7Wy0Z/UMA1x7EMOA7f+b2q0dvn8raNoqisPTgDeb9cQWAgY1c+KyPZ56/TPBxteO3t1vx7V83WPrXdf74Wy1cZ2igYWo3D0a1qipfKJVAkqTrgK2FMXcj44mU4e5CiMJgbKH2tunr2rnk4eFBixYtWL16Ne3bt+fGjRscOXKEvXv3ApCamsoXX3zBxo0buXv3LomJiSQmJmJpmbsvNy9duoSrq6s2eQBo3rx53p5PFuds3rx5hg8wLVu2JCYmhjt37uDt7U2HDh3w8vKiS5cudO7cmf79+2NnZ4e9vT0jRoygS5cudOrUiY4dOzJgwAAqVqxYoJiEHshrDMj/a+zKlSucPHlSm7gaGRkxcOBAVq9erU26g4KCGDNmTJbHBwUFYWhoSNu2bXMVZ3bc3NwyJOgAN27c4MMPP+T48eM8evSItLQ0AEJDQ/H09CQoKAgfHx9tgv5fvXv3ZsKECWzdupVBgwZpf/fu7u4FilUUriexSXyy8yJbAu9q9727IYjfzt7j095eONvkrWia3/VHzP7tIgD/6+JBxzpOzz3GzcGSLeNbMGljEPsvPWDixiAuhz/l/7rUQgN8tusSq44GA/Bmuxf4X5da+U6mTY0MmdypJt08nZm+9TwPniby9YD6NKma9b9rUfxJkq4D6RXeo6TCuxCiMGg0uRoOWxyMGjWKCRMm8O2337JmzRrc3Nzo0KEDAPPnz2fBggUsXLgQLy8vLC0tmThxIklJuVvCMqv5nwXtHVAUJdM50q+j0WgwNDRk3759+Pn5sXfvXhYvXsz06dM5ceIEVatWZc2aNbzzzjvs2bOHjRs3MmPGDPbt20ezZs0KFJcoYvIaA/L/Glu1ahUpKSlUrpyxl9DY2JgnT55gZ2eHubl5tsfn9BiAgYFBptiyKgyX1ZcRPXv2xMXFhe+//55KlSqRlpaGp6en9nfyvGubmJgwdOhQ1qxZQ9++fVm3bt1zl6wT+qMoCtuD7vHx7xeJiE1Co4FhzdywszTh27+us//SA04EH2J699oMbOySq3/fIY9iefPnQFLTFPr4VGZc29yP8rAyM+a7oY34au8Vlh28wfJDN7h2/ynW5sZsPaN+gTDjpdqMbp33kSNZqV3Rmi3jW2b53iZKFikcpwO26Wulx8la6UKIsm3AgAEYGhqybt06fvjhB15//XXtB4UjR47Qq1cvhgwZgre3N9WqVePatWu5PnedOnUIDQ3l3r1/ejz9/f0LFG+dOnXw8/PLkAD4+flhZWWlTTg0Gg0tW7bko48+4syZM5iYmLB161Ztex8fH6ZOnYqfn592+KwQhaW4vcZSUlJYu3Yt8+fPJygoSHs7e/Ysbm5u/PzzzwDUq1ePP//8M8tzeHl5kZaWxqFDh7J83NHRkadPn2oLNoLa+/48jx8/5tKlS8yYMYMOHTpQu3Ztnjx5kqFNvXr1CAoKIiIiItvzjB49mv3797N06VKSk5MzFLwTxcftiDiGrznFxI1BRMQmUdOpHL++2YKPenkysWNNfn+7Nd4utjxNSOGDLecZsuoEtyPicjxndEIyo9cGEBWfjLeLLXP6euU5+TU00DClqwcLB9bHxMiAPy8/YOuZuxgaaJj/irfOEvR/kwS95JMkXQds0tdKl+HuQogyrly5cgwcOJBp06Zx7949RowYoX2sevXq2l7pS5cuMXbsWMLDw3N97o4dO1KrVi2GDRvG2bNnOXLkCNOnT8/VsVFRURkSiKCgIEJDQxk/fjy3b9/m7bff5vLly2zfvp1Zs2YxefJkDAwMOHHiBJ9//jkBAQGEhoayZcsWHj58SO3atQkODmbq1Kn4+/tz69Yt9u7dy9WrV2VeuihUxe019vvvv/PkyRNGjRqFp6dnhlv//v21FdpnzZrF+vXrmTVrFpcuXeL8+fPMnTsXUCvMDx8+nJEjR7Jt2zaCg4M5ePAgmzZtAqBp06ZYWFgwbdo0rl+/zrp167KsHv9fdnZ2ODg4sGLFCq5fv86BAweYPHlyhjaDBw/G2dmZ3r17c+zYMW7evMmvv/6a4cuJ2rVr06xZM6ZMmcLgwYOf2/suilZKahorj9yk84LDHL76EBNDA97rpCblDVzttO1qOVux5c0WTO9eG1MjA45df0znBYdZfTQ4y2WUU9MU3l1/husPYnC2NuP7oQ1zVdAtO719KrN5bHOcrE0xNzZkxdCG9GtY5fkHijJJknQdSF8rPUqSdCGEYNSoUTx58oSOHTvi6uqq3f/hhx/SoEEDunTpQrt27bQfjHPLwMCArVu3kpiYSJMmTRg9ejSfffZZro49ePAgPj4+GW4zZ86kcuXK7Nq1i5MnT+Lt7c24ceMYNWoUM2bMAMDa2prDhw/TvXt3atasyYwZM5g/fz7dunXDwsKCy5cv069fP2rWrMkbb7zBhAkTGDt2bJ5+X0LkVXF6ja1atYqOHTtiY2OT6bF+/foRFBREYGAg7dq1Y/PmzezYsYP69evz4osvcuLECW3bZcuW0b9/f8aPH4+HhwdjxozR9pzb29vz008/sWvXLu2ycrNnz87V89mwYQOnT5/G09OTSZMmMW/evAxtTExM2Lt3LxUqVKB79+54eXnxxRdfaKvDpxs1ahRJSUmMHDnyudcVRefC3Sj6LPXj052XiE9OpUlVe3ZPbM3bHWpgYpQ5zTE00DCmTTX+mNiGplXtiU9O5ePfL/LKcj+uP3iaoe2Xey7z15WHmBoZsGJYQypY520ee1a8XWw59H/tOT61Ax1qP39euyi7NEoZW+QxOjoaGxsboqKisLa21sk5vzt0gzm7L9PXpzJfD6yvk3MKIcquhIQEgoODqVq1KmZmBf9QIIqHnP6uhfHeVNbl9DuV15jIq88++4wNGzZw/vz5HNvJv62iEZ+UysL9V1n5rBfc2syIad1rM6CRCwYGuRvqnZamsO5kKF/svkxMYgomhga827EGb7Spxvage7y/+SwAiwf7yPriQify8l4vheN0wFaGuwshhBBClDoxMTFcunSJxYsX88knn+g7nDIrNjGFsKgEwqMSCI2IY/mhG4Q+m0/+kldFZr1chwpWeftSxMBAw5BmbrzoUYFpW89z8MpD5v1xhd/O3uPmQ3UUx9svVpcEXeiFJOk6YCOF44QQQgghSp0JEyawfv16evfuLUPdC0lCciohj2O1Sbj6M57w6ETCo+IJi0rgaUJKpuMq2pjxSS/PXC2HlpNKtuasGdGYrWfu8vHvF7kcrg5771LXiUkdaxbo3ELkl16T9MOHDzNv3jxOnz5NWFgYW7duzXHu1JYtW1i2bBlBQUEkJiZSt25dZs+eTZcuXYou6CxIT7oQQgghROnj6+ubqyJ1Iu8SklP5+UQo3/51nYjY53d0WZka4WxjhrONGd5VbBnbthpWZsY6iUWj0dC3QRVa13Bk7p7LxCWnMrdfvVwPnRdC1/SapMfGxuLt7c3rr79Ov379ntv+8OHDdOrUic8//xxbW1vWrFlDz549OXHiBD4+PkUQcdZknXQhhBBCCCGeLyU1jS2Bd1m4/yr3ohIAsDIzooqdBRWfJeEVrc1wsjGj4rObk7WZzhLynDhamTLvFe9Cv44Qz6PXJL1bt25069Yt1+0XLlyY4f7nn3/O9u3b+e233/SbpKcPd49PRlEUWZtQCKETZayuZ6knf8/iR/4mQtfk31T2FEVh94Vw5u+9wo1nc76drc14t2MN+jesgrGhLDolRLoSPSc9LS2Np0+fYm9vn22bxMREEhMTtfejo6N1HofNsyXYUtMUYpNSKWdaon+tQgg9MzZW/0+Ji4uT9XhLkbg4tchR+t9X6E/68lpJSUnyGhM6Ja/zzBRF4ci1R8z74wrn70YBYGdhzPh21Rna3K1Aa48LUVqV6Gxy/vz5xMbGMmDAgGzbzJkzh48++qhQ4zAzNsDEyICklDQi45IkSRdCFIihoSG2trY8ePAAAAsLCxmhU4IpikJcXBwPHjzA1tY20/rLougZGRlhYWHBw4cPMTY2xsBAevBEwcjrPGuBoU+Yt+cK/jcfA2BpYsio1tUY07pqkQxfF6KkKrHZ5Pr165k9ezbbt2+nQoUK2babOnUqkydP1t6Pjo7GxcVFp7FoNBpszY158DSRyLhkqtjp9PRCiDLI2dkZQJuoi5LP1tZW+3cV+qXRaKhYsSLBwcHcunVL3+GIUkRe56or4U/5au8V9l28D4CJoQGvNXPlrfbVKV/OVM/RCVH8lcgkfePGjYwaNYrNmzfTsWPHHNuamppialr4/xnYWqhJepRUeBdC6EB6ElGhQgWSk+X/lZLO2NhYetaKGRMTE2rUqEFSkiyfKnSjrL3O09IUHsYkcudJHHeexGtvtx7H4n/zMYoCBhro16AK73asQRU7C32HLESJUeKS9PXr1zNy5EjWr1/PSy+9pO9wtLTF46TCuxBChwwNDcvUhz4hipKBgQFmZmb6DkOIYu3OkzgCQp78JxmP415kAkmpadke183Tmfc616R6BasijFaI0kGvSXpMTAzXr1/X3g8ODiYoKAh7e3tcXV2ZOnUqd+/eZe3atYCaoA8bNoxFixbRrFkzwsPDATA3N8fGxkYvzyGdjXatdPlGXgghhBBClGyKorAp4DYzt/9NYkrWybiBBiramFPFzpzKduZUsbOgip059arY4OFsXcQRC1F66DVJDwgIoH379tr76XPHhw8fjq+vL2FhYYSGhmof/+6770hJSeGtt97irbfe0u5Pb69Pts8qvEtPuhBCCCGEKMniklL4cNvf/Bp4B4C6laypXdGaKs8S8cq2amLubGMmS6cJUQj0mqS3a9cux/Uk/5t4Hzx4sHADKoD0ZdiiZU66EEIIIYTIp78uP+DOkzg8Klrj4WxV5FXQrz+IYfzPp7l6PwYDDbzXuRZvtn0BAwNZZUSIolLi5qQXV7YW0pMuhBBCCCHyb61/CDO3/51hn5uDBbWdralTyZo6FdWfFW3MCmVpzh1n7zH113PEJqXiaGXK4sE+NKvmoPPrCCFyJkm6jthYPCscJ3PShRBCCCFEHv10/JY2Qa/vYsv96ATCohK49TiOW4/j2PN3uLatjbmxNmGvXdGahm52VC1vme9rJ6ak8snvF/npuDrNtHk1BxYNrk8FKymsKIQ+SJKuIzInXQghhBBC5Me6E6HM2HYBgDGtqzKte200Gg1PYpO4FBbNxbBoLt5Tf15/EENUfDL+Nx/jf/Ox9hy1K1rT07siPetVwsU+98ud3Y6IY/zPgZy/GwXAhPbVmdSpJoYyvL1w7J8Nj69D35VgLF+CiKxJkq4j6cPdZZ10IYQQQgiRWxtOhjJt63kARrX6J0EHsLM0oUX18rSoXl7bPjEllWv3Y7TJ+993owkMfcKlsGguhUUzd88VvF1s6VmvIi/Vq0hFG/Nsr73v4n3e2xREdEIKdhbGfD2wPu1rVSjcJ1yWPb0PRxeo239vhfqD9RuPKLYkSdcRWSddCCGEEELkxaZTt5n6LEF/vaU7M16q/dy55qZGhnhWtsGz8j/LDz+JTWLP3+H8fu4e/jcec/Z2JGdvR/Lpzks0cbenh3dFunlWxNHKFIDk1DS++uMK3x2+CYCPqy3fvtqASrbZJ/RCB67v+2c7YJUk6SJbkqTrSHp1d+lJF0IIIYQQz7M54DZTtpxDUWBEC3dm9qiT72JwdpYmDG7iyuAmrjx8msjuC2H8dvYep0KecDIkgpMhEcze8TfNX3Cgm2dFtgfd5VTIE0DtvZ/S1QMTI1lKrdBd3fPP9p1TEHYOKtbTXzyi2JIkXUdsng13j09OJSE5FTNjQz1HJIQQQgghiqNfT9/hf7+qCfqw5m7M6pn/BP2/HK1MGdbcnWHN3QmLimfnuTB+OxfG2duRHLv+mGPX1XnsVqZGzHulHl09K+rkuuI5UhLhxl/qtmNteHhJ7U3vuUi/cYliSZJ0HbEyNcJAA2mKula6JOlCCCGEEOK/tp65w/u/nEVRYEgzVz56uW6hLKcGUNHGnNGtqzG6dTVCH8fx+/l77D4fjrmxIXP718O9ABXhRR7dOgZJMVDOCbrPhR96wrnN0OkTMLPWd3SimJEkXUcMDDTYmBvzJC6ZyPhkKlhLtUYhhBBCiNIiNU3hj7/D8fULIT4plbY1HWnvUYH6Lra5roS+Pegu721SE/RXm7ry8cuehZag/5ergwXj21VnfLvqRXI98R9X96o/a3QG99ZQvhY8ugLnNkKTMfqNTRQ7kqTrkK2FiZqkS/E4IYQQQohSISE5lS2Bd/n+yE2CH8Vq95+/G8WSv65jb2lCu1qOvOhRgdY1HLV1iv7rt7P3mLQxiDQFBjV24dNenhjIMmdlg6LA1d3qds2uoNFAo5GwZwoErIbGo9V9pVHkbYh5AFUa6juSEkWSdB2y0a6VnqTnSIQQQgghREFEJyTz0/FbrDkWwsOniYC65O6w5u642Vvw15UHHLr6kIjYJLYE3mVL4F2MDDQ0crejg4cT7T0q8IKjJRqNht/P3WPiswR9QKMqfN7HSxL0suTRNXgSAoYmUK2dus97EPz5ETy4CKHHwa25PiMsHE9uwYp2EP8ERu+HKo30HVH+JD4FU6sivaQk6TqUvlZ6pFR4F0IIIYQokR5EJ7DqWDDrjofyNDEFgEo2ZoxqXY1BjV2wNFU/PvdrWIXk1DRO33rCgcsPOHD5AdcfxHD8ZgTHb0bw2a5LuDlY0MjNnm1Bd0lNU+jfsApf9K0nCXpZc+0P9ad7KzAtp26b24JnPzjzo1pArrQl6UlxsPE1iI9Q7x9dAIN+1m9MeXUnAPy/hev74d2zYGFfZJeWJF2H0nvSoyVJF0IIIYQoUW4+jGHF4ZtsCbxLUmoaADWdyjG2zQu8XL8SxoaZlygzNjSgWTUHmlVzYFr32oQ+juPA5fv8efkBJ25GcOtxHLcexwHQt0FlvuwnCXqZdPVZkl6za8b9jUaqSfrF7dD1C7AsX/SxFQZFgd/ehfDzYG4H8ZFw+Xd4cBkqeOg7upylJsOlHXB8mbpMXrrr+6HegCILQ5J0HbLVDneXJF0IIYQQoiS4eC+axQeusefvcBRF3dfIzY5xbV/gRY8KeUqqXR0sGNGyKiNaViU2MYWj1x9x8MpDHMuZ8G7HmrkuMCdKkfhIuOWnbtfonPGxyg2gkg/cO6Mm660mFXl4heL4Mji/CTSGMPAn9f7l3+HYIuizTN/RZS0uAgJ/gJPfQ/RddZ+hCXi9As3eBGevIg1HknQdsrEwASAyXuakCyGEEEIUd0G3IxnwnT9JKWrPecfaFRjX9gUauRd8WKulqRFd6jrTpa5zgc8lSrAbf4KSqlZzt6+a+fFGo2DHBAhYAy3eBYPMIzZKlODDsHeGut3lc3WIv5G5mqSf3wTtp4Gti35j/LeHV+HEMghaDynx6j5LR7WYX6ORUK6CXsKSJF2HpCddCCGEEKJkePA0gXE/niYpJY0WLzgwq2ddajkXbXEoUQakL71Ws0vWj3v2g73TIfIW3DgANToWXWy6FhkKm0eoX0p4D4amY9X9VRpC1TZqAu//LXT7QnfXjLoDdwPVwm5mNs9utura84ZZr7SAoqi/6+PL4Pq+f/Y7eam95l79wchUdzHmgyTpOpReOC5K5qQLIYQQQhRbSSlpjP8pkPDoBKpXKMd3QxtiZZbNB3oh8istFa6lJ+lds25jYgHer6q9uQGrSm6SnhwPG4dA3GOo6A09FmRcVq7VJDVJD/wB2vwfWDoU/JqJT2FVF4i+k/Xjxpb/Stz/dQs/Bw8vP2ukgVrd1eTcvVWxWQpPknQd0lZ3l550IYQQQohia/ZvfxNw6wlWZkaskARdFJY7AWp1czMbcGmafbtGI9Uk/eoedV3x4jQcPDcUBX6fBGFnwcIBBv4MxuYZ21RrrybvYWfh5HfqsPeCOviFmqCb24FVJUiIUm9JT9XHk2PV29N7mY81KQc+Q9TefvtqBY9FxyRJ1yEbc5mTLoQQQghRnP184hbrToSi0cA3g3yo5lhO3yGJ0urqHvVn9Y5gmEPa5VgT3FtDyBG1p/nFGUUTn66c+A7OrlcLxb3im/WXDBoNtJoMm4er7Vu8889ydPkRfl4drg7Q93uo0emfx1JTIDH6n6Q9IQoSIv/ZNikHnn3VL0+KKUnSdSh9CbYo6UkXQgghhCh2AkIimL3jbwDe71yL9h76KQolyojnDXX/t0YjnyXpa6HtlOznUxc3IUfhj2e94p0/VeeeZ6d2T7B/ASJuqF9GNH8rf9dMS4PfJ6tz3+v0ypigg/qFiIV9ka5rrmslvHxg8ZI+3D06IYXUNEXP0QghhBBCiHThUQmM+ymQ5FSF7l7OjG/3gr5DEqVZ5G24fwE0BmpP+vN49ADLChBzX62EXhJE3oZNw9Vk2WuAOq87JwaG0PJdddtvCaQk5u+6Z9bCnZNqj3hXHRahK0YkSdeh9J50gGgpHieEEEIIUSwkJKcy9qfTPIpJxMPZinn9vdEUkwJRopS69of6s0qT3PXoGplAg2HqdsDqwotLV7SF4h6Bcz3ouSh3Rde8B4FVRXWe+LlNeb9u7CPYN0vdbj8NrCvl/RwlgCTpOmRsaEA5U3UGQaQk6UIIIYQQeqcoCjO2XeDs7UhszI1ZMbQRlqYy41MUsqvPkvTsll7LSsMRas978GF4dK1QwtIJRVGHm4cFgbk9DPxJrVKfG0am/wxzP7ZIrYCfF3s/VOeXO3lBk7F5O7YEkSRdx2y0a6VL8TghhBBlw9KlS6latSpmZmY0bNiQI0eOZNt2xIgRaDSaTLe6detq2/j6+mbZJiEhoSiejihlfvAL4ZfTdzDQwJJXfXB1yGUyIUR+JcWpiTbkbj56OlsXqPEsqS/Oveknv4ez69QvFF5ZA3ZueTu+4Qi1aNvja3B5Z+6PCzmmXheNusRbTsX4SjhJ0nVMuwyb9KQLIYQoAzZu3MjEiROZPn06Z86coXXr1nTr1o3Q0NAs2y9atIiwsDDt7fbt29jb2/PKK69kaGdtbZ2hXVhYGGZmZkXxlEQp4n/jMZ/svATA1G61aV3DUc8RiTIh+DCkJICNK1SonbdjG49Sfwb9rCb7xc2Nv+CPqep2p0+gWru8n8PUCpq8oW4f/VrtmX+elCTYOVndbjgcXBrn/boliCTpOiYV3oUQQpQlX3/9NaNGjWL06NHUrl2bhQsX4uLiwrJly7Jsb2Njg7Ozs/YWEBDAkydPeP311zO002g0Gdo5OzvnGEdiYiLR0dEZbqJsu/MkjrfWBZKaptC7fiVGt66q75BEWZG+9FrNzrmbp/1vL7wItq7qUmF/b8nbsanJcHYjHPlaXY88N8lvbsRHwqlV8H0H+LE3pKWAZ//8V2cHaDoOjMzh3hkIPvT89v5L4OFlsCgPHWbl/7olhCTpOpbekx4lPelCCCFKuaSkJE6fPk3nzp0z7O/cuTN+fn65OseqVavo2LEjbm4Zh0vGxMTg5uZGlSpV6NGjB2fOnMnxPHPmzMHGxkZ7c3HJYp1eUWbEJ6Uy9sfTRMQm4VnZmi/61ZNCcWVVWqruktXcUJS8Lb32XwaG0PDZl5anVuXumNRkCPwRFjeErW/Anx/Bd21goRfsnqL27Kem5C2OtFS4vh9+GQlf1VR7se8GqGuhe/aDlxfn/QuIf7Ms/0+hvKMLcm775BYcmqtud/60RC+tlluldyC/ntiYmwAQKT3pQgghSrlHjx6RmpqKk5NThv1OTk6Eh4c/9/iwsDB2797NunXrMuz38PDA19cXLy8voqOjWbRoES1btuTs2bPUqFEjy3NNnTqVyZMna+9HR0dLol5GKYrCB1vO8fe9aBwsTfhuaCPMjA31HZYoKqkpai9yyBF1De/Q42BsBm3+B41eL/z1x+9fgOi7YGwB7q3zdw6fofDX53AvUO1pruSTdbvUZDi7AQ7Pg8hb6j5LR6jcEG4egqjbcGK5ejO3g5rdwOMltbc+u0Jvj66pQ+3PboCnYf/sr1AH6r8G9QZAuQr5e17/1WICBKyCmwfhbiBUbpB1u91TICUe3Fqp1eHLAEnSdeyfOelSOE4IIUTZ8N8eSkVRctVr6evri62tLb17986wv1mzZjRr1kx7v2XLljRo0IDFixfzzTffZHkuU1NTTE1N8x68KHVWHL7J9qB7GBpo+Pa1BlS2Ndd3SKIwpaZA+Fk1IQ85Crf8IelpxjZJT2H3/8Gp79We2Br5GIaeW+lD3au2Vb8cyI9yjlCnF1z4RS0g9/LijI+nJsO5jWpy/iRE3WfpCC0nQqORagKeFKcmv5d3wpVdEB+hFl07u04dZv7Ci2rCXrOr2nv/91YIWqeuP57O3E5d/7z+q1DRW/e/M1tX8HoFzq5Xe9MH/pi5zeWdcHU3GBjBS/ML7+9WzEiSrmO2MiddCCFEGVG+fHkMDQ0z9Zo/ePAgU+/6fymKwurVqxk6dCgmJiY5tjUwMKBx48Zcu1aMlyQSevc4JpGPfrvIjrP3AJjZow7NqjnoOSqhc2mp6tJfOSXlZjbg1hLcW6k/7waoPdOPrsK6AWqxs86fgbOn7uPLz9JrWWk8Sk3Sz/+ifrFgZpO75DydiQV4dFdvqSlw+4Sa8F7+DSJD4cpO9aYxAANjSE1Uj9MYQo1OamJes6u6ZFphavmumqRf+k3txS//r9FSiTGw63/qdot3oIJH4cZSjEiSrmNS3V0IIURZYWJiQsOGDdm3bx99+vTR7t+3bx+9evXK8dhDhw5x/fp1Ro0a9dzrKIpCUFAQXl5eBY5ZlD6KovDbuTBm7/ibiNgkDDQwoX11hjXP47JQovhLS1MLl6Uvb5bO1AbcnyXl7q3AyVPtHU5Xqb7aY3tkPhxfpvYwf9cafIZA+xlglfOXirkW+wjuBKjbBU3SXZuDY214eAnO/ARmts+S82D1cUtHNcFtNBJMLHM+l6HRs99PS+jymTok//JOuPw7hJ9XE3TH2uDzmtpzrqvfR25UqA21uqu9/ccWQa8l/zx26EuIvqP2uLf5v6KLqRiQJF3H/pmTLsPdhRBClH6TJ09m6NChNGrUiObNm7NixQpCQ0MZN24coM4Vv3v3LmvXrs1w3KpVq2jatCmenpl7sj766COaNWtGjRo1iI6O5ptvviEoKIhvv/22SJ6TKDnuRycwfesF9l+6D4CHsxVf9quHt4utfgMThePaXjVBNzRVh2unJ+XOXhmT8qyY2UCnj9Wkdv9sdXh34Fq4sAVaTVIrlRsXcGrEtX2AAs71wLpSwc6l0ai96bvehz+m/bM/L8l5dud19lJv7T5Qe9WT46F8Tf0NJW81SU3Sz26A9tPU3939i3B8qfp4t3nZz6EvpSRJ1zHtEmzSky6EEKIMGDhwII8fP+bjjz8mLCwMT09Pdu3apa3WHhYWlmnN9KioKH799VcWLVqU5TkjIyN54403CA8Px8bGBh8fHw4fPkyTJk0K/fmIkkFRFDYF3ObTnZd4mpCCsaGGt9pXZ3y76pgYyeJFpdaJ5erPJmPUHuH8sHOHV3zVJcD+mAZ3T8OBTyBgDXScDV7985+sapdeK2Averp6A9QvFJJiCp6cZ8fWVXfnyi+XJmpRuFtHwf9bdf31nZPVpd48ekCtfFTJL+E0ilKUaxLoX3R0NDY2NkRFRWFtba3z818Ki6bboiOUL2dCwIxOOj+/EEKI0qew35vKIvmdll63I+KYuuU8R68/AsC7ig1z+3tTy9lKz5EJ0tLA9yWIfwKj94NpOd2d+8FlWNpUnUP9ThDY6WA6Q1oaXPhVTYSj76j7KjeCrnPUxDEvUpJg3guQGA2j/4QqjQoeH0DoCXh8Der20W1yXtxc2w8/9wNjS7WHf9+H6vaEk2BTRd/R6URe3pfkq0Yd085Jj0umjH3/IYQQQghRaFLTFNYcC6bzgsMcvf4IUyMDpnX34Nc3W0iCXlzcOgahfuo86tNrdHvuk9+pP2t1102CDmBgAPVegbcD4MUPwaScWmRuVWf4a45apC63Qv3VBN2iPFTKZimx/HBtqs6dL80JOkD1DuoQ/ORYNUEHNVkvJQl6Xslwdx2zfTYnPSVNITYplXKm8isWQgghhABIS1PYfSGc0Ig47C2NsbMwwc7SBDsLE+wtTbAxN8bQIPNQ4+sPYpjy6zlO33oCQJOq9nzZrx5Vy5fyxKWkObfhn22/xdB4dMHneYPaM3/22bmbjiv4+f7L2BzavK+uT75/llpt/NAX6pcO/VblrpDatb3qzxqd1eRf5I1Go85N/2Wker9CHWj2pn5j0iPJIHXMzNgAEyMDklLSiIxLkiRdCCGEEAJ1mPr/fjmH/83H2bbRaNTlbO0sTbB/lsBbmBiy+0I4SSlpWJoYMrV7bV5t4opBFsm80KPkeLi4Q902toSY+2pV8iZjCn7uwB8hOU6t2u7equDny46VE/RZrhal+20ihByB5a2g3/fqsm050fV89LKodi9w9IDH16HHAjA01ndEeqPXr3kOHz5Mz549qVSpEhqNhm3btuXYPiwsjFdffZVatWphYGDAxIkTiyTOvNBoNNq10iNlrXQhhBBClHFpaQo/Hr9Fl4WH8b/5GHNjQ172rsSLHhWo72KLu4MF1mZqp4aiwJO4ZG4+jCXg1hP2XbzP9qB7JKWk0bamI3snt2VIMzdJ0IujK7vV4d42LtDpI3Xf0YXqXO2CSE2Bk9+r203HFk0F8noD4I2Dam9u7ANY2xsOfpH98PfHN9TE0sBITfBF/hgaweu74a2T4NpM39HolV67eWNjY/H29ub111+nX79+z22fmJiIo6Mj06dPZ8GCBUUQYf7YmBvz4GmiVHgXQgghRJl2OyKOKb+ew++G2nvexN2eea/Uw80h8zD15NQ0IuOSeRKXRERsEpFxSUTEqvdrVChHpzpOaPS1RJR4vnMb1Z/1BqjDxg9/pRZjO7cBGgzL/3mv7oaoUDC3V9c6LyqONdUCcLv/B2d+hINz4JYf9FsJ5Sr8J8Y/1J9uLcFMClUWiIW9eivj9Jqkd+vWjW7duuW6vbu7u3a5ltWrVxdWWAWWXjxOknQhhBBClEVpaQo/nwxlzq5LxCWlYmZswJSuHgxv7p5tL7ixoQGOVqY4WpkWcbSiwGIfwfX96na9QWBsBi3fUZc4O/I1eL+q9pLmx/Fny641HKGb+e15YWIBvZaoQ+x/nwTBh54Nf18JVdv8006GugsdK/VVDRITE4mOjs5wK2w2z4rHyXB3IYQQQpQ1tyPiGLLqBB9uu0BcUipN3O3Z824bXm9ZVYapl1YXflXXtK7ko/ZAg5pUWzjAk2D4e0v+zht+Xl07W2MIjUfpLNw88x70z/D3mPuwthcc/FId/p4QrRaYA6hZ9tbzFoWj1Cfpc+bMwcbGRntzcXEp9Gtql2GLL+AcHCGEEEKIEkJRFH46fouuCw/jd+MxZsYGzOxRhw1vNMNdqrCXbumV1+sN/GefiSU0f0vdPvyVuiZ5Xp141ote52X9L8XlWEsd/u4zFJQ0OPg5/NRXHeaflgIO1cHhBf3GKEqNUp+kT506laioKO3t9u3bhX7N9MJxUdKTLoQQQogy4M4Ttfd8xrYLxCal0tjdjt3vtmFkK+k9L/UeXYN7gWpvt2f/jI81HgNmNvDoClz+LW/njX0M5zar24Wx7Fp+pA9/770cjC3g5kHY9b76WA0Z6i50p9SvD2ZqaoqpadHObdL2pEuSLoQQQohSLDVNYd3JUL7YdYnYZ3PP/6+LB6+3yH7uuciBosDl30FjAB4v6Tua3EnvRa/eAco5ZnzMzBqajIXDc9Xe9Nov5746++k1kJoIFeuDS1Odhlxg9QerQ/s3D4eHl9V9Mh9d6FCpT9L1wcbi2Zx0Ge4uhBBCiFLq8NWHfL7rEpfDnwLQyM2Oea94U1WGtudPQjT8PlGd3w3Q8xtoOLxwrqUoulnKLC0Nzm1St/891P3fmr0J/t9C+Dm4tg9qdn7+eVOT4dQqdbvpuKJZdi2vKnjAmANw4FOIj1QruwuhI3pN0mNiYrh+/br2fnBwMEFBQdjb2+Pq6srUqVO5e/cua9eu1bYJCgrSHvvw4UOCgoIwMTGhTp06RR1+tmxknXQhhBBClFJX7z/ls52XOHT1IaB+7pnUsQZDm7tjKL3n+RN2DjaPgIgb/+z7fRJYOeu2h/bpfbX3Ny0Vhm1T540XRKi/ujyaiVX2Pf8W9mrRN79v1B71Gp2en3Rf2gFP74GlI3j2LViMhcnEErrO0XcUohTSa5IeEBBA+/bttfcnT54MwPDhw/H19SUsLIzQ0NAMx/j4+Gi3T58+zbp163BzcyMkJKRIYs4N7Zx0WYJNCCGEEKXEw6eJfL3vKhtPhZKmgLGhhqHN3HmnQ3Vsn40iFHmkKOqw7t0fqEO7ratA/9UQ+AME/awm7iN+h8oNC36tp/fhh57q/HCAY99A+6kFO+e5Z0Pd6/TKeXm05hPg5Aq4cwqCD0O1tjmf98R36s9GI8FIluQTZY9ek/R27dqhKEq2j/v6+mbal1P74kLWSRdCCCFEaZGQnMrKIzdZdvAGsUmpAHSt68wH3TykantBJD6F3ybChV/U+zW6QJ/las9z5QbqUl/X98PPA2DU3oJVDo958E+CbmoNidFwbBE0GJr/qunJCfD3dnXbO5uh7umsnKDBcDj5HRyel3OSfjcQbp8AA2M1SReiDCr11d31wVbWSRdCCCFECZeWprAl8A7tvzrIV3uvEpuUincVGzaNbc7yoQ0lQS+I8Auwop2aoGsModPHMHiDmqADGBrDKz9ARW+IewQ/94fYR/m7VswD8O2hJujWldX1vl1bQEo87P8o/8/h6m5IjFJ7/91aPb99y3fUxDvkCIQez75dei963T7qcH8hyiBJ0guBzbOe9PjkVBKSU/UcjRBCCCFE3hy/+Zhe3x5j8qazhEUlUMnGjIUD67N1fEuaVLXXd3gll6LAaV9Y2QEeX1eT5td3Q8t3weA/H8tNy8Grm8HWFSJuwroBkBSbt+v9O0G3qgTDf1N75Lt+Dmjg/Ca4fSp/z+XsRvVnvVcyx54VmypQ/1V1+/BXWbd5ev+fwnnNismya0LogSTphcDK1Ij0uinRMuRdCCGEECVEQnIqb60LZNCK45y/G0U5UyP+r0stDrzfjt4+lWVZtYJIjIEtY+C3dyElAWp0hnFHwTWH5cWsnGDIFjC3h7unYfPrkJqSu+v9e4i7VSV1bnv6kPlKPlD/NXV7zwfqlwd5EfsIru9Tt+sNyv1xrSapIweu74N7ZzI/fnoNpCVDlca6mYcvRAklS7AVAgMDDdbmxkTGJRMZn0wFazN9hySEEEIIkaOE5FTGrA3gyLVHGGhgcBNXJnWqSflyUrhLKzEGfF+CR9fUHu4sb27qsPV/VzAPv6AWgXt8TU1SO8yEFu/krge6fA14daOacF/7A3ZOhp6Lcq6QHvNQbf/wcuYEPV2HD+HvrXA3AM7/ovaI59aFLZCWog7Hr+CR++Psq4LXK2rBucNfwaCf/3ksJTHjsmtClGGSpBcS2/QkXealCyGEEKKYS0pJY/zPgRy59ghzY0N8X29M02oO+g6r+PljGoQFqdsPL6m3rBhbqgm7nRtYlleT4JQENWF+ZQ24NsvbdV2aqFXfNw5RK7/bVIG2/8u6bcxD+KFHzgk6qPO9W0+GA5/A/lnqEmomFrmLJ72qe1560dO1ngznNsLl3+H+RXB6tozy39sg9gFYVVSrxQtRhslw90Ji82wpEqnwLoQQQojiLDk1jbfXB3Lg8gNMjQxYNaKRJOhZubJbTZDRwCu+MHSr2qPd+j21d9ilqZpgAiTHqgn81T1w5ic1Qa/e6dnw9jwm6Ok8XoLu89Ttvz5Tz/tfuelB/7fmb4GNK0TfVdcxz41H19Sh9xpD8Oqf9+fhWOufJPzIfPWnosCJZep241Fq4TwhyjDpSS8k6WulR8Yl6TkSIYQQQoispaSmMXFjEH/8fR8TIwO+H9aIFi+U13dYxU/MQ9jxtrrd/C218nh2khPUpPdJCESGqrfyNdRe59wMb89J49EQdReOfg073oFyTlCj0z8x/tBT/XLAquLzE3RQ1zbv9BH88jocXQg+Q8Gmcs7HnNuk/nzhRShXIX/Po/V7cHEb/L0F2k2F+Ah1jrqhKTR8PX/nFKIUkSS9kMha6UIIIYQozlLTFN7ffJad58IwNtTw3ZCGtKnpqO+wMkpLVYdFWzqCWwv9xKAoarG32IdQoQ68+GHO7Y3N1OS4IOua56TDTIi+pw453zRcTcZtXGDty/9K0Hfm/vp1+6jLnt0+Dn9+BH1XZN9WUdSh6gDe+Rjqnq5iPajZVR1pcHQBJMep+71eUacHCFHGyXD3QvJPT7ok6UIIIYQoXtLSFKb8eo5tQfcwMtDw7asNaO+Rz17RwnLjL1jeGjYNgx9eVnuQ9SHoZ7iyU13ju+8KNQnXJ40GXl4M1dqrw+rXDVB70B9czHuCnn6+rnPU7XMb4U5A9m1Dj0PkLTCxglrdC/Y8Wr//7Job4NIOdbvp2IKdU4hSQpL0QpI+Jz0yXoa7CyGEEKL4UBSF6dsu8MvpOxgaaPhmsA+d6zrrO6x/PLwCPw+AH3vDg7/VfWnJcOr7oo/lSQjsnqJuvzgDnL2KPoasGJnAwB/VeGIf/tODPjwXQ9yzUrkBeD9bwzynJdnSC8bVeTn3Reay49IYqrVTq8SnpYBbS7WHXQghSXphsZGedCGEEEIUM4qiMHvH36w/GYqBBr4e4E13r4r6DksV+xh2/R8sba4uNWZgBE3fhJeXqI8HrIGk2KKLJy0Vto6DpBhwbQEt3i66a+eGqRW89gs4eoCdu5qgl6+e//N1mKlWpb9zCi78mvnx5AR1yTaAegPyf51/a/N//2xLL7oQWjInvZCkD3eXOelCCCGEKA4UReHTnZf4wf8WGg3M7e9Nr/rPKRJWFFIS4eQKODQPEqPUfbW6Q6dP1KQzLVWtAv4kGILWQZMxRROX32II9QeTctBnGRgYFs1188LKGd70U7cLGp91RWg9CQ58CvtmqX+Df/eWX/sDEqLUqvHurQt2rXRuLaH5BIh/ArVe0s05hSgFpCe9kEjhOCGEEEIUF4qi8OWeK6w6GgzAnD5e9G9YRd9BwcUd8G1T2DtDTdCdvGDYDhi8/p9eYQNDaDZe3T6+FNLSCj+28PNqsgrQ7Uu1p7q4MjDU3RcIzSeoReii74D/koyPnX1WMK7eK7q7nkYDXT6D3kvBUPoOhUgnSXohSU/SZbi7EEIIIfRtwf5rLD90A4BPensyqImrfgO6GwhrusOmoWoPeTkn6PUtjD0E1dpmbl//VTCzgYibakXwwpScAFveUOfB13oJ6r9WuNcrTtKXZAO16nr0PXU7LgKu7VW36xWgqrsQIlfkK6tCYmP+rHCcrJMuhBBCCD1acuAa3/x5DYCZPeowtJmb7k6uKLB/FoSdzf0xKYnqMHIAI3N1rnfLd8G0XPbHmJaDhiPg2CK1N92jgJXFc3LgE7VSuqUj9Fyk9vaWJXX7PluS7QT8+TH0Wa7OUU9LVgvVOdXRd4RClHqSpBeS9J706IQUUtMUDA3K2H/wQgghhNC7lUdu8tXeqwBM7ebByFZVdXuBmwfVxDk/6g1Ui5XZ5HLYfZM3wG8JhBxRvxSo6J2/6+Yk+Aj4f6tuv7wYyhWzdeOLQvqSbN+/CGfXqzUA0tdGl150IYqEJOmFJL26O0B0fDJ2liZ6jEYIIYQQZc2Gk6F8uvMSAJM71WRs23wszfU8J58ti1a7J9TulfvjnOqAU928XcumCtTtAxd+Af+l0Pe7vB3/PAlRsO1NQIEGw6FWN92evySp3BC8B6tJ+tY34dEV0BiAV399RyZEmSBJeiExNjTA0sSQ2KRUIiVJF0IIIUQR2nH2HlO3ngdgbJtqvP1iAZbmyk5kKFzdrW6/OBMca+r+Gv/V/C01Sb/wC3ScrVYk15XdUyDqtlokrsvnujtvSdVhJlzcriboANXaq9XkhRCFTgrHFSJbCzUxlwrvQgghhCgq+y/eZ/LGIBQFXmvqygfdPNAUxrzqgNWgpEHVtkWToANUbgCuzSEtBU59r7vzXtyu9hprDKDPdznPjy8rrCtBq0n/3PeWoe5CFBVJ0gtR+pB3KR4nhBBCiKJw7Pojxq8LJCVNoY9PZT7p5Vk4CXpyApz+Qd1u8obuz5+T5m+pPwNWQ1Jswc/3NBx+m6hut5oErs0Kfs7SovkEqFBHHV3gIeuYC1FUJEkvRLJWuhBCCCGKyulbTxizNoCklDQ613FiXv96GBRW4dq/t0J8BFhXgZpdC+ca2anVXU0a45+ovd8FkZYG2yeoz8W5HrT9QCchlhomFjD2MEw4DSaW+o5GiDJDkvRCJGulCyGEEKIo/H0vihFrThKXlErrGuVZ/KoPRoaF+DHv5Ar1Z+ORYFjEJY4MDKHpm+r28WVqop1fBz6B6/vA0BT6fg9GUkMoE0Pjov8bC1HGSZJeiP5ZK12SdCGEEEIUjusPYhi26iRPE1Jo7G7Hd0MbYmpkWHgXvHMa7gWCoYlaBV0ffF4DUxt4fB2u7c3fOYLWw9Gv1e2XF0MFD93FJ4QQBSBJeiHSzkmPlznpQgghhNC92xFxDFl5gsexSXhWtmbViMZYmBRyr2d6wba6fcGyfOFeKzumVtBwmLp9/Nu8H3/LH3a8rW63fh+8B+ouNiGEKCBJ0guRdk669KQLIYQQQsfuRyfw2soThEcnUKNCOdaObIq1mXHhXjT2EVz4Vd0u6oJx/9VkLGgMIfgwhJ3L/XERwbDxNUhLhjq9oP30wotRCCHyQZL0QmRrLoXjhBBCCKF7EbFJvLbyBKERcbjaW/DT6KbYWxbBfOrAtZCaBJV8oErDwr9eTmxd1CQb1LnpuZEQBesHQdxjqFgfei8HA/k4LIQoXuR/pUKkLRwnSboQQgghdCQ6IZlhq09w/UEMztZm/Dy6KU7WZoV/4dQUddkz0H8verrmE9Sf5zerS6nlJDUFNr8ODy+DVSUYvEGtXi6EEMWMJOmF6J/CcTInXQghhBAFl5KaxmjfAC7cjcbB0oSfRjfFxb6IEs2reyDqNpjbq/PRi4MqDcGlmTp0/eT3Obf9Yxrc+BOMLWDwerCuWDQxCiFEHkmSXohknXQhhBBC6NKuC+GcDInAytSItaOaUL1CuaK7eHrBuAbDwLgIeu5zq/l49WfAakiKy7rNqZVw8jt1u893UKl+kYQmhBD5IUl6Ifr3OumKoug5GiGEEEKUZIqisOLwDQBGt65G3Uo2RXfxh1fh5kHQGECjkUV33dzw6AG2bhAfAec2ZH78xgHY9T91u8NMqPNy0cYnhBB5JEl6IUpfgi0lTSE2KVXP0QghhBCiJPO78ZgLd6MxNzZkWHO3or34qZXqz5pdwa6Ir/08BobQdJy6fXwZpKX989jDq7BpBCip4D0YWk3WS4hCCJEXkqQXInNjQ0wM1V+xzEsXQgghREEsP6T2og9s7IJdUVRyT5f4FILWqdtNxhTddfPCZwiYWsOjq3B9v7ovLgLWDYDEKHBtDj0XgUaj3ziFECIXJEkvRBqNBhuZly6EEEKIAvr7XhRHrj3C0EDDqFZVi/biZzdA0lNwqA5V2xXttXPLzFqdKw/gvwRSkmDjEHgSrA6FH/gTGJnqN0YhhMglSdILmXat9DhJ0oUQQgiRPysO3wSgu1fFoqvmDqAo/wx1bzymeK8p3nSsOmc++BCsHwi3jqm9669uBMvy+o5OCCFyrRj/T1s6yFrpQgghhCiIO0/i+P1cGABj21Qr2ouHHFHXFTe2hPqDi/baeWXrCnV6qds3DqgJe/81UKG2fuMSQog8kiS9kP2zVrok6UIIIYTIu1VHg0lNU2hVvTyelYuwojv8s/a490AwK+Jr50fzCf9sd/0CanTUXyxCCJFPRvoOoLRLr/AeGS+F44QQQgiRN09ik9hw8jYAY9vmsxc9IQoMjMEkj8Pko+7A5Z3qduNiWjDuv6o0gm5z1V70xqP1HY0QQuSLXnvSDx8+TM+ePalUqRIajYZt27Y995hDhw7RsGFDzMzMqFatGsuXLy/8QAsgfbi7zEkXQgghRF79dPwW8cmp1KloTavq+ZhX/fAqLKwH82vBgc/Uiue5FbBGXbrMrRU41cn7tfWl6Vi1Cr1UchdClFB6TdJjY2Px9vZmyZIluWofHBxM9+7dad26NWfOnGHatGm88847/Prrr4Ucaf6lF46T4e5CCCGEyIuE5FR8/UIAtRddk9ekMykONg2DhEhIjIbDc2GRt5qsxz/J+diURAj8Qd0ursuuCSFEKaXX4e7dunWjW7duuW6/fPlyXF1dWbhwIQC1a9cmICCAr776in79+hVSlAVjK0uwCSGEECIffg28w+PYJCrbmtPdq2LeDlYU2PkePLwE5Zyg42zw/xbuX1CT9RPLoek4aD4ezO0yH39xO8Q+BKuK4PGSTp6PEEKI3ClRheP8/f3p3Llzhn1dunQhICCA5OSsk+DExESio6Mz3IqSjcWzwnEyJ10IIYQQuZSapvD9s2XXRreuirFhHj+ynfkRzq5T52b3WwX1X4WxR2DAj1Ch7j896wvrwV+fZ+5ZP7lC/dloJBga6+AZCSGEyK0SlaSHh4fj5OSUYZ+TkxMpKSk8evQoy2PmzJmDjY2N9ubi4lIUoWrJcHchhBBC5NXev8MJeRyHjbkxAxrl8bNL+HnY9X/q9oszoGprddvAAOq8DOOOwoC1/yTrh77MmKzfOwN3TqnF5hoM1+0TE0II8VwlKkkHMs3HUhQly/3ppk6dSlRUlPZ2+/btQo/x32S4uxBCCCHyQlEUlh+6AcCw5m5YmuZhdmJCtDoPPSUBanSGlpMytzEwUNcTH3cUXvkBKtT5V7LuDdufLWNWpxdYOWU+XgghRKEqUUuwOTs7Ex4enmHfgwcPMDIywsHBIctjTE1NMTU1LbygUlMgOQ7MrLN82EZ60oUQQgiRByeCIzh7JwoTIwOGt3DP/YGKAjsmQMRNsHGBPt+pCXl2DAygbm+o/TJc2qEm6Q8uwv0o9XEpGCeEEHpRonrSmzdvzr59+zLs27t3L40aNcLYWA/zpY4vhy/d4fC8bJvYmqtz0uOTU0lMSS2iwIQQQghRUq14Nhf9lYZVKF8uDx0NJ75TC74ZGMMrvmBhn7vj0pP1ccfU41yaQv0h6k8hhBBFTq896TExMVy/fl17Pzg4mKCgIOzt7XF1dWXq1KncvXuXtWvXAjBu3DiWLFnC5MmTGTNmDP7+/qxatYr169fr5wlYOEDSUwg+nG0TKzMjNBr1y+2o+GQqWBkWYYBCCCGEKEmuhD/lwOUHaDQwpnW13B94JwD2zlC3O38KVRrl/eIGBlC3j3oTQgihN3rtSQ8ICMDHxwcfHx8AJk+ejI+PDzNnzgQgLCyM0NBQbfuqVauya9cuDh48SP369fnkk0/45ptv9Lf8WnohlrCz2a43amCg0Q55j5Ih70IIIYTIQXovete6zriXt8zdQXERsHkEpCWrQ9ebji28AIUQQhQ6vfakt2vXTlv4LSu+vr6Z9rVt25bAwMBCjCoPrJyhfE14dBVu+WW7jqituTGRcclESvE4IYQQQmQjLCqe7UF3AXijTS570dPSYOtYiLoN9tWg1xLIppiuEEKIkqFEzUkvlqq2UX/mMORdu1a69KQLIYQQIhtrjoWQkqbQtKo9Pq52uTvo2EK4theMzNRl1cxsCjVGIYQQhU+S9IJyfzbkPackXVvhPakoIhJCCCGK1NKlS6latSpmZmY0bNiQI0eOZNt2xIgRaDSaTLe6detmaPfrr79Sp04dTE1NqVOnDlu3bi3sp6FXUfHJrDuhTvEb1/aF3B0UchQOfKJud58Hzl6FFJ0QQoiiJEl6QaUn6Q8uQszDLJvYmsta6UIIIUqnjRs3MnHiRKZPn86ZM2do3bo13bp1y1BT5t8WLVpEWFiY9nb79m3s7e155ZVXtG38/f0ZOHAgQ4cO5ezZswwdOpQBAwZw4sSJonpaRW7diVBiElOo6VSOdrUcn3/A0/vwy0hQ0sD7VfAZWvhBCiGEKBKSpBeUpQM4PfvmOiTrngNbC1krXQghROn09ddfM2rUKEaPHk3t2rVZuHAhLi4uLFu2LMv2NjY2ODs7a28BAQE8efKE119/Xdtm4cKFdOrUialTp+Lh4cHUqVPp0KEDCxcuLKJnVbQSU1JZfSwYgDfavIDmeXPK01Lh11EQcx8ca8NLX8k8dCGEKEUkSdeF58xLl550IYQQpVFSUhKnT5+mc+fOGfZ37twZPz+/XJ1j1apVdOzYETc3N+0+f3//TOfs0qVLjudMTEwkOjo6w62k2HbmLg+fJuJsbcbL3pWef8DBOWrHgEk5dR66SS6rwAshhCgRJEnXhao5z0vXFo6TJF0IIUQp8ujRI1JTU3Fycsqw38nJifDw8OceHxYWxu7duxk9enSG/eHh4Xk+55w5c7CxsdHeXFxc8vBM9CctTdEuuzaqVVVMjJ7z0ezBJTj8lbrdcxE41izkCIUQQhQ1SdJ1wa0FaAwg4gZE3c30sK0UjhNCCFGK/Xd4tqIozx+yjbrUqq2tLb179y7wOadOnUpUVJT2dvv27dwFr2cngiO48TAWK1MjBjXJxRcLxxYBCtTuCV79Cz0+IYQQRU+SdF0ws4FKPup2FvPS0+eky3B3IYQQpUn58uUxNDTM1MP94MGDTD3h/6UoCqtXr2bo0KGYmJhkeMzZ2TnP5zQ1NcXa2jrDrSTYcVb9cr+7V0WszIxzbhwZCuc3q9utJhdyZEIIIfRFknRdyWEptn+WYJMkXQghROlhYmJCw4YN2bdvX4b9+/bto0WLFjkee+jQIa5fv86oUaMyPda8efNM59y7d+9zz1nSJKaksvNcGAC96udiLrrfEkhLgWrtoHKDwg1OCCGE3kiSriv/Lh6nKBke+qe6uwx3F0IIoX/u7u58/PHH2S6TlheTJ09m5cqVrF69mkuXLjFp0iRCQ0MZN24coA5DHzZsWKbjVq1aRdOmTfH09Mz02LvvvsvevXv58ssvuXz5Ml9++SX79+9n4sSJBY63ODl05SHRCSk4WZvStJpDzo1jH0HgWnW71aTCD04IIYTeSJKuK67NwMAYom7Dk5AMD9mYq8P4ohNSSE1TsjhYCCGEKDrvvfce27dvp1q1anTq1IkNGzaQmJiYr3MNHDiQhQsX8vHHH1O/fn0OHz7Mrl27tNXaw8LCMn0ZEBUVxa+//pplLzpAixYt2LBhA2vWrKFevXr4+vqyceNGmjZtmq8Yi6vtQfcAeNm7EoYGz5nDf+I7SIlXp9dVbVsE0QkhhNAXjaIoZSprjI6OxsbGhqioKN3PV1vdDUL9oOc30HC4dndSSho1Z+wGIGhmJ2wtTLI7gxBCiDKoUN+bcnD27FlWr17N+vXrSUlJ4dVXX2XkyJE0aFDyh1Lr63eaW08Tkmn06X4SU9L4/e1WeFa2yb5x4lNYUBcSomDAj1Dn5aILVAghhE7k5X1JetJ1KZul2EyMDLA0MQRkXroQQojiw9vbm0WLFnH37l1mzZrFypUrady4Md7e3qxevZoy9j1+kfrj7/skpqTxgqMldSs950uE075qgu5QAzx6FEl8Qggh9EeSdF3KcV66rJUuhBCieElOTmbTpk28/PLLvPfeezRq1IiVK1cyYMAApk+fzmuvvabvEEut7UFqVfde9SvnvFxdSiL4f6tut3wXDOSjmxBClHZG+g6gVKnSGIzMIPYBPLoKjrW0D1mbG3M3Ml6KxwkhhNC7wMBA1qxZw/r16zE0NGTo0KEsWLAADw8PbZvOnTvTpk0bPUZZej14msCx64+AXFR1P7sBnoaBVSWoN7AIohNCCKFvkqTrkpEpuDSF4ENqb/q/knRbc1krXQghRPHQuHFjOnXqxLJly+jduzfGxpnX565Tpw6DBg3SQ3Sl3+9nw0hToL6LLW4Oltk3TEuFY4vU7RYTwEhq2gghRFkgSbquVW3zLEk/BE3GaHf/swybJOlCCCH06+bNm9rq69mxtLRkzZo1RRRR2bL9rFrVvffzetEv/QYRN8DMFhoMz7mtEEKIUkMmNula+rIoIUchLU27W5J0IYQQxcWDBw84ceJEpv0nTpwgICBADxGVHcGPYjl7OxJDAw0v1cshSVcUOLpA3W46FkzLFU2AQggh9E6SdF2r5AMmVhD/BO5f0O5OXytdhrsLIYTQt7feeovbt29n2n/37l3eeustPURUdqQXjGtZvTyOVqbZN7z5F4QFgbEFNBlbNMEJIYQoFiRJ1zVDI3Brrm7/ayk2bU96vBSOE0IIoV8XL17Mci10Hx8fLl68qIeIygZFUdgRlMuh7um96A2Gg6VDIUcmhBCiOJEkvTD8eym2Z7SF42S4uxBCCD0zNTXl/v37mfaHhYVhZCTlagrL+btR3HwUi5mxAZ3rOmff8M5p9TOEgRE0l5ENQghR1kiSXhjSk/RbfpCaAoCNeXpPuiTpQggh9KtTp05MnTqVqKgo7b7IyEimTZtGp06d9BhZ6bb9WS96x9pOlDPN4cuQo1+rP70GgK1LEUQmhBCiOJGvywuDk5daiTUhUp1PVqURNtrCcTLcXQghhH7Nnz+fNm3a4Obmho+PDwBBQUE4OTnx448/6jm60ik1TeE3bVX3ytk3fHgFLv+ubrd8twgiE0IIUdxIT3phMDAA91bqdvAhAGylcJwQQohionLlypw7d465c+dSp04dGjZsyKJFizh//jwuLtJzWxiO33zMg6eJ2FoY06amY/YNj32j/qz1ElTwKJrghBBCFCvSk15YqrZVvwkPPgyt38uwBJuiKGg0Gj0HKIQQoiyztLTkjTfe0HcYZca2M2pV9+5eFTExyqaPJOoOnNuobreaVESRCSGEKG4kSS8sVVurP0NPQEqiNklPSVOIS0rFMqe5aEIIIUQRuHjxIqGhoSQlZZyK9fLLL+spotIpITmVPRfCAejlnUNVd/9vIS0Z3FuDS+Miik4IIURxk69M8fbt22g0GqpUqQLAyZMnWbduHXXq1JFv5dM5eoClI8Q+hDsBmLu1wMTQgKTUNCLjkyVJF0IIoTc3b96kT58+nD9/Ho1Gg6IoANpRXqmpqfoMr9T56/IDniamUMnGjMbu9lk3iouA077qdquJRRWaEEKIYihfc9JfffVV/vrrLwDCw8Pp1KkTJ0+eZNq0aXz88cc6DbDE0mgyLMWm0WiwNpficUIIIfTv3XffpWrVqty/fx8LCwv+/vtvDh8+TKNGjTh48KC+wyt1tgWpQ91frl8ZA4NsprudXAHJceBcD17oUITRCSGEKG7ylaRfuHCBJk2aALBp0yY8PT3x8/Nj3bp1+Pr66jK+ku0/66WnD3mXtdKFEELok7+/Px9//DGOjo4YGBhgYGBAq1atmDNnDu+8846+wytVouKT+evyQwB61c9mqHtiDJxYrm63mqR+0S+EEKLMyleSnpycjKmpKQD79+/Xzl3z8PAgLCxMd9GVdO7P5qXfOQVJcdjKWulCCCGKgdTUVMqVKwdA+fLluXdPXRrMzc2NK1eu6DO0UmfPhTCSUtOo5WRF7YrWWTcKXAvxT8CuKtTpVbQBCiGEKHbylaTXrVuX5cuXc+TIEfbt20fXrl0BuHfvHg4ODjoNsESzrwbWVdQiMLePZ6jwLoQQQuiLp6cn586dA6Bp06bMnTuXY8eO8fHHH1OtWjU9R1e6bDujfgHycna96ClJ4L9E3W75LhgYFlFkQgghiqt8Jelffvkl3333He3atWPw4MF4e3sDsGPHDu0weEGmeek2sla6EEKIYmDGjBmkpaUB8Omnn3Lr1i1at27Nrl27+Oabb/QcXekRHpXA8eDHALycXVX3M2sh+i6UcwbvwUUYnRBCiOIqXyXG27Vrx6NHj4iOjsbOzk67/4033sDCwkJnwZUKVVvD2XUQfATbigMAiIyXwnFCCCH0p0uXLtrtatWqcfHiRSIiIrCzs9NWeBcF99vZeygKNHKzw8U+i89HyfFw+Ct1u/V7YGxWtAEKIYQolvLVkx4fH09iYqI2Qb916xYLFy7kypUrVKhQQacBlnjp89LvBVLBOBGQwnFCCCH0JyUlBSMjIy5cuJBhv729vSToOrb9rFrVvZdP5awbBKyGp2Hq1LiGw4swMiGEEMVZvpL0Xr16sXbtWgAiIyNp2rQp8+fPp3fv3ixbtkynAZZ4ti7q3HQljVpJ6gei0Ig4PQclhBCirDIyMsLNzU3WQi9k1x/EcOFuNEYGGl7yqpi5QWIMHF2gbrf9HxiZFm2AQgghiq18JemBgYG0bq32EP/yyy84OTlx69Yt1q5dK3PZsvJsXnr9lLMABIQ8IS4pRZ8RCSGEKMNmzJjB1KlTiYiI0HcopdaOZ2ujt6npiL2lSeYGJ1dA7EO1onv9V4s4OiGEEMVZvuakx8XFYWVlBcDevXvp27cvBgYGNGvWjFu3buk0wFLBvTWc9sXm/nEq23bnbmQ8J25G0N5DpgYIIYQoet988w3Xr1+nUqVKuLm5YWlpmeHxwMBAPUVWOiiKwrYgtap7lmujJ0TBsUXqdrupYGhchNEJIYQo7vKVpFevXp1t27bRp08f/vjjDyZNmgTAgwcPsLbOZg3QbCxdupR58+YRFhZG3bp1WbhwobaXPivffvstS5YsISQkBFdXV6ZPn86wYcPy8zSKzrOedE34ebrWNWHV6XgOXX0oSboQQgi96N27t75DKNXO3I4kNCIOCxNDOtVxytzAfykkREL5WuDVv8jjE0IIUbzlK0mfOXMmr776KpMmTeLFF1+kefPmgNqr7uPjk+vzbNy4kYkTJ7J06VJatmzJd999R7du3bh48SKurq6Z2i9btoypU6fy/fff07hxY06ePMmYMWOws7OjZ8+e+XkqRaNcBXCsDQ8v0dP6Bqsoz6GrD/UdlRBCiDJq1qxZ+g6hVNvxrBe9cx0nLEz+81ErLgL8v1W320+TddGFEEJkkq856f379yc0NJSAgAD++OMP7f4OHTqwYMGCXJ/n66+/ZtSoUYwePZratWuzcOFCXFxcsi0+9+OPPzJ27FgGDhxItWrVGDRoEKNGjeLLL7/Mz9MoWlXV0QF1EoMwMtAQ/CiWW49j9RyUEEIIIXQpJTWN3889G+qeVVX3Y4sg6Sk4eUHtl4s4OiGEECVBvpJ0AGdnZ3x8fLh37x5376rFUZo0aYKHh0eujk9KSuL06dN07tw5w/7OnTvj5+eX5TGJiYmYmWVcQ9Tc3JyTJ0+SnJz1smaJiYlER0dnuOnFsyHvJqHHaOimLl13WHrThRBC6IGBgQGGhobZ3kT+hTyO5VFMEhYmhrSqXj7jgzEP1IJxAC9OB4N8fwwTQghRiuXr3SEtLY2PP/4YGxsb3NzccHV1xdbWlk8++YS0tLRcnePRo0ekpqbi5JRxrpaTkxPh4eFZHtOlSxdWrlzJ6dOnURSFgIAAVq9eTXJyMo8ePcrymDlz5mBjY6O9ubi45O3J6opbS0ADj67QzV3dJUPehRBC6MPWrVvZsmWL9rZx40Y++OADKlasyIoVK/QdXokWGad2GjhamWJs+J+PWUcXQHIcVG4INbvqITohhBAlQb7mpE+fPp1Vq1bxxRdf0LJlSxRF4dixY8yePZuEhAQ+++yzXJ9Lo9FkuK8oSqZ96T788EPCw8Np1qwZiqLg5OTEiBEjmDt3brbf/E+dOpXJkydr70dHR+snUbewh4r1IOwsHcyuMpvy+N14TFJKGiZG8k26EEKIotOrV69M+/r370/dunXZuHEjo0aN0kNUpUN6km5r/p+K7VF34dQqdfvFGZDNZx0hhBAiX9nhDz/8wMqVK3nzzTepV68e3t7ejB8/nu+//x5fX99cnaN8+fIYGhpm6jV/8OBBpt71dObm5qxevZq4uDhCQkIIDQ3F3d0dKysrypcvn+UxpqamWFtbZ7jpjbs6L73K/b8oX86UuKRUAm7JGrVCCCGKh6ZNm7J//359h1GiRcarSbqNxX/WRj/yFaQmqiPrqrXXQ2RCCCFKinwl6REREVnOPffw8CAiIndJp4mJCQ0bNmTfvn0Z9u/bt48WLVrkeKyxsTFVqlTB0NCQDRs20KNHDwxKwryuZ8usaC5uZVgV9csJGfIuhBCiOIiPj2fx4sVUqVJF36GUaJFxScB/etKfhEDgWnW7/XTpRRdCCJGjfGW23t7eLFmyJNP+JUuWUK9evVyfZ/LkyaxcuZLVq1dz6dIlJk2aRGhoKOPGjQPUoer/XgP96tWr/PTTT1y7do2TJ08yaNAgLly4wOeff56fp1H0KvmAz1AAhj1ZjCGpHLoiSboQQoiiZWdnh729vfZmZ2eHlZUVq1evZt68efoOr0SLetaTbmvxryT90FxIS4EXXgT3lnqKTAghREmRrznpc+fO5aWXXmL//v00b94cjUaDn58ft2/fZteuXbk+z8CBA3n8+DEff/wxYWFheHp6smvXLtzc3AAICwsjNDRU2z41NZX58+dz5coVjI2Nad++PX5+fri7u+fnaehHx9lw6Tdsoy4zxHA/P4R34X50Av/f3n2HR1WmfRz/zqR3CIE0Qgi9gyR0KYpGwAI2QAURbCw2RN0VXVdR98W2iKvCWigWBASExRXFSK8KSO8QIJSEEDCNkH7eP04SiKGFTDLJ5Pe5rrkyc8oz95mjPHPP0wJ93a94qoiIiC28//77xeZ/sVqt1K5dm06dOlGzZk07Rlb1lRiTnrQfts40n9/wdztFJSIiVYnFMAzjWk48ceIEH3/8MXv27MEwDFq0aMFjjz3Ga6+9xtSpU20dp82kpqbi5+dHSkqK/canb/gcfniOdIsXPc+9x9/u6c7AKDvNOi8iInZXKeomB2Ovz/SpmZv5fusJ/n5rcx7p3gDmjoAd86BpP7hvZoXFISIilUtp6qVrakkHCAkJKTGL+9atW/niiy8qdZJeKUQOh9+/xDt+Ky86z2T5vmZK0kVEpMJMmzYNb29v7r333mLb58yZQ0ZGBsOGDbNTZFVf0Zh0T1c4udNM0AFueMmOUYmISFVSBWZbc0BWJ+j3LwDudV5J6r415OVfU4cGERGRUnvrrbcuuipKnTp1qs48L5VU6rkLursvK/gsW94JQa3tGJWIiFQlStLtJawD+e2GAPBi/mdsjTtt54BERKS6OHLkCBERESW2h4eHF5sLRkqvcAm2kIw9sOd/YLFCr7F2jkpERKoSJel2ZL15HBlWb1paj5C88j/2DkdERKqJOnXqsG3bthLbt27dSq1atewQkeMonDgufNv75obWA6F2UztGJCIiVU2pxqTfddddl92fnJxclliqH68AdjcfTeTON+l4aDKkPw7ete0dlYiIOLjBgwfz9NNP4+PjQ48ePQBYsWIFzzzzDIMHD7ZzdFVXXr5BamYOkZa9eMUtA4sT9PqbvcMSEZEqplRJup+f3xX3X7iuuVxZ6E2j2LH9a1pZD5P10yu43aMWdRERKV9vvvkmR44coXfv3jg7m18F8vPzefDBBzUmvQzSMnMwDHjOZY654boh4N/AvkGJiEiVU6okfdq0aeUVR7UVVNOLt32f4P30F3DbMRM6jYCwjvYOS0REHJirqyuzZ8/mzTffZMuWLXh4eNC6dWvCw8PtHVqVlpyRw3WW/XR12gVOrtDjBXuHJCIiVdA1L8EmtlO7RXe+XduTgc4r4Ifn4LHl5gzwIiIi5ahx48Y0btzY3mE4jORzOQx1jjFftL4Xamh5VRERKT1NHFcJ9GxSm7dzB5OKFyRsg41aZ15ERMrPPffcw1tvvVVi+7vvvlti7XS5emdPn+BW63rzRcdH7RuMiIhUWUrSK4Go+jXJcPHn3ZyCL0ZL34CzSfYNSkREHNaKFSu49dZbS2zv06cPK1eutENEjsF/70zcLLnsd2kGIdfZOxwREamilKRXAm7OTnRtWIsZeTdxyrspZKbAL6/aOywREXFQ6enpuLq6ltju4uJCamqqHSJyAHm51Ds0G4DV/nfaORgREanKlKRXEj2b1iYfKx+6P25u2Pw1HN1g36BERMQhtWrVitmzZ5fYPmvWLFq0aGGHiBzA3kV4ZZ4kyfDlYO2b7B2NiIhUYZo4rpLo2cRcH/2bE8G8EnkfLttnwqLn4NFlmkRORERs6pVXXuHuu+/m4MGD3HjjjQAsWbKEb775hrlz59o5uipqw2cAzMq7AW8vbzsHIyIiVZla0iuJ8Fpe1K/lSW6+wZr6T4GbH8RvhU1a9k5ERGzrjjvuYMGCBRw4cIBRo0bx3HPPcfz4cZYuXUr9+vXtHV7Vk7gHDq0kHyszcm+ihqeLvSMSEZEqTEl6JdKjoDU9Ji4fbvy7uXHJ65pETkREbO7WW29lzZo1nD17lgMHDnDXXXcxevRoIiMj7R1a1VPQir7Zsyvx1KKGh5J0ERG5dkrSK5HCLu8r9p3CiBoOga3NSeSW/dPOkYmIiCNaunQpQ4YMISQkhI8++oh+/fqxceNGe4dVtWSmwtZZACx0M2fMV0u6iIiUhZL0SqRzg1q4Olk59sc5Dp3Jgr5vmzt+/wpSjtk3OBERcQjHjh3jzTffpEGDBtx3333UrFmTnJwc5s2bx5tvvsl112npsFLZOguy0yGgKatzmgPgq5Z0EREpAyXplYiXmzMdImoCZms69btB+PWQnwNr/m3n6EREpKrr168fLVq0YNeuXXz44YecOHGCDz/80N5hVV2GUdTVnY6PkpKZC0ANj5LL24mIiFwtJemVzIVd3s0NL5h/f/8C0k7aKSoREXEEP//8M4888gjjxo3j1ltvxclJq4eUyaEVkLQPXH0w2gwi5VwOoO7uIiJSNkrSK5meTeoAsD72NJk5eRDRE+p2gNxMWKfWDhERuXarVq0iLS2NqKgoOnXqxEcffcSpU6fsHVbV9VtBK3q7+8iweJKTZwBK0kVEpGyUpFcyTQK9CfJ1JzMnn98OnQGLBXoUtKZvmApnT9s3QBERqbK6dOnCZ599Rnx8PI8//jizZs0iNDSU/Px8YmJiSEtLs3eIVUfyUdi7yHze4RGSC1rRXZ2seLioh4KIiFw7JemVjMVioUeTAOCCLu+NoyGoDeSchV8n2zE6ERFxBJ6enowYMYLVq1ezfft2nnvuOd566y3q1KnDHXfcYe/wqoaNU8HIN3u81W5KckY2AH6eLlgsFjsHJyIiVZmS9EqosMv7ysIk/cLW9F8/gXPJ9glMREQcTtOmTXnnnXc4duwYM2fOtHc4VUNOpjlXDEDHRwFIySgYj66Z3UVEpIyUpFdC1zcKwGqB/YnpHE8+Z25sdhvUbgZZqefHwImIiNiIk5MTAwYMYOHChfYOpfLbtQAyToNvXWjSF6Cou7vGo4uISFkpSa+E/DxduK6euRRbUWu61Qrdnzefr/8YstLtFJ2IiEg199un5t8OI8DJGYDkgpZ0Py2/JiIiZaQkvZIqWopt7wWz7ra6C/wbwrk/zLFwIiIiUrGObzIfTq7QfljR5uRzBWPS1d1dRETKSEl6JVWYpK85kEROXr650eoE3ceYz9d+CDnn7BSdiIhINfXb5+bflneBV0DRZq2RLiIitqIkvZJqHeqHv5craVm5bI5LPr+jzSDwqwdnE+H3L+0Wn4iISLVz9jTsmGc+L5gwrpAmjhMREVtRkl5JWa0Wrm9k/kK/bG/i+R1OLnD9aPP5mg8gN6vigxMREbnApEmTiIiIwN3dncjISFatWnXZ47Oysnj55ZcJDw/Hzc2Nhg0bMnXq+WFc06dPx2KxlHhkZmaW96Vc3uYvIS8LQq6D0MhiuwrHpKslXUREykpJeiV2U4tAAL5ef4TT6Rck4+0eAJ9gSD0OW76xU3QiIiIwe/ZsRo8ezcsvv8zmzZvp3r07ffv2JS4u7pLnDBw4kCVLljBlyhT27t3LzJkzadasWbFjfH19iY+PL/Zwd3cv78u5tPw82FDwQ0LHx8zlUS9QNCbdUxPHiYhI2ShJr8RubR1MyxBf0jJzeXfx3vM7XNyh69Pm89UTIC/HPgGKiEi1N2HCBB5++GEeeeQRmjdvzsSJEwkLC2Py5MkXPf6nn35ixYoVLFq0iJtuuon69evTsWNHunbtWuw4i8VCUFBQsYdd7VsMKXHg4W+OR/+TZHV3FxERG1GSXok5WS2Mu6MlALM3HmXbseTzOyMfAs8ASI6D7XPtEp+IiFRv2dnZbNq0iejo6GLbo6OjWbt27UXPWbhwIVFRUbzzzjuEhobSpEkTnn/+ec6dKz4Zanp6OuHh4dStW5fbbruNzZs3XzaWrKwsUlNTiz1sqnDZtfYPmj+W/4kmjhMREVtRkl7JRdX3587rQjEMeHXhTvLzDXOHqyd0fdJ8vuo9sxueiIhIBUpKSiIvL4/AwMBi2wMDA0lISLjoObGxsaxevZodO3Ywf/58Jk6cyNy5c3niiSeKjmnWrBnTp09n4cKFzJw5E3d3d7p168b+/fsvGcv48ePx8/MreoSFhdnmIgGS9kPsMrBYIWrERQ8535Ku7u4iIlI2StKrgBf7NsPL1YnNccnM33z8/I4Oj4B7DTh9AHYtsFd4IiJSzVn+ND7bMIwS2wrl5+djsViYMWMGHTt2pF+/fkyYMIHp06cXtaZ37tyZIUOG0LZtW7p37863335LkyZN+PDDDy8Zw9ixY0lJSSl6HD161HYXuKFg2bUmfaBmeIndmTl5nMsxfyz3U0u6iIiUkZL0KiDQ152nejcGYPyPe0jLLBiD7uYDnUeZz1e+B/n5dopQRESqo4CAAJycnEq0micmJpZoXS8UHBxMaGgofn5+RduaN2+OYRgcO3bsoudYrVY6dOhw2ZZ0Nzc3fH19iz1sIiv9/CStf1p2rVBqQVd3iwV83Jxt874iIlJtKUmvIoZ3q09EgBdJ6Vn8e8kFX1I6PQauPpC4C/b9aL8ARUSk2nF1dSUyMpKYmJhi22NiYkpMBFeoW7dunDhxgvT09KJt+/btw2q1Urdu3YueYxgGW7ZsITg42HbBX61jv0HOOajVCCJ6XfSQwvHofh4uWK0X70EgIiJytZSkVxFuzk784/YWAExbc5gDiQVfbjxqnv9lf8U7YBh2ilBERKqjMWPG8PnnnzN16lR2797Ns88+S1xcHCNHjgTMbugPPvhg0fH3338/tWrVYvjw4ezatYuVK1fywgsvMGLECDw8PAAYN24cixcvJjY2li1btvDwww+zZcuWojIrVMMb4dmdcOenYL3416bkc5rZXUREbMfuSfqkSZOIiIjA3d2dyMhIVq1addnjZ8yYQdu2bfH09CQ4OJjhw4dz+vTpCorWvm5oWofezeqQm28w7vudGIUJeZcnwMUT4rfAgSV2jVFERKqXQYMGMXHiRF5//XXatWvHypUrWbRoEeHh5tjt+Pj4Ymume3t7ExMTQ3JyMlFRUTzwwAPcfvvt/Pvf/y46Jjk5mccee4zmzZsTHR3N8ePHWblyJR07dqzw6wPAJxDqRl5yd+GkcVojXUREbMFiGPZrep09ezZDhw5l0qRJdOvWjU8++YTPP/+cXbt2Ua9evRLHr169mp49e/L+++9z++23c/z4cUaOHEnjxo2ZP3/+Vb1namoqfn5+pKSk2G68WgU6nHSW6PdXkp2Xz6dDI4luWbBu7OKXYd1HENYJRiw2B8YVykqH5CPmcm1/HDGf/1HwOi0eOv8FejxvnwsSEZEqXzdVRhX5mc7ZeJQX5m6jZ5PafDHCTj8kiIhIpVaaesmus5tMmDCBhx9+mEceeQSAiRMnsnjxYiZPnsz48eNLHL9+/Xrq16/P008/DUBERASPP/4477zzToXGbU/1A7x4pHsEk5Yf5I0fdtGjSW3cXZyg61Pw22dw9Ff4/mnITDWT8OQjkHGFngZL3wD/CGh1d8VchIiIiAPRGukiImJLduvunp2dzaZNm4iOji62PTo6mrVr1170nK5du3Ls2DEWLVqEYRicPHmSuXPncuutt17yfbKyskhNTS32qOqeuKERQb7uHD1zjs9WxpobfYKgfcGYv9+/NJdkO/H7+QTdvQYEt4Xmt0OXJ6Hfe3D/HOj4uLl/wRNwYnNFX4qIiEiVd36NdCXpIiJSdnZrSU9KSiIvL6/EEi2BgYEllnIp1LVrV2bMmMGgQYPIzMwkNzeXO+6447Lrpo4fP55x48bZNHZ783JzZmy/ZjwzawsfLz/AXZF1Ca3hATe8BPk54ORmruNaox7UCDefu/tdvLBGveFMLByIgVkPwKPLzLF3IiIiclWSz2UDGpMuIiK2YfeJ4yyW4kuVGIZRYluhXbt28fTTT/OPf/yDTZs28dNPP3Ho0KHLzvY6duxYUlJSih5Hjx61afz2ckfbEDrW9yczJ5//+2G3udHTH27/APq9Y04m1/x2CG5z6QQdwOoEd38OtRpD6nH4dijkZlXMRYiIiDgAtaSLiIgt2S1JDwgIwMnJqUSreWJiYonW9ULjx4+nW7duvPDCC7Rp04ZbbrmFSZMmMXXqVOLj4y96jpubG76+vsUejsBisfDaHS2xWuCH7fGsPZh07YV51ID7ZoGbnzmm/X9jtJSbiIjIVbpwnXQREZGysluS7urqSmRkJDExMcW2x8TE0LVr14uek5GRgfVPa5Q6OTkBYMdJ6u2mRYgvD3Qyl7gZt3AXuXn5115YQCO4dypYrLDla/j1PzaKUkRExLFp4jgREbElu3Z3HzNmDJ9//jlTp05l9+7dPPvss8TFxRV1Xx87diwPPvhg0fG333473333HZMnTyY2NpY1a9bw9NNP07FjR0JCQux1GXb1XHQTani6sPdkGl+vP1K2whrdBNFvms8XvwQHl5Y9QBEREQdX1N1dSbqIiNiAXZP0QYMGMXHiRF5//XXatWvHypUrWbRoEeHhZutwfHw8cXFxRcc/9NBDTJgwgY8++ohWrVpx77330rRpU7777jt7XYLd1fB05fnopgBMiNnH6fQyjifvPAra3g9GPswZDqcP2iBKERERx5WcUTBxnIcmjhMRkbKzGNWsn3hpFpGvKvLyDW7/cDW74lO5r2MY4+9qU7YCczLhi9vg2AYIaAqPxFx+8jkRESkTR6yb7K2iPtO8fIOGLy0CYOPfbyLA263c3ktERKqu0tRLdp/dXcrOyWphXP+WAMzacJRtx5LLVqCLOwz6GnxCIGkvzHsU8vPKHqiIiIiDSS0Yjw6aOE5ERGxDSbqD6FDfnwHtQjAMGPvddnLKMokcgE8QDJ4Bzu6wfzEsed02gYqIiDiQ5IIk3dvNGRcnfa0SEZGyU23iQF7q15wani7sPJHKh0sPlL3A0PbQ/2Pz+ZqJsO3bspcpIiLiQM6PR1cruoiI2IaSdAdSx9edN/q3AuDjZQfYejS57IW2vgeuf9Z8vvApOL6p7GWKiIg4iGStkS4iIjamJN3B3N42hFvbBJOXbzDm2y1k5thgLPmNr0CTPpCbCbMegLSEspcpIiLiAFK1RrqIiNiYknQH9Gb/VtT2cePgqbO8u3hv2Qu0OsFdn5kzvafFw+whkJtd9nJFRESqOK2RLiIitqYk3QHV9HLl7btbAzB1zSHWx54ue6HuvnDfTHMptmMbYPHYspcpIiJSxRUm6VojXUREbEVJuoO6sVkgg6LCMAx4fs5W0rNyy15orYZmizrAhs9hyzdlL1NERKQKSz5n9ixTS7qIiNiKknQH9vfbmhNaw4Njf5zjnz/ssk2hTW6Bni+az//3LMRvtU25IiIiVVBKYXd3TRwnIiI2oiTdgfm4u/DevW0BmPnbUZbtSbRNwT3/Bo2jzYnkZg+BjDO2KVdERKSKSdbEcSIiYmNK0h1cl4a1GNEtAoC/zdtWtJ5rmVitcNenULM+JMfBvEcg3wazyIuIiFQx59dJ15h0ERGxDSXp1cBf+zSlYW0vEtOyeOW/O21TqEdNGDQDnD3g4BJY9n+2KVdERKQKUUu6iIjYmpL0asDdxYl/DWyHk9XC91tP8L9tJ2xTcFAruOPf5vNV78GeH2xTroiISBVRuE66n8aki4iIjShJrybahdVgVK+GAPx9wQ4SUzNtU3CbgdBppPl8/khIOmCbckVERCo5wzC0TrqIiNickvRq5KkbG9MyxJfkjBxe/G47hmHYpuDoN6FeF8hKhdkPQFa6bcoVERGpxM5m55Gbb9alNTQmXUREbERJejXi6mxlwsB2uDpZWbonkW83HrVNwU4ucO908A6CU3tg4ZNQ2h8Ako/C0jfho46waoJt4hIRESlHhZPGuTpbcXfRVyoREbEN1SjVTNMgH8ZENwHg9e93cfRMhm0K9gmCgV+C1Rl2zod1H135nPw82LcYvhkEH7SBle9C0l5YMg7WXsX5IiIidpR8wRrpFovFztGIiIijUJJeDT3avQFR4TU5m53HC3O3kp9vo27v9TpBn7fM5zGvwqFVFz8uLcFMyD9oC98MhH0/gZEPET0gcrh5zM8vw+avbROXiIhIOUjRzO4iIlIOlKRXQ05WC+/d2xYPFyfWx57hr/O2cSjprG0K7/AItBkMRh7MeQhSjpvb8/Mhdjl8+yC839Ls2p5yFNxrQJcn4cmNMOx7uO196PqUec7Cp2D3/2wTl4iIiI2db0nXeHQREbEdZ3sHIPZRP8CLV25rwUvztzN30zHm/X6MG5vWYXi3CLo1qnXt3fYsFjPRPrkTTm6Hb4dCyzth4zQ4c/D8cWGdIGoEtOgPLh7Fz7/5DTj3h9mSPnc4PDAXGvQs2wWLiIjYWPI5c0y6n1rSRUTEhpSkV2P3d6pHeC1PPl8Vy7K9p1iyJ5ElexJpEujNQ10juPO6UDxcnUpfsKsnDPoKPu0FxzeZDwBXH2g7yOzSHtTq0udbLHDbB5CZAru/h1n3w7CFEBp5TdcpIiJSHi4cky4iImIrStKruW6NAujWKIDYU+l8sfYwczYdY9/JdF6av513Fu9hcId6PNglnJAaHlcu7EL+EXDPFJj9INRqCB0ehlb3gJv31Z3v5Ax3T4EZ98KhFfD1PTDiJ6jdtPQXWRqGASnHIH4LnNgCp/dDWGczfme38n1vERGpUlILxqT7KUkXEREbshg2Wyy7akhNTcXPz4+UlBR8fX3tHU6lk5qZw7cbjvLFusMcPXMOMMew92kZxPBu9YkMr1m6rvD5+WAtw9QHWWnwZX+zNd4nBB5eDDXqXXt5FzIMSI6D+K3nk/L4LZBxuuSxNepB71eh5V1lux4RkYtQ3WR7FfGZ/m3uNmZvPMrz0U148sbG5fIeIiLiGEpTL6klXYrxdXfhke4NGN4tgiW7TzJtzWHWxZ7mh+3x/LA9ntahfrzYtxndGgVcXYFlTWjdfMwx6dP6mmuwfznAbFH3rlP6snLOwYElcHxjQUK+Fc6duUjMzlC7OYS0Bb96sGmamczPe9hcWi76Tah/fdmuS0REqrzzY9I1cZyIiNiOknS5KCerheiWQUS3DGJ3fCrT1xxm/pbjbD+ewpApv/LczU0Y1asRVmsFrAvr6Q9D58PUW8zJ576+Cx76Adz9rnxufj4cWQPbZsGuhZCVWny/1QXqNIeQdhBc8AhsCS7u54/p+iSsnwSrJ8KJzTD9VmjSF256Deo0s9lliohI1aIx6SIiUh7U3V2u2pmz2bz94x5mbzwKwI3N6vD+wHYVN6vt6YMwtQ+cTYR6XWHIPHOSuos5tc9MzLd9ay71VsivHjS84XxSHtjy6seap5+CFW+ZM9UbeWCxQvsHoddL4BNY1qsTkWpMdZPtVcRn2mfiSvYkpPHVwx3p3rh2ubyHiIg4htLUS0rSpdS+3XCUV/67g6zcfML8PZj8QCStQq+iVdsWErbDtFshKwUa3wKDZ4BTwY8EZ5NgxzzYOgtO/H7+HDdfaDnAXL+9Xpeyd8FP2g+/vAZ7CtZwd/Ey13bv+tTVT4wnInIB1U22VxGfaef/W0JCaibfP3k9retWUD0oIiJVkpL0y9AXIdvYcTyFv8zYxNEz53B1tvJG/5YM6mCjCd2uJG69OTY99xy0vhea3QpbZ8OBGMjPNY+xOkOjm6DNIGjat/ha7LZyZB3EvALHNpivvQOh11i4bqg5O72IyFVS3WR7FfGZNnvlRzJz8ln11xsI879Ezy4RERGUpF+WvgjZTkpGDmO+3cKSPYkADIoKY1z/lri7XMPa6qW1PwZmDj6flBcKuQ7a3get7gavq5zcriwMA3YtgF/GwR+HzG31u8P9s8HVq/zfX0Qcguom2yvvzzQzJ49mr/wEwLbXovF117h0ERG5tNLUS1pLSq6Zn6cLnz0YxQu3NMVqgdkbj3L35LXEnc4o/zdvfDPc+Yk58ZtfGHR/Dp7YAI8th06PV0yCDmCxQMs74YnfoM/b4OoNh1fBN4Mg+2zFxCAiIhWucI10J6sFHzf1nhIREdtRki5lYrVaeOKGRnw5ohP+Xq7sPJHKbR+uYumek+X/5q3vgb8dhme2Qe9/QO0m5f+el+LsCp1HmrPQu/qYifqMgUrURUQcVHJBku7r7ozFUgErnYiISLWhJF1s4vrGAfzvqetpF1aD1MxcRkzfyL9+3ktefjmPpnDzLvtEcLYU1hGGfmcm6kdWO1ainpMJJ3fCzgWw4l347jGYcgv89pm9IxMRqXBFy69pjXQREbEx9c8Smwmp4cG3j3fhnz/s4ot1R/hw6QG2HE1mwsB21Pa5ymXOHEFYR7NF/as7CxL1e+GBOVVjjLphQPpJcwb70/vNv0n7IWkfJMcBF/nR5fhGiOgBtZtWeLgiIvaSnJENgJ/WSBcRERtTki425epsZVz/VlxXryZjv9vOqv1J9Hx3GQ91rc9jPRpUnxaHsA5mov71XXBkjZmo3/9txS/RFr/VTK4zUyEr9fzfC59f+DczxZw1/1Lc/SCgCdRqDAGNIXYZHFoJP/4Vhi4wx+iLiFQDhd3da3gqSRcREdtSki7lYsB1oTQP9uWFuVvZdiyFScsP8tW6IzzcPYIR10dUj1lwCxP1r+48n6g/MKdiEvWTO+Hnv8PBpaU/12KFGuFmMh5QkIzXamy+9goonoi3HAAfd4bY5bD7e2hxh62uQESkUksp7O6ulnQREbExJelSbpoG+fDfJ7oRs+skE2L2sSchjYm/7GfamsM81qMBD3Wtj5ejz4hbN+p8oh63tvwT9bSTsOyfsPkrMPLN2e9DrgN3X3DzBTefgud+57f9eZ9vKDhf5fAE/wbQ9SlY9R4sftlcm95VawWLiONLPmd2d682PcRERKTCOHiGJPZmsViIbhnETc0D+XFHAu//so8Diem8u3gvU1cfYmTPhgzpHI6HawWsrW4vFZGo55yDdR/D6vchO93c1qI/3PSamUiXp+5jYOssSImDNR/ADWPL9/1ERCqBwonjNCZdRERsze7TYk+aNImIiAjc3d2JjIxk1apVlzz2oYcewmKxlHi0bNmyAiOWa2G1Wri1TTCLR/fg/UFtqV/Lk9Nns/nnot30eHcZ09ccIis3r1RlGoZBZk7pzrGbulHmmG0334JE/R7ISi97ufn5sO1b+DAKlr5hJuihkTD8Jxj4Zfkn6GBOiBf9hvl8zUT440j5v6eIiJ1pTLqIiJQXu7akz549m9GjRzNp0iS6devGJ598Qt++fdm1axf16tUrcfwHH3zAW2+9VfQ6NzeXtm3bcu+991Zk2FIGTlYLd15Xl9vbhPDd78f5YMl+jief47Xvd/HJylieuKERjet480dGDn9kZPNHRjbJGTn8cTabPzJySL5gW/K5HPINg7vb1+Xtu9vgZK3kk5bVjTQT9a/uhLh1ZqL+wByzm/m1OLLW7GJ+4nfztV+Y2XLe8q6KX5au5Z2wcaq5Pvzil2DwjIp9fxGRCpZ6Ti3pIiJSPiyGYZTzQtaX1qlTJ9q3b8/kyZOLtjVv3pwBAwYwfvz4K56/YMEC7rrrLg4dOkR4ePhVvWdqaip+fn6kpKTg6+t7zbGLbWTn5vPtxqN8tPQACamZ11zO4A5hjL+rNZaqMLv48U3w5Z2QlQJhneHO/4BHTXD1Bqer+N3sTCzEvAq7F5qvXb3NLuedR4GLR/nGfjknd8F/rgcjz+ze3/DGspdpGJoxXqoF1U22V96f6e0frmb78RSmPhTFjc0CbV6+iIg4ltLUS3ZrSc/OzmbTpk28+OKLxbZHR0ezdu3aqypjypQp3HTTTZdN0LOyssjKyip6nZqaem0BS7lwdbYypHM490TWZeZvccz4NY78fIMani7U9HSlhqcrNT1dqOnlesE2829NT1c2HD7DM7M2M2vDUfw8XHixb7PKn6iHRsKD881E/eh6+He78/ucPcyx6m4+ZvJ94V83b8jNhu1zID/HnIW9/TC44SXwrmO3yykS2AI6Pgq//gd+/BuMXAPOZZhQ6fAamPewue78nZ+Ci7vtYhURKaPCieP8PDRxnIiI2JbdkvSkpCTy8vIIDCz+63NgYCAJCQlXPD8+Pp4ff/yRb7755rLHjR8/nnHjxpUpVil/7i5ODO8WwfBuEaU67/a2IZzLzuOv87bxycpY/DxdGNWrUTlFaUOFifqCUWbLeJ75ZY/cc+bj7KnLn9+wN0S/aSbGlUmvsbB9LiTtg98+MWd+vxYHl8LM+83PYtd/ISsNBn9j354CIiIXKJw4TmPSRUTE1uw+u/ufWz0Nw7iqltDp06dTo0YNBgwYcNnjxo4dy5gxY4pep6amEhYWdk2xSuU0sEMYKedy+Oei3bzz01583V0Y0vnqhj/YVWgkPPGr+Tw325z0LSvVnFAuO73gb5qZoBZuy8mA+t2hUW/7xn4pHjXgpldh4VOw/G1ofS/4BJWujL0/wbcPQl4W1OsC8VsLkvb74L6ZStRFxO5y8/JJy8wFtE66iIjYnt2S9ICAAJycnEq0micmJpZoXf8zwzCYOnUqQ4cOxdX18t3M3NzccHO7yjWfpcp6tEcDUs7l8NGyA7zy3x34erhwR9sQe4d19ZxdwdkfPP3tHUnZtRsCG6eZE9r98po55v5q7VxgdnHPz4Vmt8E90+DYBnPZuthlStRFpFJILUjQQRPHiYiI7dltCTZXV1ciIyOJiYkptj0mJoauXbte9twVK1Zw4MABHn744fIMUaqY56KbMLRzOIYBY2ZvYdmeRHuHVD1ZrdDvXfP51pkQ9+vVnbftW5g73EzQW90D9043f7yo3w2GzAUXr4JEfTBkZ5Rb+CIiV5KcYQ5R8nFzxtnJ7qvZioiIg7FrzTJmzBg+//xzpk6dyu7du3n22WeJi4tj5MiRgNlV/cEHHyxx3pQpU+jUqROtWrWq6JClErNYLIy7oyX924WQm28w8utN/HbojL3Dqp7qRpkt6gA/vgD5V1jP/vcv4bvHwMg3z7vrU3C6oHUqvCsMmWdOohe7XIm6iNhV4RrpfhqPLiIi5cCuSfqgQYOYOHEir7/+Ou3atWPlypUsWrSoaLb2+Ph44uLiip2TkpLCvHnz1IouF2W1Wnjv3rbc2KwOWbn5PDx9AzuOp9g7rOrpplfBzdccU/77l5c+7tdPzTHsGBD1MNzxIVidSh4X3uV8on5oBcwcVPGJ+plYSFcPDZHqLuWcJo0TEZHyY9d10u1Ba9FWD5k5eTw49Td+O3SGWl6ufDuyCw1re9s7rOpn3SRYPBY8/OGpTSXH3K/5AGL+YT7v8qQ5Y/2VJo6MWw9f321OpFe/O9w/G1y9yif+Qvn5sOo9WPZ/gAHBbaFxNDS62ew1cLEfFURKQXWT7ZXnZ7pg83FGz95Ct0a1mPFIZ5uWLSIijqk09ZIGUolDcndx4vNhUbQK9eX02WyGfv4rJ5LP2Tus6qfjo1C7GZw7U5DgFjAMc/b3wgS9+/NXl6AD1OsMQ74DVx84vAq+GQTZZ8snfjBn1/92KCz7J1Dwm2b8Vlj5LkyNhncbwtwRsGUmpF9h6TwRcQiFY9JraI10EREpB0rSxWH5urvwxfCONKjtxYmUTIZM+ZWk9Cx7h1W9OLlA33fM5xunQMJ2M0FfMg6WFyTtN/4der9ydQl6oXqdYGgFJOpJB+Cz3rDnf+DkanbFf24fDJgMLe8Cdz849wfsmAcLRsJ7jeDTXuYPEkc3XHksvohUSRqTLiIi5UlJuji0Wt5ufPVwJ0L83Ik9dZZhU38jNTPH3mFVLw16Qov+5qRwi/4KP42F1e+b+275P+jxwrWVG9axeKI+Y6BtE/V9i+GzGyFpL/gEw0OLoP2D4BMI7e6He6fBC7EwYjF0fw6C2pjnndgMK96GKTfBu41gw+e2i6k6++MIfNIT5jwEudn2jkaqueSMgjHpWn5NRETKgZJ0cXihNTz46pFO1PJyZeeJVB747FdOpmbaO6zqJfqf4OwBcWvh18nmtlv/BV2eKFu5YR1h6Hxzgrojq8311LPSy1Zmfj6seNdsnc9KgbDO8NgKCOtQ8lgnZ7P7fe9/wMhV8Nxe6P8xtBgAbn5mN/9Ff4U/Dpctpuou7SR8NQDit8DO+fDDs2aPDBE70cRxIiJSnpSkS7XQsLY3X4zoSE1PF7YfT6H/R2s063tFqhEG3ccUvLCYiWyHR2xTdliHCxL1NfBxJ1jyhtlVvbSy0mDOg7DsTYpmmx/2vdl6fjV8guC6ITDwC/hrLDS4AYw8WPWv0scipnPJ8PVd5sz63kFgscLmr2Htv+0dmVRjGpMuIiLlSUm6VButQv1Y8EQ3Gtb2IiE1k3v/s47FOxPsHVb10W003DTO7KJ+3RDbll03CoYuAK/akHrMnIn9o0j4/GbYONVM9K7k9EH4/CbY/f358ee3TQDna/wS7uQMvcaaz7d8A8lxlz++IiTth7NJ9o7i6mWfhW8Gwskd4FUHhi+CW8ab+2JehT0/2Dc+qbY0Jl1ERMqTknSpVsJrefHdqG50bxzAuZw8Rn69icnLD1LNViK0D2dXuH40NLyxfMqvGwmjd8A908zl0SxWOPYb/O9ZeK+JOZZ538+Ql1vy3P0x8OkNcGpP8fHnZVWvE0T0hPxcWDWh7OVdK8MwW/M/ioIP2sL6yZV/UrvcbJg9FI7+ak7QN3Q+1GoInR43ezhgwLxHIH6bvSMVYNKkSURERODu7k5kZCSrVq267PFZWVm8/PLLhIeH4+bmRsOGDZk6dWqxY+bNm0eLFi1wc3OjRYsWzJ8/vzwvoVRSNCZdRETKkZJ0qXb8PFyY9lAHHuwSjmHA2z/t4YW528jKreRJi1yZizu0ugsemANj9pjLutVpCXlZ5ljmb+6F91vA4pchYYeZvK58r2AsewqEdYLHll98/Pm16vWi+Xfz15ByzHblXq3cLJg/Epa8br7OToefXjQnxTuxueLjuRr5efDdo3BwCbh4wgNzIaiVuc9igb5vm0MJcjJg5mBIU48Ye5o9ezajR4/m5ZdfZvPmzXTv3p2+ffsSF3fp3iMDBw5kyZIlTJkyhb179zJz5kyaNWtWtH/dunUMGjSIoUOHsnXrVoYOHcrAgQP59ddfK+KSruj8mHR1dxcREduzGNWsCbE0i8iL4/ti7WHGfb+TfAM61vfnP0Mj8ffSly6HYhiQsM1cx3z7HMi4oLu3TwiknTCfR42APm9fe/f2y5l+mzkDfYdHzAnzKkr6KZj9gNkabXEyk1snV4h5BTJTzN4GnUbCDS+Bm0/FxXU5hgHfPwO/fwFWF7h/NjTqXfK4c8kw5WZI2gch7c2u8C4eFR6urVTluqlTp060b9+eyZMnF21r3rw5AwYMYPz48SWO/+mnnxg8eDCxsbH4+/tftMxBgwaRmprKjz/+WLStT58+1KxZk5kzZ15VXOX1mRqGQaOXfyQv32D92N4E+bnbrGwREXFcpamX1JIu1dqwrvWZ+lAHfNyc+e3wGQZ8vIYDiWn2DktsyWKB4LbQ9y14bg8MngnNbzcTwLQTZtJ6+7/htvfLJ0EH6Pk38+/vX0LqifJ5jz87uQs+v9FM0N38YMhc6PgoRA6DJzdC64HmsnjrJ5mT7VWW8d2/vGom6BYr3P35xRN0AI8aZgLv4Q8nfjd7C+TnV2ioAtnZ2WzatIno6Ohi26Ojo1m7du1Fz1m4cCFRUVG88847hIaG0qRJE55//nnOnTtXdMy6detKlHnLLbdcskwwu9CnpqYWe5SH9Kxc8vLN9g3N7i4iIuVBSbpUe72a1uG7UV0J8/cg7kwGd05ay8p9p+wdlpQHJxdo1g8GfQ3P74P+k+DRpWbiWp7qXw/1ukJeNqyeWL7vBebY+ynR5mR1NSPgkV+KzwXgXQfu/swc510zAlKPw6z7YdYD9umSX2j1+7DmA/P5bROh5YDLH+/fwLyXVhfYtQCWl2y1lfKVlJREXl4egYHFV0AIDAwkIeHiwxBiY2NZvXo1O3bsYP78+UycOJG5c+fyxBPnl2RMSEgoVZkA48ePx8/Pr+gRFhZWhiu7tMI10t2crbi7OJXLe4iISPWmJF0EaBzow4JR3ehQvyZpmbkMn76BL9cdtndYUp48/eG6ByCodfm/l8UCPf9qPt80vfzGUBsGrPsYZg6C7DSo3938EaJ2k4sf3/BGGLUOuj8HVmfY8z+zVd0eE8ttnAq/vGY+v/mNq//hpH43uH2i+XzlO7Dt2/KITq7AYrEUe20YRolthfLz87FYLMyYMYOOHTvSr18/JkyYwPTp04u1ppemTICxY8eSkpJS9Dh69GgZrujStEa6iIiUNyXpIgVqebvx9SOduKt9KHn5Bv/4705e/e8OTSgnttGglzkxXV7W+dZiW8rNNsdyL37J7Mbe/kEY8p35Y8TluHhA73/AyNUQ1tk+E8vtmAf/G2M+v34MdHu6dOdfN8Rc4g/gv09AXOWYXKw6CAgIwMnJqUQLd2JiYomW8ELBwcGEhobi5+dXtK158+YYhsGxY2ZPjqCgoFKVCeDm5oavr2+xR3lILprZXfOXiIhI+XC2dwAilYmbsxP/urctjep4885Pe/li3RG+WHcEHzdnani54O/pSk0v16K/NT1dir3293Klnr+nukBKSYWt6V/fbbYadxsNPpdOOEol4wx8+6A5OR0WuOWf0HmU+Z5Xq05zGP4jbP4SYv4B8VvMRL1pPwhtD0FtzbH93rVtE3Oh/THw3WOAYU7e1/sf11ZO71fh9AGzN8Cs+80eBDXDbRqqlOTq6kpkZCQxMTHceeedRdtjYmLo37//Rc/p1q0bc+bMIT09HW9vbwD27duH1Wqlbt26AHTp0oWYmBieffbZovN+/vlnunbtWo5Xc3WSz2UDWiNdRETKj5J0kT+xWCyM6tWIBgHejP1uG39k5JCWlUtaVi5Hz5y74vlOVguN63jTKtSP1qF+tAr1o0WwLx6uStyrvYa9ITQKjm+EdR+aS8SVVdJ++GYgnIkFV2+4Zyo0ueXayrJaIfIhMzFf/JI5G/6e/5mPQj4hENzGTNiD20JQG/CrW7ofBADyciBunbkWen4utLob+r1X+nIujP2uT2FqH3M2/5mDYcRicK9aM6VXRWPGjGHo0KFERUXRpUsXPv30U+Li4hg5ciRgdkM/fvw4X375JQD3338/b7zxBsOHD2fcuHEkJSXxwgsvMGLECDw8zBn6n3nmGXr06MHbb79N//79+e9//8svv/zC6tWr7XadhZK1RrqIiJQzJekil9CnVRDRLQJJzczhzNls/sjI4Y+z2ZzJyOaPi77O5lRaFqmZuexJSGNPQhpzN5ldN60WaHSRxN3LTf8LVisWiznT+zf3woYp0PWZsrVMH1wGc4aZy6n51YP7Z0Fgy7LH6V3HnFm901/gyBqI32o+Th8wZ8RPOwH7fjp/vId/QdLeBtx8ISvt/CM7veB56gXb0yH3gh+8GkfDnZ+AtYw/ZLl6wX2zzB4Aibtg3sPm67KWK5c1aNAgTp8+zeuvv058fDytWrVi0aJFhIebPRni4+OLrZnu7e1NTEwMTz31FFFRUdSqVYuBAwfy5pvnf7Tq2rUrs2bN4u9//zuvvPIKDRs2ZPbs2XTq1KnCr+/PNCZdRETKm9ZJF7EhwzBISM1k+7EUdhxPYceJVLYfT+FUWlaJYy0WaFjbm4a1vajn70k9f0/qFvwNreGhLvOOyjDgsxvM8d7dRsPN466tnM0zYOFTYOSZY90HzbB9V/Q/y0qHkzvOJ+3x2+DUbrMl/Fo16Wu2/rt62i7O47/DtH7mDwGt7jHXga/V0HbllwPVTbZXXp/p/y3azacrY3msRwNe6tfcZuWKiIhjK029pGY8ERuyWCwE+3kQ7OdBdMugou0nCxL37ccLk/cUTqZmcSAxnQOJ6RctK8jXvSBx96CevydhNT2pV8uT1qF+SuCrssLW9JmD4bfPoOvT4FXr6s83DFj1HiwtaHVsPRDu+BBc3Msn3gu5eUO9zuajUE6m2WqdsM1M2vOyzdZ0Nx/zeDefgoev2R2/6HXBw6kcWiND28Od/zF7GeyYaz4a9oYOj5hDAdSyLmWQnFEwJl3d3UVEpJwoSRepAIG+7gS2cOemFucnCktMy2TniVSOJJ0l7sw54s5kcOyPDOLOZJCRnUdCaiYJqZn8drh4WQHebjzWI4IHOoWru3xV1aSPOZY7YRus//jqJ0vLz4NFL8DGKebr6581J0y71nHctuDibibFoe3tF8PFtBwAHgth3Ufm5HQHl5gPvzCIGg7XPVj+PQ/EIRWOSVeSLiIi5UXf8EXspI6PO3WaukPT4tsNw+DM2WzizmQUJO7niDudwdE/Mth3Mo2k9Cz+b9EeJi0/yMPdIniwa319WaxqClvTZz8Av34KXZ688lJpOedg3iMFk7hZoO870OmxCgm3ymrQ03yciYWN02DzV5ByFJa8DsvGm4l8h0fM4QL2/KFDqpRkjUkXEZFypiRdpJKxWCzU8najlrcb19WrWWxfTl4+8zcfZ9KyAxw+ncG/Yvbx6cpYhnWtz4jrI/D30rq9VUazWyGwNZzcDusnw40vX/rYjDNm9/ijv4KTG9z9GbS4+PJWchH+DSD6DXNs+s4FsOFzc4b97XPMR2Br6PAwtL7X7KIvchkpWiddRETKmSaOE6mC8vIN/rftBB8vO8C+k+aYdg8XJ4Z0rsej3RtQx7cCxidL2e36r7m+uZsvjN4GHjVLHpMcZ66tnrQP3P3M2crD7b9WdJV3YrOZrG+fC7mZ5jY3X3OWeiPfHFqQn2tOzJefW/D6wm0Fz53d4ZktZQ5HdZPtlddn2un/fuFkahb/e+p6WoX62axcERFxbJo4TsTBOVkt9G8Xyu1tQvh510k+WrafHcdT+WzVIb5Yd4TBHcJ4vGdDQmt42DtUuZxmt0OdFubEa79+Ar1eLL4/YTt8fQ+kJ4BvKAyZB3U0m7RNhFwH/T+Gm9+ALd+Y4/zPxMLhVaUrx1n/j1U3GpMuIiLlTUm6SBVmtVro0yqIW1oGsnzfKT5aeoBNR/7gy3VH+ObXOO5uX5ehXcJpGeKLRWNuKx+rFXq8AHOHw/pJ0PkvZms5QOwKmD3EXF+8Tgt4YC74hdo3Xkfk6Q9dn4TOoyBuLaQlmLO/W5zA6mw+L/Haufg2qTYyc/LIys0HNCZdRETKj75diDgAi8XCDU3r0KtJbdbFnuajpQdYe/A0szceZfbGo4T4mTPL39wikE4RtXB1tto7ZCnUYgDUfhtO7TEnkev5gtkFe/5IyM+B8Oth8AzwqGHvSB2b1Qr1r7d3FFLJpRRMGudkteCt1TVERKScqIYRcSAWi4WuDQPo2jCATUf+YMrqWJbtOcWJlEy+XHeEL9cdwcfdmV5N63Bzi0B6Na2Nr7tag+yqsDV93sPmcmEWzq+B3mIA3PlJxayBLiJXdGFXd/VOEhGR8qIkXcRBRYbXJDI8ksycPNYcSCJm10l+2Z1IUnoW3289wfdbT+DiZKFzg1rc1NxsZQ+5yBj2/HyDtKxcUs/lkPKnR3pmLtfVq0FU/SssHyaX1/JOWP4WnN5/PkHv9Be45f/MJF5EKoXkjGwAamg8uoiIlCMl6SIOzt3Fid7NA+ndPJD8fIPNR5OJ2XWSmF0JHDx1llX7k1i1P4lXF+6kZYgvNT1diyXiaZk55F9hDYgO9WvyxA2N6NmktlqXroXVyWxNn1+w7vnNb0DXp7R2t0glU7hGup/Go4uISDlSki5SjVitloIW9pq82LcZsafSCxL2k2yK+4OdJ1Ivea6bsxU/D5diD6vVwoq9p9hw+A8emraBliG+PHFDI25pGYSTVQlmqbS+F3LOQo1waNTb3tGIyEWcXyNdSbqIiJQfJeki1ViD2t483tObx3s2JCk9izUHksjNM6jhWTwZ9/Vwwd3F6aJlJKRk8vmqWGb8GsfOE6mMmvE7DWp78ZeeDRlwXSguTuqufVWsVogaYe8oROQyks8VdHf3dLVzJCIi4siUpIsIAAHebvRvV/olvoL83Pn7bS0YdUMjpq89zPQ1h4g9dZYX5m5j4i/7eaxHAwZ1CLtkki8iUlVojXQREakIauISEZvw93JlzM1NWPPijYzt24wAbzeOJ5/j1YU7uf7tpUxafoDUzBx7hykics0Kx6RrjXQRESlPakkXEZvycXfh8Z4NGda1PnM2HeM/yw9yPPkc7/y0l8nLD9KlQS1q+7gVPer4uBc9D/B2xc1ZLe4iUjkVrpOuMekiIlKelKSLSLlwd3FiaOdwBncI4/utJ5i0/CAHEtP5edfJy57n5+FiJu3eZuLeq2lt+rcL1UR0ImJ3RRPHaUy6iIiUIyXpIlKuXJys3NW+LgPahbL24GkOnz5LYloWpwof6VkkFTzPzssvWvrtQGI6AAu3nuDTlbH8rU8zejXVEm8iYj+FE8dpTLqIiJQnJekiUiGsVgvXNw7g+sYBF91vGAYp53KKJe8HE9OZvvYwexLSGD59A50i/BnbrzntwmpUbPAiIlwwcZzGpIuISDlSki4ilYLFYqGGpys1PF1pHOhTtH3E9RFMXn6QaWsP8+uhMwz4eA39Wgfxwi3NiAjwsmPEIlLdaJ10ERGpCHaf3X3SpElERETg7u5OZGQkq1atuuzxWVlZvPzyy4SHh+Pm5kbDhg2ZOnVqBUUrIhWthqcrY/s1Z9nzvbgnsi4WCyzansBNE1bw9wXbSUzLtHeIIlIN5OTlk5aVC2hMuoiIlC+7JumzZ89m9OjRvPzyy2zevJnu3bvTt29f4uLiLnnOwIEDWbJkCVOmTGHv3r3MnDmTZs2aVWDUImIPoTU8eO/etvz4THdubFaHvHyDr9fH0evd5UyI2Ud6wZdnEZHykHru/BKSvu7qiCgiIuXHYhiGYa8379SpE+3bt2fy5MlF25o3b86AAQMYP358ieN/+uknBg8eTGxsLP7+/tf0nqmpqfj5+ZGSkoKvr+81xy4i9rU+9jTjf9zD1qPJANTycuXp3o25r2M9XJ3t3klIpFRUN9merT/Tg6fS6f2vFfi4O7P9tVtsEKGIiFQnpamX7PZNNjs7m02bNhEdHV1se3R0NGvXrr3oOQsXLiQqKop33nmH0NBQmjRpwvPPP8+5c+cu+T5ZWVmkpqYWe4hI1de5QS0WjOrKpAfaExHgxemz2by6cCeRb8QwasYmvt14lMRU23SFP3M2m592xDP+x93M+i2OzJw8m5QrIlVHctHyaxqPLiIi5ctu/bWSkpLIy8sjMDCw2PbAwEASEhIuek5sbCyrV6/G3d2d+fPnk5SUxKhRozhz5swlx6WPHz+ecePG2Tx+EbE/i8VCv9bB3NwikNkbjvLh0v2cTM1i0fYEFm03/x1pFerLDU3rcEOzOrStW+Oq1ltPSs/i19gz/HroNOtjT7PvZHqx/e/9vJeHutZnSOdwjU0VqSYKu7vX8ND/8yIiUr7sPqjqz2seG4ZxyXWQ8/PzsVgszJgxAz8/PwAmTJjAPffcw8cff4yHh0eJc8aOHcuYMWOKXqemphIWFmbDKxARe3NxsjKkczj3dazH9uMpLN2TyPK9iWw7lsKO46nsOJ7Kh0sPUNPThZ5NanNDszr0aFybml7ml+3EtEx+jT3D+tjT/HroTNEa7RdqGuhDu7AarD6QxPHkc7z38z4mLT/I4A71eLh7BKE1Sv77IyKOo3CNdLWki4hIebNbkh4QEICTk1OJVvPExMQSreuFgoODCQ0NLUrQwRzDbhgGx44do3HjxiXOcXNzw83NzbbBi0il5GS10C6sBu3CajDm5iacSstixb5TLNubyMp9p/gjI4cFW06wYMsJrBZoU7cGqedyiE06W6KsZkE+dG5Qi84N/OkYUQv/goQ+Jy+fRdvj+c+KWHbHpzJ1zSG+WHeY29sE81iPhrQI0XhiEUdU2N3dV8uviYhIObNbku7q6kpkZCQxMTHceeedRdtjYmLo37//Rc/p1q0bc+bMIT09HW9vbwD27duH1Wqlbt26FRK3iFQdtX3cuCeyLvdE1iU3L5/f45JZtjeRZXsS2ZOQxpaCSecsFmge5EvnBrXo1MCfjvX9i1rZ/8zFyUr/dqHc0TaEVfuT+GTlQdYcOF2U/PdoUpuRPRrQpWGtS/YKEpGqJ1lrpIuISAWxa3f3MWPGMHToUKKioujSpQuffvopcXFxjBw5EjC7qh8/fpwvv/wSgPvvv5833niD4cOHM27cOJKSknjhhRcYMWLERbu6i4gUcnay0jHCn44R/vytTzPiU86x9sBpfD1c6FjfH79SdmG1WCz0aFKbHk1qs/1YCp+sPMii7fGs3HeKlftO0TrUj8d6NCCqfk0yc/LJys0jMyefzJw8snLNv4XPs3LMfdl5+XRtWIvr6tUsp09BRK5VyjlNHCciIhXDrkn6oEGDOH36NK+//jrx8fG0atWKRYsWER4eDkB8fHyxNdO9vb2JiYnhqaeeIioqilq1ajFw4EDefPNNe12CiFRRwX4e3B1pmx44rev68dH97Yk7ncGU1bHM3niU7cdTeGrm5lKXZbHAE70aMfqmxjg7aSk5kcoiOaNgTLomjhMRkXJm13XS7UFr0YpIeTtzNpsv1x3m6/VHSDmXg5uzE+4uVtycnXBzseJ+wWt3FyvuLk64OVtJy8xlyZ5EAKLCa/LBfddpQrpqQnWT7dn6M31o2m8s33uKd+5pw8AoTUArIiKlU5p6ye6zu4uIOBp/L1dG39SE0Tc1KfW5/9t2grHztrPxyB/0nbiSd+5pS59WQeUQpYiUhsaki4hIRVFfShGRSuS2NiEseqY7bcNqkJqZy8ivN/HKgh1k5uTZOzSRaq1onXRPdXcXEZHypSRdRKSSCfP3ZO7ILjzeswEAX60/woCP13AgMe2aykvJyOHbDUd5/KuNvLJgB/tPXls5ItVZsiaOExGRCqLu7iIilZCLk5WxfZvTrWEAY77dwp6ENG7/cA3j7mjJvVF1r7i8W1pmDr/sPsn3W+NZtf8UOXnnpx/5av0RujcOYES3CHo2qY3VWval4o6eyWD5vlMEeLnSu3kgrs76DVgcR36+ccHEcUrSRUSkfClJFxGpxHo0qc2iZ7rz3LdbWbU/ib/O28bqA0n8885W+LgXTxbOZeexZM9J/rc1nqV7E8nOzS/a1yzIhz6tgtgTn8bPuxJYtT+JVfuTaBDgxUPd6nN3+7p4uZWuSog7ncGiHfH8uD2ercdSirbX9nHjvg5h3NepHsF+mvhOqr707FzyC37n8lWSLiIi5Uyzu4uIVAH5+QafrIzlvZ/3kpdvUM/fkw/vu45mwT6s2HuK77fFs2T3STKyz49db1Dbi9vahHB7m2AaB/oUbT96JoMv1x1m1oajpGXmAuDj7szgDmE82KU+Yf6el4zjcNJZftgez4874tlxPLVou9UCkeE1OXw6g1NpWQA4WS3c1LwOQzvXp2vDWjZpsXdUqptsz5af6dEzGXR/ZxnuLlb2vNHXRhGKiEh1Upp6SUm6iEgVsunIHzw9czPHk8/hbLXg4eJEWlZu0f4wfw9uaxPCbW2CaRHse9lu8Wezcpn3+zGmrTnMoaSzgJlsR7cIYni3+nSM8MdisXDwVDo/bo/nh+0J7I4vnph3aViLvq2CuaVlELV93MjJy+fnnSf5av1h1seeKTq2QYAXD3QO5572dfGroDG9efkGFqgSPw6obrI9W36m24+lcPtHqwnydWf9S71tFKGIiFQnStIvQ1+ERKSqSzmXw9jvtrFoewIAQb7u3NYmmNvahtC2rt8Vx6v/WX6+wYp9p5i65hCr9icVbW8R7EtevsHeCyaac7Ja6NqwFv1aBxPdIpBa3m6XLHffyTRmrD/CvN+Pk17wQ4K7i5X+bUMZ2iWcVqF+pYrzasWdzmD62sN8u/EoDWp7MX14R/y9KveM3KqbbM+Wn+mq/acYOuU3mgX58NPoHjaKUEREqhMl6ZehL0Ii4ggMw0ysvdyciaxX02atxftPpjFt7WG++/0YmTnmmHZnq4XrGwfQr1UwN7cIpGYpE970rFz+u+U4X607wp6E8wl/u7AaDIwK46YWdajj416muA3DYH3sGaauOcQvu09yYc3WItiXbx7tVKmXzlLdZHu2/Ey/33qCp2ZuplOEP7Mf72KjCEVEpDpRkn4Z+iIkInJlyRnZ/G9bPB4uTtzUPNAmXdQNw2DjkT/4at0RftwRX2zG+XZhNYhuGUh0i0Aa1va+6t4AmTl5LNx6gqmrDxX7AaBHk9rc0TaEt37cTVJ6Nq1D/fj6kU74VdJJv1Q32Z4tP9Ov1h/hlQU7uKVlIJ8MjbJRhCIiUp2Upl7S7O4iIlJCDU9XhnQOt2mZFouFDvX96VDfn1NpLZi76Rg/7Uxg69FkthQ83vlpLxEBXtzcIpCbmgcSGV4Tp4v0EkhMy+Tr9XHMWH+E02fNpbE8XJy4q30ow7vVp1Edc6K81qF+3PfZerYfT2HY1N/46uGOJWbFF7mS1MI10j0qb28MERFxHErSRUSkwtX2ceMvvRryl14NOZmayS+7TxKz6yRrD5zmUNJZPl0Zy6crY/H3cqV3szrc3CKQ7o1rcyAxnWlrDvH9thNFLfEhfu482LU+gzuElejS3jTIh68f7sT9n69ny9FkHpq2gS9GdMS7lMvNSfVWtEZ6BU16KCIi1Zu+pYiIiF0F+rrzQKdwHugUTnpWLiv3nSJm10mW7knkzNls5mw6xpxNx3B1spKdd37t98jwmozoFsEtLQNxdrJesvwWIb5mov7ZejYd+YMR0zcwfXgHPF3LXgVm5uTh7uJU5nKkckvOMFvStUa6iIhUBCXpIiJSaXi7OdOvdTD9WgeTk5fPhsNniNlltrIf+8Ncdu62NsEM7xZB27AaV11uq1A/vnq4E0M+/5XfDp3hkS82MmVYBzxcry3B3nTkDyb+sg9nq4VpwzteUxlSdSQXdndXS7qIiFQAJekiIlIpuThZ6dowgK4NA/jHbS04lHQWXw8XAi6z7NvltA2rwfQRHXlwyq+sPXiax77ayGcPRpWqJXzTkTNM/GV/0VJ1zlYLR89kEObveU0xSdWQkqEx6SIiUnEu3T9QRESkkrBYLDSo7X3NCXqhyPCaTB/REU9XJ1btT2Lk15vIys274nkbD59h6JRfuXvyOlbtT8LZamFQVBhLn+ulBL0aSD6nMekiIlJx1JIuIiLVSof6/kx9qAMPTfuN5XtP8cSM35n0QCSuziV/t95w+Awf/LKf1QfOt5zfE1mXJ25opOS8Gikck15Zl/ATERHHoiRdRESqnc4NajFlWAdGTN/AL7sTeWrm73x0f3tcCiag++3QGT5Yso81B04DZnJ+b1RdRvVScl4daUy6iIhUJCXpIiJSLXVrFMCnD0bx6BcbWbzzJKNnbWFI53A+XLqftQcvTM7DGNWroZLzaiozJ4/sXHNVgT8v8SciIlIelKSLiEi11bNJbf4ztD2Pf7WJH7bH88P2eABcnM4n53VrKjmvzgq7ujtbLXhd42oAIiIipaEkXUREqrUbmwXy8f3tGTXjdywWGBgVxqgbGhFaw8PeoUklcOGkcRaLxc7RiIhIdaAkXUREqr3olkEse74Xbi5W6vi42zscqUTq+XsyZ2SXoi7vIiIi5U1JuoiICGjMuVyUp6szHer72zsMERGpRrROuoiIiIiIiEgloSRdREREREREpJJQki4iIiIiIiJSSShJFxEREREREakklKSLiIiIiIiIVBJK0kVEREREREQqCSXpIiIiIiIiIpWEknQRERERERGRSkJJuoiIiIiIiEgloSRdREREREREpJJQki4iIiJlMmnSJCIiInB3dycyMpJVq1Zd8tjly5djsVhKPPbs2VN0zPTp0y96TGZmZkVcjoiIiF052zsAERERqbpmz57N6NGjmTRpEt26deOTTz6hb9++7Nq1i3r16l3yvL179+Lr61v0unbt2sX2+/r6snfv3mLb3N3dbRu8iIhIJaQkXURERK7ZhAkTePjhh3nkkUcAmDhxIosXL2by5MmMHz/+kufVqVOHGjVqXHK/xWIhKCjI1uGKiIhUeuruLiIiItckOzubTZs2ER0dXWx7dHQ0a9euvey51113HcHBwfTu3Ztly5aV2J+enk54eDh169bltttuY/PmzZctLysri9TU1GIPERGRqqjataQbhgGgyltERCqNwjqpsI6qKpKSksjLyyMwMLDY9sDAQBISEi56TnBwMJ9++imRkZFkZWXx1Vdf0bt3b5YvX06PHj0AaNasGdOnT6d169akpqbywQcf0K1bN7Zu3Urjxo0vWu748eMZN25cie2q70VEpDIoTV1vMaraN4IyOnbsGGFhYfYOQ0REpISjR49St25de4dx1U6cOEFoaChr166lS5cuRdv/+c9/8tVXXxWbDO5ybr/9diwWCwsXLrzo/vz8fNq3b0+PHj3497//fdFjsrKyyMrKKnp9/PhxWrRoUYqrERERKX9XU9dXu5b0kJAQjh49io+PDxaLpcT+1NRUwsLCOHr0aLEJbRyVrtex6Xodm67XcRiGQVpaGiEhIfYOpVQCAgJwcnIq0WqemJhYonX9cjp37szXX399yf1Wq5UOHTqwf//+Sx7j5uaGm5tb0Wtvb+9L1veO/N/Sxeh6HZuu17Hpeh1Haer6apekW63Wq2ql8PX1dbj/MC5H1+vYdL2OTdfrGPz8/OwdQqm5uroSGRlJTEwMd955Z9H2mJgY+vfvf9XlbN68meDg4EvuNwyDLVu20Lp166su82rqe0f9b+lSdL2OTdfr2HS9juFq6/pql6SLiIiI7YwZM4ahQ4cSFRVFly5d+PTTT4mLi2PkyJEAjB07luPHj/Pll18C5uzv9evXp2XLlmRnZ/P1118zb9485s2bV1TmuHHj6Ny5M40bNyY1NZV///vfbNmyhY8//tgu1ygiIlKRlKSLiIjINRs0aBCnT5/m9ddfJz4+nlatWrFo0SLCw8MBiI+PJy4uruj47Oxsnn/+eY4fP46HhwctW7bkhx9+oF+/fkXHJCcn89hjj5GQkICfnx/XXXcdK1eupGPHjhV+fSIiIhVNSfqfuLm58eqrrxYb1+bIdL2OTdfr2HS9UlmMGjWKUaNGXXTf9OnTi73+61//yl//+tfLlvf+++/z/vvv2yq8Eqrbf0u6Xsem63Vsut7qqdrN7i4iIiIiIiJSWVntHYCIiIiIiIiImJSki4iIiIiIiFQSStJFREREREREKgkl6SIiIiIiIiKVhJL0C0yaNImIiAjc3d2JjIxk1apV9g6pXLz22mtYLJZij6CgIHuHZTMrV67k9ttvJyQkBIvFwoIFC4rtNwyD1157jZCQEDw8POjVqxc7d+60T7A2cKXrfeihh0rc786dO9snWBsYP348HTp0wMfHhzp16jBgwAD27t1b7BhHusdXc72OdI8nT55MmzZt8PX1xdfXly5duvDjjz8W7Xekeyv2UV3qelB972j/XlSn+l51vep6R7m310pJeoHZs2czevRoXn75ZTZv3kz37t3p27dvsbVdHUnLli2Jj48vemzfvt3eIdnM2bNnadu2LR999NFF97/zzjtMmDCBjz76iA0bNhAUFMTNN99MWlpaBUdqG1e6XoA+ffoUu9+LFi2qwAhta8WKFTzxxBOsX7+emJgYcnNziY6O5uzZs0XHONI9vprrBce5x3Xr1uWtt95i48aNbNy4kRtvvJH+/fsXVc6OdG+l4lW3uh5U3zvSvxfVqb5XXa+63lHu7TUzxDAMw+jYsaMxcuTIYtuaNWtmvPjii3aKqPy8+uqrRtu2be0dRoUAjPnz5xe9zs/PN4KCgoy33nqraFtmZqbh5+dn/Oc//7FDhLb15+s1DMMYNmyY0b9/f7vEUxESExMNwFixYoVhGI5/j/98vYbh+Pe4Zs2axueff+7w91bKX3Wq6w1D9b0j/3tR3ep71fWOfX8NQ3X9n6klHcjOzmbTpk1ER0cX2x4dHc3atWvtFFX52r9/PyEhIURERDB48GBiY2PtHVKFOHToEAkJCcXutZubGz179nTYew2wfPly6tSpQ5MmTXj00UdJTEy0d0g2k5KSAoC/vz/g+Pf4z9dbyBHvcV5eHrNmzeLs2bN06dLF4e+tlK/qWNeD6vvq9u+FI9YFoLq+kCPeX9X1F6ckHUhKSiIvL4/AwMBi2wMDA0lISLBTVOWnU6dOfPnllyxevJjPPvuMhIQEunbtyunTp+0dWrkrvJ/V5V4D9O3blxkzZrB06VL+9a9/sWHDBm688UaysrLsHVqZGYbBmDFjuP7662nVqhXg2Pf4YtcLjnePt2/fjre3N25ubowcOZL58+fTokULh763Uv6qW10Pqu+hev174Wh1QSHV9SZHu7+q6y/P2d4BVCYWi6XYa8MwSmxzBH379i163rp1a7p06ULDhg354osvGDNmjB0jqzjV5V4DDBo0qOh5q1atiIqKIjw8nB9++IG77rrLjpGV3ZNPPsm2bdtYvXp1iX2OeI8vdb2Odo+bNm3Kli1bSE5OZt68eQwbNowVK1YU7XfEeysVpzr996P6vnrdb0erCwqprjc52v1VXX95akkHAgICcHJyKvHrTGJiYolfcRyRl5cXrVu3Zv/+/fYOpdwVzmpbXe81QHBwMOHh4VX+fj/11FMsXLiQZcuWUbdu3aLtjnqPL3W9F1PV77GrqyuNGjUiKiqK8ePH07ZtWz744AOHvbdSMap7XQ+q76F63e+qXheA6vrLqer3V3X95SlJx/yPJDIykpiYmGLbY2Ji6Nq1q52iqjhZWVns3r2b4OBge4dS7iIiIggKCip2r7Ozs1mxYkW1uNcAp0+f5ujRo1X2fhuGwZNPPsl3333H0qVLiYiIKLbf0e7xla73Yqr6Pf4zwzDIyspyuHsrFau61/Wg+r66/XtRlesC1fWq6x3l3l6zip6prrKaNWuW4eLiYkyZMsXYtWuXMXr0aMPLy8s4fPiwvUOzueeee85Yvny5ERsba6xfv9647bbbDB8fH4e51rS0NGPz5s3G5s2bDcCYMGGCsXnzZuPIkSOGYRjGW2+9Zfj5+RnfffedsX37duO+++4zgoODjdTUVDtHfm0ud71paWnGc889Z6xdu9Y4dOiQsWzZMqNLly5GaGholb3ev/zlL4afn5+xfPlyIz4+vuiRkZFRdIwj3eMrXa+j3eOxY8caK1euNA4dOmRs27bNeOmllwyr1Wr8/PPPhmE41r2Viled6nrDUH3vaP9eVKf6XnW96npHubfXSkn6BT7++GMjPDzccHV1Ndq3b19s2QNHMmjQICM4ONhwcXExQkJCjLvuusvYuXOnvcOymWXLlhlAicewYcMMwzCX7Xj11VeNoKAgw83NzejRo4exfft2+wZdBpe73oyMDCM6OtqoXbu24eLiYtSrV88YNmyYERcXZ++wr9nFrhUwpk2bVnSMI93jK12vo93jESNGFP07XLt2baN3795FlbZhONa9FfuoLnW9Yai+d7R/L6pTfa+6XnW9o9zba2UxDMOwffu8iIiIiIiIiJSWxqSLiIiIiIiIVBJK0kVEREREREQqCSXpIiIiIiIiIpWEknQRERERERGRSkJJuoiIiIiIiEgloSRdREREREREpJJQki4iIiIiIiJSSShJFxEREREREakklKSLSIWzWCwsWLDA3mGIiIhIOVFdL3LtlKSLVDMPPfQQFoulxKNPnz72Dk1ERERsQHW9SNXmbO8ARKTi9enTh2nTphXb5ubmZqdoRERExNZU14tUXWpJF6mG3NzcCAoKKvaoWbMmYHZPmzx5Mn379sXDw4OIiAjmzJlT7Pzt27dz44034uHhQa1atXjsscdIT08vdszUqVNp2bIlbm5uBAcH8+STTxbbn5SUxJ133omnpyeNGzdm4cKF5XvRIiIi1YjqepGqS0m6iJTwyiuvcPfdd7N161aGDBnCfffdx+7duwHIyMigT58+1KxZkw0bNjBnzhx++eWXYhXz5MmTeeKJJ3jsscfYvn07CxcupFGjRsXeY9y4cQwcOJBt27bRr18/HnjgAc6cOVOh1ykiIlJdqa4XqcQMEalWhg0bZjg5ORleXl7FHq+//rphGIYBGCNHjix2TqdOnYy//OUvhmEYxqeffmrUrFnTSE9PL9r/ww8/GFar1UhISDAMwzBCQkKMl19++ZIxAMbf//73otfp6emGxWIxfvzxR5tdp4iISHWlul6katOYdJFq6IYbbmDy5MnFtvn7+xc979KlS7F9Xbp0YcuWLQDs3r2btm3b4uXlVbS/W7du5Ofns3fvXiwWCydOnKB3796XjaFNmzZFz728vPDx8SExMfFaL0lEREQuoLpepOpSki5SDXl5eZXoknYlFosFAMMwip5f7BgPD4+rKs/FxaXEufn5+aWKSURERC5Odb1I1aUx6SJSwvr160u8btasGQAtWrRgy5YtnD17tmj/mjVrsFqtNGnSBB8fH+rXr8+SJUsqNGYRERG5eqrrRSovtaSLVENZWVkkJCQU2+bs7ExAQAAAc+bMISoqiuuvv54ZM2bw22+/MWXKFAAeeOABXn31VYYNG8Zrr73GqVOneOqppxg6dCiBgYEAvPbaa4wcOZI6derQt29f0tLSWLNmDU899VTFXqiIiEg1pbpepOpSki5SDf30008EBwcX29a0aVP27NkDmLOxzpo1i1GjRhEUFMSMGTNo0aIFAJ6enixevJhnnnmGDh064Onpyd13382ECROKyho2bBiZmZm8//77PP/88wQEBHDPPfdU3AWKiIhUc6rrRaoui2EYhr2DEJHKw2KxMH/+fAYMGGDvUERERKQcqK4Xqdw0Jl1ERERERESkklCSLiIiIiIiIlJJqLu7iIiIiIiISCWhlnQRERERERGRSkJJuoiIiIiIiEgloSRdREREREREpJJQki4iIiIiIiJSSShJFxEREREREakklKSLiIiIiIiIVBJK0kVEREREREQqCSXpIiIiIiIiIpXE/wOwdgKmMZicIwAAAABJRU5ErkJggg=="
+      "image/png": "iVBORw0KGgoAAAANSUhEUgAAA+4AAAHUCAYAAACkmnJcAAAAOnRFWHRTb2Z0d2FyZQBNYXRwbG90bGliIHZlcnNpb24zLjEwLjAsIGh0dHBzOi8vbWF0cGxvdGxpYi5vcmcvlHJYcgAAAAlwSFlzAAAPYQAAD2EBqD+naQAA0CRJREFUeJzs3Xd8U9X7wPFP0pXu3dJCadkttOyN7I2ggCiggAioyBcUcSLLjeOHoiAoMioKMlQQlY0ytGwoIGWPllFa2tK90uT+/rgQqG1ZTUnH83698qK5Offe56YNyZNzznM0iqIoCCGEEEIIIYQQolTSWjoAIYQQQgghhBBCFE0SdyGEEEIIIYQQohSTxF0IIYQQQgghhCjFJHEXQgghhBBCCCFKMUnchRBCCCGEEEKIUkwSdyGEEEIIIYQQohSTxF0IIYQQQgghhCjFJHEXQgghhBBCCCFKMUnchRBCCCGEEEKIUkwSd1GhaTSau7pt3bq1WOd5++230Wg097Xv1q1bzRJDcc79008/PfBz34/Dhw/zzDPPUK1aNXQ6HU5OTjRu3JhPPvmEpKQkS4cnhBDiLsh7891bs2YNGo0GT09PcnJyLBpLWZSamsoHH3xA06ZNcXFxwc7OjqCgIEaMGMGBAwcsHZ4Q+VhbOgAhLGnnzp357r/33nv89ddf/Pnnn/m2161bt1jnGTVqFD169LivfRs3bszOnTuLHUN59+233zJmzBjq1KnDa6+9Rt26ddHr9ezbt4+vv/6anTt3smrVKkuHKYQQ4g7kvfnuLViwAICkpCRWr17NwIEDLRpPWXLmzBm6detGfHw8o0eP5p133sHJyYnz58+zYsUKmjRpQnJyMq6urpYOVQgANIqiKJYOQojSYvjw4fz000+kp6fftl1mZiYODg4PKCrL2bp1Kx07dmTlypUMGDDA0uEUaefOnbRt25auXbuyevVq7Ozs8j2em5vL+vXreeSRR4p9rqysLHQ63X330gghhLg38t5cuCtXrhAQEEC7du2IiIigbdu2bNy40dJhFaq0/W4MBgONGjUiOjqaf/75h9DQ0AJt1q1bR/v27Ysdt6IoZGdnY29vX6zjCCFD5YW4gw4dOhAaGsr27dtp3bo1Dg4OjBgxAoDly5fTrVs3/Pz8sLe3JyQkhDfffJOMjIx8xyhsOF5QUBC9e/dm/fr1NG7cGHt7e4KDg1m4cGG+doUNxxs+fDhOTk6cPn2aXr164eTkREBAAK+88kqBoXIXL15kwIABODs74+bmxlNPPcXevXvRaDSEh4eb5Tn6999/efTRR3F3d0en09GwYUO+++67fG2MRiPvv/8+derUwd7eHjc3N+rXr88XX3xhanP16lWee+45AgICsLOzw9vbmzZt2rB58+bbnv/DDz9Eo9Ewb968Akk7gK2tbb6kXaPR8PbbbxdoFxQUxPDhw033w8PD0Wg0bNy4kREjRuDt7Y2DgwPLly9Ho9GwZcuWAseYO3cuGo2Gw4cPm7bt27ePRx55BA8PD3Q6HY0aNWLFihW3vSYhhBBFk/dm+O6778jLy+Pll1+mf//+bNmyhejo6ALtkpOTeeWVV6hevTp2dnb4+PjQq1cvjh8/bmqTk5PDu+++S0hICDqdDk9PTzp27EhERAQA58+fLzK2/76n3nheDxw4wIABA3B3d6dGjRqA+n44aNAggoKCsLe3JygoiMGDBxca96VLl0yfCWxtbfH392fAgAHExcWRnp6Om5sbzz//fIH9zp8/j5WVFZ9++mmRz93q1as5cuQIEydOLDRpB+jZs6cpaR8+fDhBQUEF2hT2N6TRaBg7dixff/01ISEh2NnZMX/+fHx8fBg6dGiBYyQnJ2Nvb8+ECRNM21JTU3n11VepVq0atra2VK5cmfHjxxf4GxYViwyVF+IuxMbGMmTIEF5//XU+/PBDtFr1O69Tp07Rq1cvxo8fj6OjI8ePH+fjjz9mz549BYb0FebQoUO88sorvPnmm/j6+jJ//nxGjhxJzZo1adeu3W331ev1PPLII4wcOZJXXnmF7du389577+Hq6srUqVMByMjIoGPHjiQlJfHxxx9Ts2ZN1q9fb9ahdCdOnKB169b4+Pjw5Zdf4unpyQ8//MDw4cOJi4vj9ddfB+CTTz7h7bffZvLkybRr1w69Xs/x48dJTk42HWvo0KEcOHCADz74gNq1a5OcnMyBAwdITEws8vwGg4E///yTJk2aEBAQYLbrutWIESN4+OGH+f7778nIyKB37974+PiwaNEiOnfunK9teHg4jRs3pn79+gD89ddf9OjRgxYtWvD111/j6urKsmXLGDhwIJmZmfm+KBBCCHH3Kvp788KFC/Hz86Nnz57Y29uzdOlSwsPDmTZtmqlNWloaDz30EOfPn+eNN96gRYsWpKens337dmJjYwkODiYvL4+ePXuyY8cOxo8fT6dOncjLy2PXrl3ExMTQunXre4rrhv79+zNo0CBGjx5tSjjPnz9PnTp1GDRoEB4eHsTGxjJ37lyaNWtGVFQUXl5egJq0N2vWDL1ez1tvvUX9+vVJTExkw4YNXLt2DV9fX0aMGMG8efP45JNP8g1nnzNnDra2tqYvcgpzY2RC37597+va7mT16tXs2LGDqVOnUqlSJXx8fDh37hxff/01X331FS4uLqa2P/74I9nZ2TzzzDOAOjqhffv2XLx40XTtR48eZerUqRw5coTNmzfLqL+KShFCmDz99NOKo6Njvm3t27dXAGXLli233ddoNCp6vV7Ztm2bAiiHDh0yPTZt2jTlvy+3wMBARafTKdHR0aZtWVlZioeHh/L888+btv31118KoPz111/54gSUFStW5Dtmr169lDp16pjuf/XVVwqgrFu3Ll+7559/XgGURYsW3faabpx75cqVRbYZNGiQYmdnp8TExOTb3rNnT8XBwUFJTk5WFEVRevfurTRs2PC253NyclLGjx9/2zb/deXKFQVQBg0adNf7AMq0adMKbA8MDFSefvpp0/1FixYpgDJs2LACbSdMmKDY29ubrk9RFCUqKkoBlFmzZpm2BQcHK40aNVL0en2+/Xv37q34+fkpBoPhruMWQoiKSN6bC9q+fbsCKG+++abpOqtVq6YEBgYqRqPR1O7dd99VAGXTpk1FHmvx4sUKoHz77bdFtjl37lyRsf33PfXG8zp16tQ7XkdeXp6Snp6uODo6Kl988YVp+4gRIxQbGxslKiqqyH3PnDmjaLVa5fPPPzdty8rKUjw9PZVnnnnmtuft0aOHAijZ2dl3jFFR1N9tYGBgge2F/Q0Biqurq5KUlJRv++HDhxVAmTdvXr7tzZs3V5o0aWK6P336dEWr1Sp79+7N1+6nn35SAGXt2rV3FbMof2SovBB3wd3dnU6dOhXYfvbsWZ588kkqVaqElZUVNjY2tG/fHoBjx47d8bgNGzakatWqpvs6nY7atWsXOmTsvzQaDX369Mm3rX79+vn23bZtG87OzgWK7wwePPiOx79bf/75J507dy7Q2z18+HAyMzNNRYaaN2/OoUOHGDNmDBs2bCA1NbXAsZo3b054eDjvv/8+u3btQq/Xmy3O4njssccKbBsxYgRZWVksX77ctG3RokXY2dnx5JNPAnD69GmOHz/OU089BUBeXp7p1qtXL2JjYzlx4sSDuQghhChnKvJ7842idDd6lTUaDcOHDyc6OjrfNK5169ZRu3ZtunTpUuSx1q1bh06nu20P9f0o7L0zPT2dN954g5o1a2JtbY21tTVOTk5kZGTk+92sW7eOjh07EhISUuTxq1evTu/evZkzZw7K9ZJdS5cuJTExkbFjx5r1Wu5Vp06dcHd3z7ctLCyMJk2asGjRItO2Y8eOsWfPnnzP/e+//05oaCgNGzbM97mhe/fupWIlA2E5krgLcRf8/PwKbEtPT6dt27bs3r2b999/n61bt7J3715++eUXQC1idieenp4FttnZ2d3Vvg4ODuh0ugL7Zmdnm+4nJibi6+tbYN/Ctt2vxMTEQp8ff39/0+MAEydO5P/+7//YtWsXPXv2xNPTk86dO7Nv3z7TPsuXL+fpp59m/vz5tGrVCg8PD4YNG8aVK1eKPL+XlxcODg6cO3fObNf0X4VdX7169WjWrJnpDdhgMPDDDz/w6KOP4uHhAUBcXBwAr776KjY2NvluY8aMASAhIaHE4hZCiPKsor43p6WlsXLlSpo3b463tzfJyckkJyfTr18/NBqNKakHtXZMlSpVbnu8q1ev4u/vb5pqYC6F/X6efPJJZs+ezahRo9iwYQN79uxh7969eHt753t+7yZugJdeeolTp06xadMmAL766itatWpF48aNb7vfjS9mSuqzQ2HXDuoXLTt37jTVF7jxhf+tX9rExcVx+PDhAp8bnJ2dURRFPjdUYDLHXYi7UNhcoj///JPLly+zdetW0zf5QL4525bm6enJnj17Cmy/XSJ8P+eIjY0tsP3y5csApvlq1tbWTJgwgQkTJpCcnMzmzZt566236N69OxcuXMDBwQEvLy9mzpzJzJkziYmJYc2aNbz55pvEx8ezfv36Qs9vZWVF586dWbduHRcvXryrN3o7O7tC17stai59UXPJnnnmGcaMGcOxY8c4e/YssbGxpjlqt177xIkT6d+/f6HHqFOnzh3jFUIIUVBFfW/+8ccfyczMZM+ePQV6dQFWrVrFtWvXcHd3x9vbm4sXL972eN7e3vz9998YjcYik/cbX0b8973zdjVo/vv7SUlJ4ffff2fatGm8+eabpu05OTkkJSUViOlOcYPasx0aGsrs2bNxcnLiwIED/PDDD3fcr3v37sybN4/Vq1fni6UoOp2u0M8NRSXRRX1uGDx4MBMmTCA8PJwPPviA77//nr59++b7PXp5eWFvb1+gIOKtj4uKSXrchbhPN/5T/m8V82+++cYS4RSqffv2pKWlsW7dunzbly1bZrZzdO7c2fRB6VaLFy/GwcGBli1bFtjHzc2NAQMG8L///Y+kpCTOnz9foE3VqlUZO3YsXbt25cCBA7eNYeLEiSiKwrPPPktubm6Bx/V6Pb/99pvpflBQUL6q76B+2LvTUkP/NXjwYHQ6HeHh4YSHh1O5cmW6detmerxOnTrUqlWLQ4cO0bRp00Jvzs7O93ROIYQQRasI780LFizA2dmZLVu28Ndff+W7ffrpp+Tk5LBkyRJArYx+8uTJ2xbl69mzJ9nZ2betZu/r64tOpyvw3vnrr7/eVcyg/m4URSnwu5k/fz4Gg6FATH/99dddTSd78cUX+eOPP5g4cSK+vr48/vjjd9zn0UcfJSwsjOnTp/Pvv/8W2mbDhg1kZmYC6ueG+Ph400g6UJea3bBhwx3PdSt3d3f69u3L4sWL+f3337ly5UqBKQq9e/fmzJkzeHp6Fvq5obDq9qJikB53Ie5T69atcXd3Z/To0UybNg0bGxuWLFnCoUOHLB2aydNPP83nn3/OkCFDeP/996lZsybr1q0zvdHc7bC4Xbt2Fbq9ffv2TJs2jd9//52OHTsydepUPDw8WLJkCX/88Ue+Sq99+vQhNDSUpk2b4u3tTXR0NDNnziQwMJBatWqRkpJCx44defLJJwkODsbZ2Zm9e/eyfv36Inurb2jVqhVz585lzJgxNGnShBdeeIF69eqh1+s5ePAg8+bNIzQ01DTvcOjQoUyZMoWpU6fSvn17oqKimD17dr6qtHfDzc2Nfv36ER4eTnJyMq+++mqB5/Sbb76hZ8+edO/eneHDh1O5cmWSkpI4duwYBw4cYOXKlfd0TiGEEEUr7+/N//77L3v27OGFF14odH5/mzZtmDFjBgsWLGDs2LGMHz+e5cuX8+ijj/Lmm2/SvHlzsrKy2LZtG71796Zjx44MHjyYRYsWMXr0aE6cOEHHjh0xGo3s3r2bkJAQBg0ahEajYciQISxcuJAaNWrQoEED9uzZw9KlS+/6ul1cXGjXrh2ffvopXl5eBAUFsW3bNhYsWICbm1u+tu+++y7r1q2jXbt2vPXWW4SFhZGcnMz69euZMGECwcHBprZDhgxh4sSJbN++ncmTJ2Nra3vHWKysrFi1ahXdunWjVatWvPDCC3Ts2BFHR0eio6P56aef+O2337h27RoAAwcOZOrUqQwaNIjXXnuN7OxsvvzyywJfONyNESNGsHz5csaOHUuVKlUK1B8YP348P//8M+3atePll1+mfv36GI1GYmJi2LhxI6+88gotWrS45/OKcsCytfGEKF2Kqlxbr169QttHREQorVq1UhwcHBRvb29l1KhRyoEDBwpUXi2qcu3DDz9c4Jjt27dX2rdvb7pfVOXa/8ZZ1HliYmKU/v37K05OToqzs7Py2GOPKWvXrlUA5ddffy3qqch37qJuN2I6cuSI0qdPH8XV1VWxtbVVGjRoUKDy7IwZM5TWrVsrXl5eiq2trVK1alVl5MiRyvnz5xVFUZTs7Gxl9OjRSv369RUXFxfF3t5eqVOnjjJt2jQlIyPjtnHeEBkZqTz99NNK1apVFVtbW8XR0VFp1KiRMnXqVCU+Pt7ULicnR3n99deVgIAAxd7eXmnfvr0SGRlZZFX5/1Z2vdXGjRtNz8fJkycLbXPo0CHliSeeUHx8fBQbGxulUqVKSqdOnZSvv/76rq5LCCEqMnlvvmn8+PEKoERGRhbZ5s0331QAZf/+/YqiKMq1a9eUl156SalatapiY2Oj+Pj4KA8//LBy/Phx0z5ZWVnK1KlTlVq1aim2traKp6en0qlTJyUiIsLUJiUlRRk1apTi6+urODo6Kn369FHOnz9fZFX5q1evFojt4sWLymOPPaa4u7srzs7OSo8ePZR///23wPuvoijKhQsXlBEjRiiVKlVSbGxsFH9/f+WJJ55Q4uLiChx3+PDhirW1tXLx4sUin5fCJCcnK++9957SuHFjxcnJSbGxsVGqVq2qDBkyRPnnn3/ytV27dq3SsGFDxd7eXqlevboye/bsIqvK/+9//yvynAaDQQkICFAAZdKkSYW2SU9PVyZPnqzUqVNHsbW1VVxdXZWwsDDl5ZdfVq5cuXJP1yjKD42iXC/DKISoMD788EMmT55MTEzMXc0JF0IIIUTJkvfm+5Obm0tQUBAPPfQQK1assHQ4QpQYGSovRDk3e/ZsAIKDg9Hr9fz55598+eWXDBkyRD4YCCGEEBYg783Fd/XqVU6cOMGiRYuIi4u7qyJzQpRlkrgLUc45ODjw+eefc/78eXJycqhatSpvvPEGkydPtnRoQgghRIUk783F98cff/DMM8/g5+fHnDlz7rgEnBBlnQyVF0IIIYQQQgghSjFZDk4IIYQQQgghhCjFJHEXQgghhBBCCCFKMUnchRBCCCGEEEKIUqzCFaczGo1cvnwZZ2dnNBqNpcMRQgghUBSFtLQ0/P390WrlO3VzkPd7IYQQpUlx3+srXOJ++fJlAgICLB2GEEIIUcCFCxdkKSgzkfd7IYQQpdH9vtdXuMTd2dkZUJ8wFxcXC0cjhBBCQGpqKgEBAab3KFF88n4vhBCiNCnue32FS9xvDJdzcXGRN3IhhBCligzpNh95vxdCCFEa3e97vUykE0IIIYQQQgghSjFJ3IUQQgghhBBCiFJMEnchhBBCCCGEEKIUq3Bz3IUQoqxQFIW8vDwMBoOlQxHFZGVlhbW1tcxhL2UMBgN6vd7SYYhyQl7nQoiSJIm7EEKUQrm5ucTGxpKZmWnpUISZODg44Ofnh62traVDEUB6ejoXL15EURRLhyLKEXmdCyFKiiTuQghRyhiNRs6dO4eVlRX+/v7Y2tpKD04ZpigKubm5XL16lXPnzlGrVi20WpmpZkkGg4GLFy/i4OCAt7e3vL5EscnrXAhR0iRxF0KIUiY3Nxej0UhAQAAODg6WDkeYgb29PTY2NkRHR5Obm4tOp7N0SBWaXq9HURS8vb2xt7e3dDiinJDXuRCiJMlXgUIIUUpJb035Ir/P0kd62oW5yetcCFFS5H8XIYQQQgghhBCiFJPEXQghhBBCCCGEKMUkcRdCCFFqdejQgfHjx1s6DCHKLXmNCSFE2SCJuxBCiGLTaDS3vQ0fPvy+jvvLL7/w3nvvFSu24cOH07dv32IdQwhLK82vsRsiIiKwsrKiR48eZjmeEEKIm6SqvBBCiGKLjY01/bx8+XKmTp3KiRMnTNv+W7lbr9djY2Nzx+N6eHiYL0ghyrCy8BpbuHAh48aNY/78+cTExFC1alWzHfte3e31CyFEWSE97sWw+uAleszczvS1xywdihCiHFMUhczcPIvcFEW5qxgrVapkurm6uqLRaEz3s7OzcXNzY8WKFXTo0AGdTscPP/xAYmIigwcPpkqVKjg4OBAWFsaPP/6Y77j/HcYbFBTEhx9+yIgRI3B2dqZq1arMmzevWM/vtm3baN68OXZ2dvj5+fHmm2+Sl5dnevynn34iLCwMe3t7PD096dKlCxkZGQBs3bqV5s2b4+joiJubG23atCE6OrpY8ZRFc+bMoVq1auh0Opo0acKOHTtu2/6rr74iJCQEe3t76tSpw+LFiwu0+fnnn6lbty52dnbUrVuXVatWlVT48hozw2ssIyODFStW8MILL9C7d2/Cw8MLtFmzZg1NmzZFp9Ph5eVF//79TY/l5OTw+uuvExAQgJ2dHbVq1WLBggUAhIeH4+bmlu9Yq1evzrcqwNtvv03Dhg1ZuHAh1atXx87ODkVRWL9+PQ899BBubm54enrSu3dvzpw5k+9YFy9eZNCgQXh4eODo6EjTpk3ZvXs358+fR6vVsm/fvnztZ82aRWBg4F3/7oQQpZPBqHDmajprDl3mo3XHGbZwD098s5N3fjvKqoMXOR2fjtFYel7n0uNeDKnZeo5fSaO6t6OlQxFClGNZegN1p26wyLmj3u2Og6153ireeOMNZsyYwaJFi7CzsyM7O5smTZrwxhtv4OLiwh9//MHQoUOpXr06LVq0KPI4M2bM4L333uOtt97ip59+4oUXXqBdu3YEBwffc0yXLl2iV69eDB8+nMWLF3P8+HGeffZZdDodb7/9NrGxsQwePJhPPvmEfv36kZaWxo4dO1AUhby8PPr27cuzzz7Ljz/+SG5uLnv27KlwS4wtX76c8ePHM2fOHNq0acM333xDz549iYqKKrTHde7cuUycOJFvv/2WZs2asWfPHp599lnc3d3p06cPADt37mTgwIG899579OvXj1WrVvHEE0/w999/3/Zv437Jayy/+3mNLV++nDp16lCnTh2GDBnCuHHjmDJliun18Mcff9C/f38mTZrE999/T25uLn/88Ydp/2HDhrFz506+/PJLGjRowLlz50hISLin6z99+jQrVqzg559/xsrKClC/UJgwYQJhYWFkZGQwdepU+vXrR2RkJFqtlvT0dNq3b0/lypVZs2YNlSpV4sCBAxiNRoKCgujSpQuLFi2iadOmpvMsWrSI4cOHV7jXuhDlhaIovPNbFMv3XiBLbyjw+J5zSaafneysCa3swus9gmlc1f1BhlmAJO7FoLNW3xSycgv+woUQQuQ3fvz4fD1sAK+++qrp53HjxrF+/XpWrlx526SiV69ejBkzBlATlc8//5ytW7feV+I+Z84cAgICmD17NhqNhuDgYC5fvswbb7zB1KlTiY2NJS8vj/79+xMYGAhAWFgYAElJSaSkpNC7d29q1KgBQEhIyD3HUNZ99tlnjBw5klGjRgEwc+ZMNmzYwNy5c5k+fXqB9t9//z3PP/88AwcOBKB69ers2rWLjz/+2JS4z5w5k65duzJx4kQAJk6cyLZt25g5c2aBHmNxkyVfYwsWLGDIkCEA9OjRg/T0dLZs2UKXLl0A+OCDDxg0aBDvvPOOaZ8GDRoAcPLkSVasWMGmTZtM7atXr34vlw5Abm4u33//Pd7e3qZtjz32WIE4fXx8iIqKIjQ0lKVLl3L16lX27t1rmjZQs2ZNU/tRo0YxevRoPvvsM+zs7Dh06BCRkZH88ssv9xyfEKJ0WLnvIuER5wHQ2WgJruRCPX8X6vq7oLO24silFI5cSuHo5RTSc/LYdTYJG63lB6pL4l4MOls1cc/WGy0ciRCiPLO3sSLq3e4WO7e53NpjBWAwGPjoo49Yvnw5ly5dIicnh5ycHBwdbz+KqX79+qafbwwXjo+Pv6+Yjh07RqtWrfL1nLVp04b09HQuXrxIgwYN6Ny5M2FhYXTv3p1u3boxYMAA3N3d8fDwYPjw4XTv3p2uXbvSpUsXnnjiCfz8/O4rlrIoNzeX/fv38+abb+bb3q1bNyIiIgrdJycnB51Ol2+bvb09e/bsMc1L3rlzJy+//HK+Nt27d2fmzJlFxnLj7+eG1NTUu74OeY3ld6+vsRMnTrBnzx5TMmttbc3AgQNZuHChKRGPjIzk2WefLXT/yMhIrKysaN++/V1dZ1ECAwPzJe0AZ86cYcqUKezatYuEhASMRvUzW0xMDKGhoURGRtKoUaMi5/r37duXsWPHsmrVKgYNGsTChQvp2LEjQUFBxYpVCGEZcanZvPdHFACvda/D6PY1sNLmHz3zWJMqAOQZjJy+ms7hCynUruT0wGP9L0nci0FnrX7zkp0nPe5CiJKj0WjMNpTWkv6bLMyYMYPPP/+cmTNnEhYWhqOjI+PHjyc3N/e2x/lvwSmNRmP6MH6vFEUpMNz1xrxVjUaDlZUVmzZtIiIigo0bNzJr1iwmTZrE7t27qVatGosWLeLFF19k/fr1LF++nMmTJ7Np0yZatmx5X/GUNQkJCRgMBnx9ffNt9/X15cqVK4Xu0717d+bPn0/fvn1p3Lgx+/fvZ+HChej1ehISEvDz8+PKlSv3dEyA6dOn5+vNvRfyGsvvXl9jCxYsIC8vj8qVK5u2KYqCjY0N165dw93dvUDxvFvd7jEArVZbYD65Xq8v0K6wLyT69OlDQEAA3377Lf7+/hiNRkJDQ03PwZ3ObWtry9ChQ1m0aBH9+/dn6dKlt/0CSQhReimKwqRVR0jLzqNBFVeeb1e9QNJ+K2srtTc+uJLLA4yyaJbv8y/DdDYyVF4IIe7Xjh07ePTRRxkyZAgNGjSgevXqnDp16oHGULduXSIiIvIlBRERETg7O5uSEI1GQ5s2bXjnnXc4ePAgtra2+QqlNWrUiIkTJxIREWEaelvRFPblR1Hzf6dMmULPnj1p2bIlNjY2PProo6alzG7MS77XY4I6nD4lJcV0u3Dhwn1eTfnxIF5jeXl5LF68mBkzZhAZGWm6HTp0iMDAQJYsWQKovfhbtmwp9BhhYWEYjUa2bdtW6OPe3t6kpaWZikKC2kt/J4mJiRw7dozJkyfTuXNnQkJCuHbtWr429evXJzIykqSkpCKOog6X37x5M3PmzEGv1xeYjiCEKBvWHLrM5mPx2Fhp+GRAA6ytylYqXLaiLWVuJO45eTJUXggh7lXNmjVNvdnHjh3j+eefv22PanGkpKTkSyoiIyOJiYlhzJgxXLhwgXHjxnH8+HF+/fVXpk2bxoQJE9BqtezevZsPP/yQffv2ERMTwy+//MLVq1cJCQnh3LlzTJw4kZ07dxIdHc3GjRs5efJkhZrn7uXlhZWVVYHfW3x8fIEe8xvs7e1ZuHAhmZmZnD9/npiYGIKCgnB2dsbLywtQK6jfyzEB7OzscHFxyXer6B7Ea+z333/n2rVrjBw5ktDQ0Hy3AQMGmCrDT5s2jR9//JFp06Zx7Ngxjhw5wieffAKoleyffvppRowYwerVqzl37hxbt25lxYoVALRo0QIHBwfeeustTp8+zdKlSwutWv9f7u7ueHp6Mm/ePE6fPs2ff/7JhAkT8rUZPHgwlSpVom/fvvzzzz+cPXuWn3/+mZ07d5rahISE0LJlS9544w0GDx58x156IUTpk5iewzu/qUPkx3WqRZ1KzhaO6N5J4l4MN+alZRdSjVAIIcTtTZkyhcaNG9O9e3c6dOhg+vBcErZu3UqjRo3y3aZOnUrlypVZu3Yte/bsoUGDBowePZqRI0cyefJkAFxcXNi+fTu9evWidu3aTJ48mRkzZtCzZ08cHBw4fvw4jz32GLVr1+a5555j7NixPP/88yVyDaWRra0tTZo0YdOmTfm2b9q0idatW992XxsbG6pUqYKVlRXLli2jd+/eaK8X/2nVqlWBY27cuPGOxxT5PYjX2IIFC+jSpQuurq4FHnvssceIjIzkwIEDdOjQgZUrV7JmzRoaNmxIp06d2L17t6nt3LlzGTBgAGPGjCE4OJhnn33W1MPu4eHBDz/8wNq1a01L2r399tt3jE2r1bJs2TL2799PaGgoL7/8Mp9++mm+Nra2tmzcuBEfHx969epFWFgYH330Ub7RHwAjR44kNzeXESNG3MezJIR4ELL1Bv4+lcDfpxLIyMnL99i0NUdJysgluJIzL3SoYaEIi0ejVLBFKFNTU3F1dSUlJaXY38afikuj6+fbcXew4eDUbmaKUAhR0WVnZ3Pu3DnTutiifLjd79Wc700P2vLlyxk6dChff/01rVq1Yt68eXz77bccPXqUwMBAJk6cyKVLl0xrtZ88eZI9e/bQokULrl27xmeffcamTZvYv3+/qeBXREQE7dq144MPPuDRRx/l119/ZfLkyfe0HNztnlN5jYl79cEHH7Bs2TKOHDly23bytyXEg3U5OYs/j8fz1/F4/jmTYCoabq3VEFrZlRbVPHCxt+HTDSew0mpYPaYNYVUKftH4IBT3vb7sV2KxINMcd+lxF0IIUUENHDiQxMRE3n33XWJjYwkNDWXt2rWm5fNiY2OJiYkxtTcYDMyYMYMTJ05gY2NDx44diYiIyFelu3Xr1ixbtozJkyczZcoUatSowfLly0tkDXchbic9PZ1jx44xa9Ys3nvvPUuHI0S5cuJKGh+uPUbzah6FVnf/r5QsPUcvpxB1OZV/ry/ZduZqRr42lVx0WGk1XErOIvJCMpEXkk2PPdeuusWSdnOQxL0Y7GyuV5XXG+9YNEcIIYQor8aMGWNa9/u//jsXOSQkhIMHD97xmAMGDGDAgAHmCE+I+zZ27Fh+/PFH+vbtK8PkhTCj9f9eYcKKSDJzDWw7eZVdZxP5YlAjPBxtC7TdeiKej9ef4FhswWU+tRpoXNWdjsE+dKzjQ4ifMxqNhovXMtl7Pok959Sbv5s9L3Wu9SAurcRI4l4Mt669mpNnNPXACyGEEEKIsi88PPyuCuEJIe6O0agw68/TfL75JAANqrhyIi6NHacS6P3lDuYMaULDADcALiRl8u7vUWyKijPtX8XdnlB/V+r5uxBa2ZWGAW64F5LsV3F3oIq7A/0aVXkg1/UgSOJeDLcm6jl6SdyFEEIIIYQQZV9mbh57ziXh5WRHDW8n7G3vnOekZuvZHBXHH4dj2Xk2kQB3B1rV8KR1DU9aVPfEWqvhlRWHWH9UXd3imTZBTOoVwumr6bzwwwHOJWTw+NcRTOoVwrVMPXO3nSE3z4i1VsPw1kG80KEGnk52JX3ppZYk7sVgY6XFSqvBYFTI0htwxcbSIQkhhBBCCCHEfTt6OYWxSw9yLkGdP67RQIC7A7V8nKjh44SLzhqdjRX2tlbY21ihNxjZFBXP9pNXyTXcXCb7RFwaJ+LSCI84j1YDrvY2XMvUY2ul5f1+oTzRNACA4Eou/Dq2Da+vPMz6o1d4+/qybQBtanrydp961PIte8u3mZsk7sWks9aSkWuQJeGEEEIIIYQQZZaiKCzdE8M7v0WRm2c0zTdPysglJimTmKRMthyPv+0xang70ru+P51DfLh4LYuIMwlEnEnk7NUMrmXq8Xa245uhTWhc1T3ffi46G+YOacz8Hef4aP1xfJ3tmNy7Lj1DK0kdseskcS8me1srNXHPk8RdCCGEEEIIUTpl5uZxJSWb9Jw8/N3s8XS0NSXFadl6Jv5yhN8PxwLQKdiHGY83wN3RlsT0HE7GpXM6Po1zCZlk5uaRpTeQlWsgS28gz6DQLMidh+v7U9vXyXTM+lXc6BXmB8CVlGwiL1yjaZAHXkUMd9doNDzbrjqPNamCk501ttbaB/CslB2SuBeTnfX1JeFyJXEXQgghhBBCPHiXk7PYdTaR5Ew9yVl6UrP0JGfmkpiRS1xqNldSsknNzsu3j4OtFQHuDgR4OHAqPo3oxEystRpe71GHUQ9VR3t9eTZPJztaOdnRqobnfcdXyVVHD1e/u2pbWGV5IYl7seluWRJOCCGEEEIIIcwlW28gNVuPj7Ou0MfTc/KYu/U03+44R27enfMRB1srnOysuZqeQ2auwTQPHcDfVcesJxvTJND9DkcRliCJezHdqLAoQ+WFEKL4OnToQMOGDZk5cyYAQUFBjB8/nvHjxxe5j0ajYdWqVfTt2/eBxChEWSavMSHKBkVRWHPoMtPWHCU5U0/9Kq70qe/Pw/X98Hezx2BU+Hn/RT7deIKraTkAhFV2JdDTAVd7G9wcbK7/a0slFx1+rjp8XXU421mj0WjIyTNw6VoWF65lEZOUSW6ekf6NKhe6tJooHSRxLybd9aHyOVKcTghRgfXp04esrCw2b95c4LGdO3fSunVr9u/fT+PGje/puHv37sXR0bFYsQ0fPpzk5GRWr15drOMIYUml+TV2Q0REBG3btqVr166sX7/eLMcUoiKKT8tm8qp/2XjL+uWHL6Zw+GIKH6w9RrMgdzJzDRy9nApAkKcDb/UKoWtd37su5GZnbUV1byeqezuVyDUI85PEvZhurN2eJYm7EKICGzlyJP379yc6OprAwMB8jy1cuJCGDRvec0IB4O3tba4QhSjTysJrbOHChYwbN4758+cTExND1apVzXbse6XX67GxkWV6heUlpufw3u9RnE3IwNXexnRzc7DBy8mOAHcHqnjYU8XdAUdbq3y97DZWGsZ1qsXAZgFsjIrjt8jL7DmfxN7z1wBw1lnzUudaDGsVJIXcKgD5DReTzHEXQpQ4RYHcDMvcFOWuQuzduzc+Pj6Eh4fn256Zmcny5csZOXIkiYmJDB48mCpVquDg4EBYWBg//vjjbY8bFBRkGtILcOrUKdq1a4dOp6Nu3bps2rTpXp/NArZt20bz5s2xs7PDz8+PN998k7y8mwV8fvrpJ8LCwrC3t8fT05MuXbqQkaGubbt161aaN2+Oo6Mjbm5utGnThujo6GLHJB4weY2Z7t/vaywjI4MVK1bwwgsv0Lt37wJxAqxZs4amTZui0+nw8vKif//+psdycnJ4/fXXCQgIwM7Ojlq1arFgwQIAwsPDcXNzy3es1atX5+tZfPvtt2nYsCELFy6kevXq2NnZoSgK69ev56GHHsLNzQ1PT0969+7NmTNn8h3r4sWLDBo0CA8PDxwdHWnatCm7d+/m/PnzaLVa9u3bl6/9rFmzCAwMRLnL352ouI7FpvLI7H9YHXmZwxdT2HEqgd8Px7Jkdwxf/XWGd36LYtTiffSYuYPQaRuo/85GXloWSXKmnnr+LqwZ+xAvdq6Fr4uOoS0DWTG6FRFvdmLywyG81r0OW1/twKi21SVpryCkx72YbvS4yzruQogSo8+ED/0tc+63LoPtnYfRWltbM2zYMMLDw5k6darpA/XKlSvJzc3lqaeeIjMzkyZNmvDGG2/g4uLCH3/8wdChQ6levTotWrS44zmMRiP9+/fHy8uLXbt2kZqaett5uXfj0qVL9OrVi+HDh7N48WKOHz/Os88+i06n4+233yY2NpbBgwfzySef0K9fP9LS0tixYweKopCXl0ffvn159tln+fHHH8nNzWXPnj2y3mxZJK8xoHivseXLl1OnTh3q1KnDkCFDGDduHFOmTDHF+ccff9C/f38mTZrE999/T25uLn/88Ydp/2HDhrFz506+/PJLGjRowLlz50hISLirc99w+vRpVqxYwc8//4yVlfr5LCMjgwkTJhAWFkZGRgZTp06lX79+REZGotVqSU9Pp3379lSuXJk1a9ZQqVIlDhw4gNFoJCgoiC5durBo0SKaNm1qOs+iRYsYPny4vNbFba07EsuEFYfI0hsI8nTgte7B5OQZSM7Uk5Kl3uJSs7lwLZOL17JIztSTlp1n6mV/oUMNbKwKJuT+bvaMalvdAlckLE0S92K6mbhLj7sQomIbMWIEn376KVu3bqVjx46AOnS2f//+uLu74+7uzquvvmpqP27cONavX8/KlSvvKqnYvHkzx44d4/z581SpUgWADz/8kJ49e953zHPmzCEgIIDZs2ej0WgIDg7m8uXLvPHGG0ydOpXY2Fjy8vLo37+/aXhyWFgYAElJSaSkpNC7d29q1KgBQEhIyH3HIsSdlObX2IIFCxgyZAgAPXr0ID09nS1bttClSxcAPvjgAwYNGsQ777xj2qdBgwYAnDx5khUrVrBp0yZT++rV7z0xyc3N5fvvv883/P+xxx4rEKePjw9RUVGEhoaydOlSrl69yt69e/Hw8ACgZs2apvajRo1i9OjRfPbZZ9jZ2XHo0CEiIyP55Zdf7jk+UTEYjQpfbDnFF1tOAdC2lhezBzfG1eH2UzfSsvVcvJaFp5NtkRXkRcUmiXsx3RgqL3PchRAlxsZB7ZWz1LnvUnBwMK1bt2bhwoV07NiRM2fOsGPHDjZu3AiAwWDgo48+Yvny5Vy6dImcnBxycnLuujDWsWPHqFq1qimhAGjVqtW9XU8hx2zVqlW+nrM2bdqQnp7OxYsXadCgAZ07dyYsLIzu3bvTrVs3BgwYgLu7Ox4eHgwfPpzu3bvTtWtXunTpwhNPPIGf392tUytKEXmNAff/Gjtx4gR79uwxJbPW1tYMHDiQhQsXmhLxyMhInn322UL3j4yMxMrKivbt299VnEUJDAwsMGf/zJkzTJkyhV27dpGQkIDRqHa0xMTEEBoaSmRkJI0aNTIl7f/Vt29fxo4dy6pVqxg0aJDpuQ8KCipWrKJsMhgVLl7L5FRcOifj0zgTn0FGTh4KCooCRgWupudw6EIyACMfqsbEnsFYF9Jz/l/OOhtC/KQugyiaJO7FJFXlhRAlTqO5q6G0pcHIkSMZO3YsX331FYsWLSIwMJDOnTsDMGPGDD7//HNmzpxJWFgYjo6OjB8/ntzc3Ls6dmHzSYs7VFVRlALHuHEejUaDlZUVmzZtIiIigo0bNzJr1iwmTZrE7t27qVatGosWLeLFF19k/fr1LF++nMmTJ7Np0yZatmxZrLjEAyavMeD+X2MLFiwgLy+PypUr5zuWjY0N165dw93dHXt7+yL3v91jAFqttkBser2+QLvCvqDo06cPAQEBfPvtt/j7+2M0GgkNDTU9J3c6t62tLUOHDmXRokX079+fpUuX5qsJICqGf04nMHPzSY5cSrmrUba2Vlo+6BfK400DHkB0oqKQSgbFZFrHXRJ3IYTgiSeewMrKiqVLl/Ldd9/xzDPPmD7479ixg0cffZQhQ4bQoEEDqlevzqlTp+762HXr1iUmJobLl2/2jO7cubNY8datW5eIiIh8SUFERATOzs6mJESj0dCmTRveeecdDh48iK2tLatWrTK1b9SoERMnTiQiIsI09FaIklLaXmN5eXksXryYGTNmEBkZabodOnSIwMBAlixZAkD9+vXZsmVLoccICwvDaDSybdu2Qh/39vYmLS3NVBQS1F76O0lMTOTYsWNMnjyZzp07ExISwrVr1/K1qV+/PpGRkSQlJRV5nFGjRrF582bmzJmDXq/PV1RPlG/RiRk8t3gfT83fzd7z18jWG7G11hLi58KjDf15pWtt3usbyvt9Q/mgXygf9gvjo/5hrBvfVpJ2YXbS415MshycEELc5OTkxMCBA3nrrbdISUlh+PDhpsdq1qzJzz//TEREBO7u7nz22WdcuXLlrueFd+nShTp16jBs2DBmzJhBamoqkyZNuqt9U1JSCnzQ9/DwYMyYMcycOZNx48YxduxYTpw4wbRp05gwYQJarZbdu3ezZcsWunXrho+PD7t37+bq1auEhIRw7tw55s2bxyOPPIK/vz8nTpzg5MmTDBs27G6fLiHuWWl7jf3+++9cu3aNkSNH4urqmu+xAQMGsGDBAsaOHcu0adPo3LkzNWrUYNCgQeTl5bFu3Tpef/11goKCePrppxkxYoSpOF10dDTx8fE88cQTtGjRAgcHB9566y3GjRvHnj17Cq1a/1/u7u54enoyb948/Pz8iImJ4c0338zXZvDgwXz44Yf07duX6dOn4+fnx8GDB/H39zdNEwgJCaFly5a88cYbjBgx4o699KLsS8/JY85fp5m/4xy5BiNWWg1DWwYyrFUggZ6OWGmlMKF48KTHvZjsrGU5OCGEuNXIkSO5du0aXbp0ybeO85QpU2jcuDHdu3enQ4cOVKpUib59+971cbVaLatWrSInJ4fmzZszatQoPvjgg7vad+vWrTRq1CjfberUqVSuXJm1a9eyZ88eGjRowOjRoxk5ciSTJ08GwMXFhe3bt9OrVy9q167N5MmTmTFjBj179sTBwYHjx4/z2GOPUbt2bZ577jnGjh3L888/f0/PlxD3qjS9xhYsWECXLl0KJO2gFoaLjIzkwIEDdOjQgZUrV7JmzRoaNmxIp06d2L17t6nt3LlzGTBgAGPGjCE4OJhnn33W1MPu4eHBDz/8wNq1a01L3L399tt3dT3Lli1j//79hIaG8vLLL/Ppp5/ma2Nra8vGjRvx8fGhV69ehIWF8dFHH5mq0t8wcuRIcnNzGTFixB3PK8q2S8lZdP1sG3O2niHXYOShml6se6ktbz9Sj+reTpK0C4vRKBVsEcrU1FRcXV1JSUnBxcWl2MdbsjuaSav+pVtdX+YNa3rnHYQQ4g6ys7M5d+4c1apVQ6eTyrLlxe1+r+Z+bxK3f07lNSbu1QcffMCyZcs4cuTIbdvJ31bZ9/aao4RHnKeymz3T+tSla11fWfpPmEVx3+st2uO+fft2+vTpg7+/PxqNhtWrV99xn23bttGkSRN0Oh3Vq1fn66+/LvlAb+NGcbrsPOlxF0IIIYQoT9LT09m7dy+zZs3ixRdftHQ4ooSl5+Tx0/6LAEzvH0a3epUkaRelhkUT94yMDBo0aMDs2bPvqv25c+fo1asXbdu25eDBg7z11lu8+OKL/PzzzyUcadFM67jnyhx3IYQQQojyZOzYsTz00EO0b99ehslXAKsOXCQ9J4/qXo48VNPL0uEIkY9Fi9P17NmTnj173nX7r7/+mqpVq5qW4QgJCWHfvn383//9H4899lgJRXl7N9Zxz86TxF0IIYQQojwJDw+/q0J4ouxTFIXvdkYDMKxVIFqZyy5KmTJVnG7nzp1069Yt37bu3buzb9++QtfzBMjJySE1NTXfzZzsbWQ5OCGEEEIIIcqynWcSOR2fjqOtFY81qWLpcIQooEwl7leuXMHX1zffNl9fX/Ly8khISCh0n+nTp+Pq6mq6BQSYd01FO1PiLnPchRDmVcFqh5Z78vssfeR3IsxN/qbKru92ngegf+MqOOtsLBuMEIUoU4k7UKBAxI3/IIsqHDFx4kRSUlJMtwsXLpg1nhtD5WUddyGEudjYqB8YMjMzLRyJMKcbv88bv19hOTeW+srNzbVwJKK8kdd52XQpOYtNUXGAOkxeiNLIonPc71WlSpW4cuVKvm3x8fFYW1vj6elZ6D52dnbY2dmVWEw6GSovhDAzKysr3NzciI+PB8DBwUGq2pZhiqKQmZlJfHw8bm5uBdaHFg+etbU1Dg4OXL16FRsbG7TaMtePIUoZeZ2XbUt2RWNUoHUNT2r5Ols6HCEKVaYS91atWvHbb7/l27Zx40aaNm1qsW82b8xxz5Gh8kIIM6pUqRKAKXkXZZ+bm5vp9yosS6PR4Ofnx7lz54iOjrZ0OKIckdd52ZOtN7Bsrzoid1irIMsGI8RtWDRxT09P5/Tp06b7586dIzIyEg8PD6pWrcrEiRO5dOkSixcvBmD06NHMnj2bCRMm8Oyzz7Jz504WLFjAjz/+aKlLMPW45xqMGIwKVlKBUghhBjcSCx8fnyKLb4qyw8bGRnrgShlbW1tq1aolw+WF2cjrvGz643AsSRm5+Lvq6BLiY+lwhCiSRRP3ffv20bFjR9P9CRMmAPD0008THh5ObGwsMTExpserVavG2rVrefnll/nqq6/w9/fnyy+/tNhScHBzjjuo39g52pWpQQxCiFLOyspKPggKUUK0Wi06nc7SYQghLGjx9aJ0T7UMxNpKps2I0suiWWaHDh1uW32zsHUz27dvz4EDB0owqnujs775gVoSdyGEEEIIIUq35Mxctp28ypZj8Ry6mIKttZZBzcy78pQQ5iZZZjFptRpsrbXk5hnJzpN57kIIIYQQQjwoeoOR/dHXyNYb0Gg0aDWgQZ26mpNnIEtvICvXQLbeQGJGLv+cTmB/9DWMt/QdPt0qEE+nkitmLYQ5SOJuBrrriXtWrlSWF0IIIYQQ4kGIupzKKysPcSw29Z73rePrTMdgHzrW8aZ5NY8SiE4I85LE3Qx0NlakZufJknBCCCGEEEKUML3ByNytZ/hyyynyjAouOmsCPBxQFFDANBXXzsYKnbUWe1sr7G2scLC1pmFVNzrW8aaKu4NlL0KIeySJuxnY215fEi5PEnchhBBCCCHuVXpOHucTMohOzOR8YgbRiRkkZeQS6OlILR8navk6UdPHmSsp2byyMpJ/L6m97N3r+fJ+3zC8nWWouyjfJHE3gxsF6rJlLXchhBBCCCGKdCoujQ1Hr3A5JZvY5CxiU7KJTckmJevulj7VaEBRwNXehncfrccjDfzRaGQ5ZlH+SeJuBjeWhJM57kIIIYQQQhSUmZvHF5tPMf/vcxiMha8q5eloS6CnA0GejgR6OuLhaMO5hExOxadxOj6d2JRsFAW6hPjwYb8wfFxkOUdRcUjibgZ2Ntd73GWovBBCCCGEEPlsiorj7TVHuZScBUD72t40CHDDz1WHn6sOfzd7/Fx1OOtsbnuc1Gw9adl5+LvqpJddVDiSuJuBvY0MlRdCCCGEEOXPmavpZOUaqOfvcs/JcnRiBu//cYxNUXEAVHG3591H69Ep2Pe+YnHR2eByh+ReiPJKEnczMA2Vl6ryQgghhBCijDMaFbadvMq3O84ScSYRgAAPex5tUJm+jfyp6eN82323n7rK4p3R/HUiHkUBa62GZ9tV58VOtUxFnYUQ90YSdzPQXe9xz5HEXQghhBBClFHZegOrDl5iwd/nOB2fDoCVVoOdtZYLSVnM/us0s/86TT1/FzrU8cbdwRYXe7UX3MXemqjLqfywK5rziZmmY7av7c2kh0Oo7Vt0si+EuDNJ3M3gZlV5SdyFEEIIIUTZciEpkyW7Y1i+N4ZrmWp1d2c7awY1D2B4m2p4ONiy6Vgcvx68xLaTVzl6OZWjl1OLPJ6zzprHmwQwpGVVqns7PajLEKJck8TdDG4M+ZE57kIIIYQQoiwwGhV2nE7g+53n2XJcHdIOUNnNnmfaBDGwWUC+YnGPNPDnkQb+JGXksvZILMevpJKWnUdKlp7ULD2p2XmmhL1vI38cbCXNEMKc5BVlBnYyx10IIYQQQpQyiqLw5/F45u84x6XkLPIMRvRGhTyDkZw8I5m3LGX8UE0vhrUKpFOwD9ZW2iKP6eFoy5CWgQ8ifCHELSRxNwMZKi+EEEIIIUoLo1FhY1Qcs/48dfsh7XbWPNakCkNbBVJDhrQLUapJ4m4GOlkOTgghhBBCWFBunpHziRkcvpjC/B1nOX4lDQAHWyuGtgqkW11fbKy0WGu12FhpsLbS4ueqM32OFUKUbpK4m4H99aHy2XnS4y6EEEIIIUqW0ahw+FIKfx6L49iVNM7EpxOdlInBqJjaONlZM7x1ECMeqoaHo60FoxVCmIMk7mZg6nHPlcRdCCGEEEKYn95gZM+5JDYcvcLGo3FcSc0u0MbJzpoaPk50rOPNM62r4epgU8iRhBBlkSTuZmBK3KXHXQghhBBCmNnaI7FMWf0viRm5pm0OtlZ0rONDsyB3avo4U9PHCV8XOzQajQUjFUKUlKJLRoq7JnPchRBCVGRz5syhWrVq6HQ6mjRpwo4dO27bfsmSJTRo0AAHBwf8/Px45plnSExMND0eHh6ORqMpcMvOLtjDKER5lq038NaqI4xZcoDEjFw8HG15omkVFjzdlANTuvLVU40Z3qYaD9XyopKrTpJ2Icox6XE3A92N5eBkqLwQQogKZvny5YwfP545c+bQpk0bvvnmG3r27ElUVBRVq1Yt0P7vv/9m2LBhfP755/Tp04dLly4xevRoRo0axapVq0ztXFxcOHHiRL59dTpdiV+PEKXFybg0xi09yIk4tcjcCx1qMKFrbWxus1SbEKL8ksTdDGSovBBCiIrqs88+Y+TIkYwaNQqAmTNnsmHDBubOncv06dMLtN+1axdBQUG8+OKLAFSrVo3nn3+eTz75JF87jUZDpUqVSv4ChLCgbL2B6MRMcvIMGIwKRgWMikLU5VSmrztGtt6Il5Mtnz3RkHa1vS0drhDCgiRxN4MbiXuODJUXQghRgeTm5rJ//37efPPNfNu7detGREREofu0bt2aSZMmsXbtWnr27El8fDw//fQTDz/8cL526enpBAYGYjAYaNiwIe+99x6NGjUqMpacnBxycnJM91NTi167WoiSFpeaza+RlzAY1dWH7G2t0NlYodVoOHM1nRNX0jgRl8b5hAxuKQRfQNtaXsx4ogE+zjLaRIiKThJ3M7A3zXGXHnchhBAVR0JCAgaDAV9f33zbfX19uXLlSqH7tG7dmiVLljBw4ECys7PJy8vjkUceYdasWaY2wcHBhIeHExYWRmpqKl988QVt2rTh0KFD1KpVq9DjTp8+nXfeecd8FyfEfdp3PonRPxwgIT3nzo0BF501TnbWaLUatBoNVloNNlYaBjSpwqiHqqPVyrx1IYQk7mZhmuMuibsQQogK6L8FsRRFKbJIVlRUFC+++CJTp06le/fuxMbG8tprrzF69GgWLFgAQMuWLWnZsqVpnzZt2tC4cWNmzZrFl19+WehxJ06cyIQJE0z3U1NTCQgIKO6lCXFPftwTw9Rf/0VvUKjt60SDKm5k6Q1k5RrI0hvQG4xU9XAkuJIzdSo5E1zJGW9nqQQvhLgzSdzNQHdLj/vtPqwIIYQQ5YmXlxdWVlYFetfj4+ML9MLfMH36dNq0acNrr70GQP369XF0dKRt27a8//77+Pn5FdhHq9XSrFkzTp06VWQsdnZ22NnZFeNqhLh/eoOR936PYvHOaAAeDvPj08fr42ArH7WFEOYhZSnNQGetJu5GBfSG20xUEkIIIcoRW1tbmjRpwqZNm/Jt37RpE61bty50n8zMTLTa/B8/rKzU91FFKfw9VFEUIiMjC03qhbAko1Hh6OUUhszfzeKd0Wg08Fr3Osx+spEk7ULcSlEg4TQY9JaOpMyS/1HMQGd78wNIdp4BW2v5PkQIIUTFMGHCBIYOHUrTpk1p1aoV8+bNIyYmhtGjRwPqEPZLly6xePFiAPr06cOzzz7L3LlzTUPlx48fT/PmzfH39wfgnXfeoWXLltSqVYvU1FS+/PJLIiMj+eqrryx2nULcEJ+azY5TCew4dZW/TyeQkJ4LgJOdNTMHNqRL3cJHmwhRIRn0cHQV7JwNsYegegcY8gtorSwd2U056ZB6CVIugI0jVG0JpXAEtSTuZmBrpUWjUb9Iys414KKzsXRIQgghxAMxcOBAEhMTeffdd4mNjSU0NJS1a9cSGBgIQGxsLDExMab2w4cPJy0tjdmzZ/PKK6/g5uZGp06d+Pjjj01tkpOTee6557hy5Qqurq40atSI7du307x58wd+faLiSkjPYd2/V7iQlMnFa5lcvJbFxWtZJGXk5mvnYGtF6xqevNEjmFq+zhaKVohSJisZ9ofD7m8g7fLN7We3wl8fQucpFgoM0GfB1ulw5k9IuQhZ1/I/7h0Crf4H9Z8A69IzBUujFDUurZxKTU3F1dWVlJQUXFxczHbckCnrydIb2P5aR6p6OpjtuEIIIcq/knpvqsjkORXFcTIujaELdhOXWrAyvEYD9Su78lAtL9rW8qZxVXcZbSnKptxMSDwNCSchOwUaPgk29sU/7sEfYN0bkJuu3nf0gRbPgb0H/HG9iOiTK6B29+Kf617FHYWfRsLVY/m327mCa2VIjskfd/PnoNlIcPAo9qmL+74kPe5mYm9rRZbeQHaeVJYXQgghhCirDsZc45nwvSRn6qnm5UiHOt5UcXcgwN2eKu4OVPV0wMlOPkKLEmY0wLE1EDEbcjPg8UXgE1J4W0WBXXPh1AZw8gXXKtdvAaBzhfR4SL8CaXHqvykX1fnmKTH5jxOzE/p/e//DxHMzYe2rELlEve9TV+25Dnv8Zs91/DHY+y388hyM3gFuVe/vXPdKUWDvfNgwCQw5alLe/QOoFAYulUF3PZHOToH938Hur9Xh83+9DztmQP95UPeRBxNrEeR/HTPRXf+mVdZyF0IIIYQom3acusrz3+8nM9dAwwA3wp9phpuDraXDEqXVpQNqj3Vm4i23JKjcGFq8ANb38beTlwOHfoR/voCksze3L+gOg36Aau3yt9dnwa9j4d+f7u8a7N3BqzZc3AdHVkJgG2j6zL0fJ+EUrHga4o+CRgsd34KHXoH/FCOl+wdwaT9cPqC2H7G++MPRFQWyk29+MWHIU0cO2NiDjYMaz+ZpcGKt2r5WN3h0Djh5FzyWzhXavAgtX4Cjq2HnLLWXvnLj4sVoBpK4m8mNJeGyciVxF0IIIYQojbJyDWyMukJ0YibVvByp5etENS9H7Kyt+ONwLOOXH0RvUGhby4uvhzTBUXrWK6aUS2oCZ+dU+ONXT8DGyXBqY+GPR62GyB/hkVkQ0Oz258q6pia9CSfV3ugjP6nJJ4DOTR2qfW47XNgF3/eHvnOh/uPq42lxsOxJuLQPtNbQ7nU1CU65ePOWk6L2LjtXUnvjnSuBsx941QLPWuDoqR7rny9g01R1iHvlJuBX/+6fr39/gTXj1CHmjj4wYEHBLxhusLaDJ76Dr9uqyfuGSfDw/xXe1mhQk+aYXXBhN2QlqcXujHnX/9Wrz19anNqLfidWttD1XWgx+s6jCqxs1Oc5bID6u3GtcufjlzD538hM7G6s5Z5ntHAkQgghhBDiBkVROHghmZX7LvL7ocuk5eTle9xKqyHQw4FziRkoiroG+2cDG2BnXYqqXosH5/w/8F1vsLKD4F7qMO8andXe88wktajZ3gWgGNRkuWorcPQGB0/1ZmWjDlu/egwWdIUWz0OnKeqXAIoC8VFqUbQzf8GVw5BxtWAMzv7Qeiw0flrdT/8KrHoOon6FX0ap1c9rdYWlgyD1oprgD/y+6GT5brQaB9ERcHI9rHwantt2c/h4UZLOqcn+sTXq/cCH1KTdudLt93Orqg7JX/q4Omw+K0n9okRroz5/Gg3ERcGFPZCbdvfXoHNTz21lq45E0GeBPlO9eQerX6TcyxcSoMbiXefe9ikhUpzOTPrP+YcDMcl8M7QJ3evd4Y9VCCGEuIUUUjM/eU4rFr3ByLYTVzkRl0ZOnpEcvYFsvYFsvZH9Mdc4HZ9ualvF3Z7mQR6cT8zgVHw6adk3E/knW1TlvUdDsdKWvqWgxAPy4+CbQ6pv0LlBzS5wepM6BxqgTi/o+h541Sx4jMwk2PCWOuQd1Lnmga3ViurpcQXbu1RWe8C9akOVZlC3b8Fh9kYjbJqiLqsG6pcGxjzwrKkWevOsUYyLviXub9qpXwzU6wcDFhXeM52dAtv/T50HbshVh6K3GQ8dJ4HVPfQLb3kPdhTR236DrTMENFeXaHOrql63lc3NJN/O5eZoAhvdPV3ugybF6UqJG0PlZY67EEIIIcSDcTo+nZX7LvDzgUskpBc9VFZno6VXqB8DmlahZTVPtNcTc0VRiE/L4VRcOrbWWpoFuaMphes3iwckOUbtcQZ4PBwu7IV/f1aHrt+YQ+4bqs7Trt6h6OM4eEC/r9Vh1r+9rBaBO7xcfczaHoIeghqd1GTUq3bRQ/JvpdWq53UNgPVvqkl7tfbqsHN79+Jcdf64ByyCRT3UtdcD20DzZ/PPIT+/Qx11kJmo7lO9oxqXb717P1/Ht6BSKCSeyT/83WgAt0D1+fGtV7rWfLcgSdzNRBJ3IYQQQoiSl603sObQZZbvvcD+6JvrL3s52dKuljeOdtbobLTYWVuhs9Hi52pPt3q+OOtsChxLo9Hg66LD16V099SJB2TfQlCMalJer5966/YenP8bTm9Wq7rXH3j3iWTNLjBmpzocPDtVPW7VlsUrxtZytJrMJpxQh9JbFfy7LpaAZuo88A1vwfqJEPGlWpU+Lzt/O89aasJeq9v9V6HXWqnPsbgrkribic7mRlV5meMuhBBCCGFuF5Iy+WF3NMv3XiA5Uw+o89M71vHm8aYBdAr2wcZK1lMX90mfDQcWqz83G3Vzu9YKqrdXb/fDzgkeern48d2qWlv1VlJajlHnux//XR2FcIPOFVyqQJOnoekI839pIG5LEnczkR53IYQQQgjz23MuiXnbz7LleBw3KjNVdrPnqZZVGdC4Cj7SW172xeyC6H+g+fN3N2y8JET9qg7/dqkCtXtaJobSQqOBAQvh3A61QJ2Tb5mYQ17eSeJuJqbl4CRxF0IIIYQottw8I59uOM63O86ZtrWt5cWwVkF0CvaRAnLlgdEA2z6B7Z+oQ9QTTkO/uZaJZe+36r9Nh99bgbXyytoOanWxdBTiFvJXaSY66xs97jJUXgghhBCiOKITMxj340EOX1QreD/RtArPtatBTR8L9caKO1MUtcL4f2mtC58TnhoLvzyrFju74dBSdQm2kD4lF2dhLkfCxb1qpfLGTz/YcwtxlyRxN5Obc9ylx10IIYQQ4n6tOXSZt345QnpOHm4ONnzyWH26yVK75nd2m5poV2tfcOmxu2HIU9chj9kFMTvVfzPiC7az1kHlJmpRtqqt1OXOLu5T1yXPTAQbR+j9ubq++T8z4beXoEpzcPa9/2vLSVfXFq/cFLxr37n93vnqv3UfBSef+z+vECVIEnczsb8+VD4nTxJ3IYQQQoh7ZTAqTF79Lz/uUYthNQty54tBjfB3s7dwZOXQ0VWwcrj6s727um542ACo2lpdduxO/poOEbNAn3HntnnZ6vz16H+ub9AA14sVVAqDAeHqWuh5OXB6C8QdgTXj4MnlBauVKwpkJICTd9HnS4uDJQPULxVA/dKgwWAIfUxd7uy/sq7BketLvTV/9s7XI4SFSOJuJqY57rmSuAshhBBC3Kvpa4/x454YNBoY17EmL3auhbVUiTe/uKOweoz6s42jmrjuX6TeXCqrFdBvl8DGHYVtH6k/61whoOXN3nSfYND8Z1h8etz1XvnrPfNJZ9TtzZ6Fbu/fLHhmbQf958G89nBqAxz4DpoMv3mcC3vU5cku7VN7xnt+WrBXPuE0/NAfkqPB1hn0mXBpv3rb8BbU7qHuW6PTzST+4BLIywLfMAhocV9PqRAPgiTuZiLLwQkhhBBC3J9le2KY/7dahO6LQY14pIG/hSMqpzKTYNmTakJbrT08tVJd9uvfnyDqN0i9BGtfhSpNwb9R4cfYdb14XHBveOL7O/fQ61zAqxY0HqreT49Xh+i7VinY1rcudJ4KGyfD+regWjt1jvymaXD0l5vton5Vh/p3/wAaPqX2zF/cD0sfV4ffu1eDIT+DnTMcWQmHfoQrR9Th88fWgEarJum1ut1cAq75qPtfj1yIB0ASdzMxLQcnQ+WFEEIIIe5axJkEJq/+F4CXu9SWpL2kGA3w80i4dh7cqsLj4Wovd42O6q3XDHXeedSvsP3/YNCSgsfISFQTYYBWY+9uWP1/3WkOecv/wYn1EP03fN9PLWJnyAE00Ogpdcj75rch9hD8+j84vALCHod1r6tfSPg1hKd+ujmcvtX/1NuVf9XYT26Aq8euz8vfqbaxc1WPIUQpJom7mchQeSGEEEKIe3MuIYMXfjhAnlGhTwN/Xuxc09IhlV9b3oUzf4K1PQxaWnC+t40OOrwFUWvg+O8QF6X2gN9q/yJ1zrpfQ3V4fEnQatUl4ea0Vr9kAAhqq/au+zW4fr8d7JwNW6fDuW3qDaBGZ3hiceFrwVcKVW9d34Fr0XBqo3q7uBfavgK2jiVzPUKYiSTuZnKzx12GygshhBBC3ElKpp6R3+0lJUtPwwA3Ph1QH40MVQZ9trqmuGIEB8/8N2sdWNlcX2LNWh3ynXRG7X2+cYs7qiahPnXBt576b26GWrEd4NHZalG4wvgEQ91H1F73HTNgwIKbjxn0N6uvt3yhZIeV3xgRsGceNHka6vTKfz4ra3hovLps3G8vqUvKNRgMj8xSn587cQ9U5/FLMTpRhkjibiY35rjnyHJwQgghhKjAkjNzORWfzvmEDKITMzmfmEFMUiYGo4K9jRX2tlbY21gRk5TJ2asZ+LvqmDesiakTpMLb/omaNBeHPjN/T/QNrV9Uq8ffTttX1MT96C/Q8S3wrKFuj/oV0mLB0Qfq9StefHejVhf1djueNeDp3yDtCrj4lXxMQliQJO5mcmM5OFnHXQghhBAVUUZOHl9uOcWCv8+RZ1Tuah8HWyvmP90MH2ddCUdnBooC2z5Rl1Lr9SlUa2v+c2QkwK6v1Z9r91DnpWcmQmYCZF5Th6kb9fn30bmpQ8j96qtD2CuFQW66OtQ9Pgri/oWrJyCwNXSeducY/BpAre5qZfe/P4NHv1K33yhK12ykOje+tNBoJGkXFYIk7mZimuMuibsQQgghKhBFUdhwNI53fzvK5ZRsACq72VPNy5FATweqeTlS1cMBW2st2XoDWXoDWblGsvUGOtTxprp3IfOR7+TKEdBnQZVmD6YSuCEP/nj5ZgXyHwfBsDVQpUnR+xgNoL3HUQT/zFTXRvdrCIOXFX5tiqIe26gHYx7YOhXervJtYruTdq+qifuhZdD+DbUS/KV9YGULTUfc/3GFEPdNEnczkeXghBBCCFHRXEjKZNqao/x5PB6AKu72vPNIPTqH+N5hz/ukz4Y/34OdXwGK2jvcahzU63t3c5vv95w/j1QLtmm04FVHrUq+5DF4Zh34hORvn5MG69+EIz9B1/egxXN3d560K7DnW/XnjpOK/kJCo1HneFuV4Mf4gObqcnHntsE/X0BWsro9dMCdq8ILIUqEJO5mYmctQ+WFEEIIUXHsOZfEsIW7ydYbsbHS8Fy76oztWAt72xKaq375IKwaDVePq/etbNVibL+MUpcHa/E8NBgE9h7mS2qzU9V1z8/vUM/32AJ16bTFj8Kl/epyZSPWg3uQ2j5mt7qk2o1q6OteB+dKasG3O9nxmToUvkozqNXVPPEXR7vX1MT9wPegXP9823K0ZWMSogKTxN1MbrxJ5eQZMRoVtFqpiiqEEEKI8ikhPYexSw+QrTfSLMid6f3DqOnjXDInM+jVYm3bP1WHhjv6wCNfQpXmsG+BWnk89SJsmqLeAGwcQecCdi5qBfHQx9QK5Hez5JfRCOlxkBwNa1+DK4fB1hkGL4Vq7dQ2T/0Ei3qpPe+L+8Lw32H/d7Dj/9Rq8K4B4N8Ijq2BX54FF3+o0rToc6ZcVJdag9v3tj9IQQ9BQEu4sEu9X7X1zeXYhBAPnCTuZnJrJdScPGPJfdsshBBCCGFBBqPCy8sjiU/LoaaPE9+NaI6DbQl9pEy/Cj8OVHu3Aer2hYc/A0dP9X7719VK6UdWqsXT4o+q2/UZ6i0tFhJOqOt1//EK1H1UXTasaktIjoGEk5BwSv036RykXIDUy/kLwDl4wZCfwb/hLds8YOgqWNgdrp2DLxuBIVd9rP4g6PWJ+uXBsifVueJLB8KozeBRrfDr3P5/6v6BD0H1DmZ8AotBo1F73Zc8pt6X3nYhLEoSdzPRWWtNP2frDZK4CyGEEKJMOhWXxicbTtC9XiUea1y5wNrqX/11mh2nErC3sWLOU41LLmlPuagOSU88rVZOf3iG2nP+395oGx00HqreDHp1jnl2inrLSYXonXBoqTp8PXKJekMD3KbyvcZK7SX3DoaeH99cEu1WLn4wbDUs7KH20Otcoffnaow3DFgIi3qqvfZLHoeRG9Wk/1ZJ5+Dg9+rPnUpJb/sNNTtD46fVdeDrPGzpaISo0CRxNxNrKy3WWg15RoXsPJnnLoQQQoiyR28wMu7Hgxy/ksamqDg2HL3C9P5heDmpy39FnElg5uaTALzXN5TaviU0PD7hNHzfV+0Bd6kCw34Fr5p33s/KRk2Mb02Oq7VTe+ZjdqkJ/NHVakJvrQPPWuBVC7xqq8m5W1VwrQJOle5unrxHdbVA3dFVak++a+X8j9s5wZMrYH4XSDwFy56CR2eDS2X1Cwe4OQWgekd1ybbSRKNRpyUIISxOEnczsrexIi0nj6xcSdyFEEIIUfZ8s+0Mx6+k4WxnTXaegU1RcRyIvsb0/mE0rOrGiz9GYlTg8SZVGNCkSskEEXsYfugPGVfBsyYMXQ1uAcU7pkYDga3UW89P1bXRnf1Aq73zvnfiWUNdPq0oLn7w1Ep1WH1MBMxqrG539FG/JIiNVO93mlz8WIQQ5ZYk7mZkdz1xlyXhhBBCCFHWnI5P48stpwF4v18oNX2cmLD8ECfi0nju+/34utiRkJ5DHV9n3n00tGSCiNkFS56AnBSoFAZDVoGTt3nPYaMr2DNe0nzrquuyr38DEs+APhMy4tUbQO0ety9eJ4So8CRxNyPTWu4yVF4IIYQQZYjRqPDGz0fINRjpFOzDIw380Wg0rBnXhs82nmTejrPEpebgYGvFV081LplaPvHH4fv+alG5gJbw5HKwdzP/eSwlqA2M/hsUBbKuqdMAUi5CZhIEy/xxIcTtSeJuRvY2spa7EEIIIcqe73dFsz/6Gk521rzfN9RUkM7O2oqJvULoHOLLtzvO8lSLqtT0cSr6QEYDHF6urkXuVevuA8jNgJVPq0l74EPq0HJbh2JeVSml0dychy/Lqwkh7pIk7makk8RdCCGEEGXMxWuZfLz+OABv9AzG382+QJvm1TxoXs2jwPYCdn6lrqXuHgRj999dgTdFUZdqu3pcLQr3eHj5TdqFEOI+maEih7jBNFRe5rgLIYQQogxQFIVJq/4lM9dA8yAPnmpe9f4PlnIJtn6k/nztvLq2+t04+AMc+hE0WhiwwPxz2oUQohyQxN2MpMddCCGEEGXJ4p3RbDt5FVtrLdMfC0OrLcYa4hsnqUPdba73lm//FAx5t9/nyr+w9npF9o6TIOih+z+/EEKUY5K4m9HNxF163IUQQghRui3dHcO0NUcBmNC1NjW8bzN3/U7O/KWuZa7RwpBfwN4Dks7A0V+K3icnTZ3XnpcNNbvAQxPu//xCCFHOSeJuRjcS9yzpcRdCCCFEKbZ0dwxvrToCwMiHqvF8u+r3f7C8HFj7mvpzs2fVtdJbj1Xvb/9ULVj3X4oCv42HxNPg7A/95plnTXUhhCin5H9IM9JZ35jjLom7EEIIIUqn/ybtkx8OMVWRvy87v4LEU+DoAx3fUrc1exZ0bpBwEqJW52+vKLD5bfj3J9BYweOLwNHz/s8vhBAVgCTuZnSjxz1HEnchhBBClEK3Ju0j2pghaU++oPaqA3R77+a66zoXaPU/9edtn4LxlmmE2z6Gf2aqPz/8f1C15f2fXwghKghJ3M3I3laGygshhBCi9LmcnMW0X//Nl7RP6V3MpB1gw0TQZ0LV1lB/YP7Hmj8Hdq5w9Rgc/03d9vfnsHW6+nP36dB0RPHOL4QQFYSs425GN4fKS3E6IYQQQlje2avpfL3tDL8cuESeUQHMmLQf+x2O/aYOd3/4/+C/x7N3g5aj1R72bZ+oy8Vtflt9rPM0aDWmeOcXQogKRBJ3M7KT5eCEEEIIUQpcTs7ig7XHWHskFkXN12ldw5MxHWrSpqZn8ZP2a+fh1+uJd+ux4Fuv8HYtRsPOORD3r9o7D9D+DWgrFeSFEOJeSOJuRvY3Evc86XEXQgghhGVk6w08s2gvJ+LSAOgS4suYjjVoXNXdPCfIy4WVz0B2ClRuCh0nF93WwQNaPAc7Zqj327wEHSaaJw4hhKhAJHE3I9NycLnS4y6EEEIIy/ho3XFOxKXh5WTH4hHNqevvYt4TbJoClw+oVeMfXwTWtrdv33ocxB6CKs3U3vbi9vYLIUQFJIm7Gels1DnuOXmSuAshhBDiwfvzeBzhEecB+L/H65s/aY/6FXZ/rf7c7xtwq3rnfezdYcjP5o1DCCEqGEnczUgnc9yFEEIIYSHxadm8tvIwAM+0CaJDHZ/7O1D6Vdj1lZqUV20FXnVAq4Wks/DrWLVN6xehTg8zRS6EEOJOLL4c3Jw5c6hWrRo6nY4mTZqwY8eO27ZfsmQJDRo0wMHBAT8/P5555hkSExMfULS3Z5rjLlXlhRBCCPEAGY0Kr608TGJGLsGVnHmjR3DhDfNybn8gQx6sGKYu2/b7yzCnJXxSDZY8AT8OhpxUCGgBnaea/yKEEEIUyaKJ+/Llyxk/fjyTJk3i4MGDtG3blp49exITE1No+7///pthw4YxcuRIjh49ysqVK9m7dy+jRo16wJEXzu76UHlZx10IIYQQD9KiiPNsO3kVO2stswY3Mo0CzOevD2F6Fdj+f0Uf6K/3ISYCbJ2hWjuwcYDsZDi1Aa4eB3sPGLAIrGxK7FqEEEIUZNGh8p999hkjR440Jd4zZ85kw4YNzJ07l+nTpxdov2vXLoKCgnjxxRcBqFatGs8//zyffPJJkefIyckhJ+fmt8upqalmvoqbZKi8EEIIIR60qMupfLzuOACTHw6hlq9zwUaHV6jrqQP8+Z6aeLd5KX+bkxvUnnaAR2dDvb5g0MOVwxCzC+KOQpPh4Fq5xK5FCCFE4SzW456bm8v+/fvp1q1bvu3dunUjIiKi0H1at27NxYsXWbt2LYqiEBcXx08//cTDDz9c5HmmT5+Oq6ur6RYQEGDW67iVzlqGygshhKh4SmLa288//0zdunWxs7Ojbt26rFq1qiQvocw6cjGFYQt3k2sw0iXEhyEtAws2unQA1oxTf67STP1301TY/c3NNskXYNXz6s/Nn1OTdlAT/MpNoNX/oO8cCGheYtcihBCiaBZL3BMSEjAYDPj6+ubb7uvry5UrVwrdp3Xr1ixZsoSBAwdia2tLpUqVcHNzY9asWUWeZ+LEiaSkpJhuFy5cMOt13MreVnrchRBCVCwlMe1t586dDBw4kKFDh3Lo0CGGDh3KE088we7dux/UZZUJO05dZdC8nSSk51LXz4VPBjRA89+l1tLiYPkQyMuG2j1gxEZo95r62LrXYX+42qv+0wjIugb+jaDb+w/8WoQQQtyexYvT/fcNRlGUgm8610VFRfHiiy8ydepU9u/fz/r16zl37hyjR48u8vh2dna4uLjku5WUG8vBSeIuhBCiorh12ltISAgzZ84kICCAuXPnFtr+1mlv1apV46GHHuL5559n3759pjYzZ86ka9euTJw4keDgYCZOnEjnzp2ZOXPmA7qq0u/XyEuMCN9LRq6BNjU9Wf58Szwc/7Oeel4OrBgKqZfAqzb0n6dWh+84SV1bHeC38fB9P7i4B+xc4fFwsLZ70JcjhBDiDiyWuHt5eWFlZVWgdz0+Pr5AL/wN06dPp02bNrz22mvUr1+f7t27M2fOHBYuXEhsbOyDCPu2bgyVzzMq5BlkuLwQQojyraSmve3cubPAMbt3717kMUGtaZOamprvVl7N33GWl5ZFojco9Gngz8LhzXDW/adYnKLAH6/Ahd1qQj7oR9C5qo9pNND1PXVIPAqcvz61oe9X4B70IC9FCCHEXbJY4m5ra0uTJk3YtGlTvu2bNm2idevWhe6TmZmJVps/ZCsrNVlWFKVkAr2dqyfgny/gxHqAfBVcs/MkcRdCCFG+ldS0tytXrtzTMeHB1rSxFKNR4YM/onj/j2MAjGhTjS8GNsTOupAK8ru/gYPfg0YLjy8Er5r5H9dooMfH0HSEer/NSxDSp4SvQAghxP2y6FD5CRMmMH/+fBYuXMixY8d4+eWXiYmJMQ19nzhxIsOGDTO179OnD7/88gtz587l7Nmz/PPPP7z44os0b94cf3//B38Bx35Ti7scWgqAnfXNpzMrV4bLCyGEqBhKYtrbvRwTHmxNG0vI1hsYt+wg3+44B8CbPYOZ0jsErbaQ5+ToKlj/pvpzl3egZpfCD6rVQu/P4dVT0PXdEopcCCGEOVh0ObiBAweSmJjIu+++S2xsLKGhoaxdu5bAQLUiamxsbL7iNsOHDyctLY3Zs2fzyiuv4ObmRqdOnfj4448tcwGBbdR/oyNAUdBqNdhZa8nJM8o8dyGEEOVecae9AdSvXx9HR0fatm3L+++/j5+fH5UqVbqnY4Ja08bOrnzOzU7OzOW5xfvZcz6JZ23W8T+PvbhVegc0NQo2PrcDfrk+BL7ZqJtz2W/HycfsMQshhDAvixenGzNmDOfPnycnJ4f9+/fTrl0702Ph4eFs3bo1X/tx48Zx9OhRMjMzuXz5Mj/88AOVK1toPdHKjcFaBxlXIeEkcHO4fE6eJO5CCCHKt5Ka9taqVasCx9y4cWORxyzPLiRl8tjcCPacT6KhXSxvWS/FLeU4/DgQ1r4O+uybja8cgWVPgiEXQh6Bnp+oQ+KFEEKUeRZP3Ms0a7ub66FG/wOAvY2s5S6EEKLiKIlpby+99BIbN27k448/5vjx43z88cds3ryZ8ePHW+ISLebfSyn0nxvBmasZ+LnYsaTKL2gUA7hVVRvs+Qa+7QTxx+FaNPwwAHJS1RGB/b8FbSFz34UQQpRJFh0qXy4EtlGrsZ7/B5qOMC0JlyVD5YUQQlQAJTHtrXXr1ixbtozJkyczZcoUatSowfLly2nRosUDvz5L0RuMvLBkP1fTcgiu5MyPbeJw/OMfdaTf07+rBXJXvwDxR2Fee3D0hvQr4FMXBi0FG52lL0EIIYQZaRSLlGO3nNTUVFxdXUlJSTHPmu7ntsN3fcDZDyYco8cXOzh+JY3vRzanbS3v4h9fCCFEuWf29yZR5p/T5XtjeOPnI3g52fHnS81wmd8GUi5A+zeh40S1UVocrB4NZ/5U77tUgVGbwMUCBXuFEELcVnHfl6THvbiqNAOtDaTFQtJZ7GSovBBCCCGKQW8wMuvP0wCMbl8dl72z1aTdNUBdtu0GZ1946mfYMw/O/qWuzS5JuxBClEsyx724bOyhchP15+gI7GWovBBCCCGK4ZcDF7l4LQsvJzuG1Fbgny/UB7p/ALYO+RtrtdByNDy5HLxrP/hghRBCPBCSuJtD0I1l4f4xVZWX5eCEEEIIca/+29uu+3MKGHKgWnu1UrwQQogKSRJ3c7ixnvv5f9BZX18OThJ3IYQQQtyjW3vbh3qfhhN/gNZalnYTQogKTua4m0NAC9BYQUoMfl7xgMxxF0IIIcS9ydfb3q4adlufVh9o/jz4BFswMiGEEJYmPe7mYOcE/g0BCMk9AsgcdyGEEELcm3xz2wOuwpUj6vJv7V61dGhCCCEsTBJ3c7k+XL5m5mFA5rgLIYQQ4u4VmNt+6Hv1gXr9wMHDgpEJIYQoDSRxN5egh9R/MiIBGSovhBBCiLt3a2/7Uw3c4N+f1QeaDLdkWEIIIUoJSdzNJaAFoMEj+wLeXCM7T3rchRBCCHFnOXkGvtxys7fd/vjPkJcF3iHXP18IIYSo6CRxNxd7N6gUBkBL7TGycyVxF0IIIcSdfb8zmkvJWfi62PFU86qwb5H6QJPhUkleCCEEIIm7eV0fLt9ce1x63IUQQghxR6nZemb/pfa2T+haG/v4gxB/VC1K12CghaMTQghRWkjibk7XC9S10B4jS3rchRBCCHEHX289Q3Kmnpo+TjzWuArsv97bXq8f2LtbNjghhBClhiTu5lS1FQC1tZfISIqzcDBCCCGEKM2upGSz8J9zALzRIxjr3FT49xf1wSbPWDAyIYQQpY0k7ubk6EmuZzAAPsn7MRoVCwckhBBCiNJq5uaTZOuNNA10p0uIDxxecUtRuuaWDk8IIUQpIom7mVlXU+e5N1GiuJScZeFohBBCCFHqpF8l5uhOftoXDcDEXsFo4OYw+abPSFE6IYQQ+VhbOoDyRhvYCvbNp6H2DOcSMgjwcLB0SEIIIYQoLYxGWNidqkln2GnryjGXh2iSYwPndRAfpRalqy9F6YQQQuQnibu5+YYCUFtzgZXxqbSr7W3hgIQQQghRalzcA0lnAPDWpOCd9gcs/ePm4/X6q0vMCiGEELeQofLm5lmTPI0Njpocki+ftnQ0QgghhChFjEd/BeB3Q0vmV/scmj0Lzv7XH9VAs1GWC04IIUSpJT3u5mZlTZpzDdxTj6PERwHdLR2REEIIIUoDRSEz8hecgA2ah5jc70lw0UGvT+HyQVAUqNLE0lEKIYQohaTHvQQYvEMAcEw+aeFIhBBCiIKCgoJ49913iYmJsXQoFUrMkR045VwhQ7GjXc+B+Lro1Ac0GqjcWJJ2IYQQRZLEvQTYVw4DwD/nLNl6g4WjEUIIIfJ75ZVX+PXXX6levTpdu3Zl2bJl5OTkWDqsck1vMLJnbTgA/zq2ZEDLWpYNSAghRJkiiXsJcAioD0AdzQXOJ2ZYOBohhBAiv3HjxrF//372799P3bp1efHFF/Hz82Ps2LEcOHDA0uGVS7O3nKJZ5g4A6nQagkaWexNCCHEPJHEvARrfegBU08Ry/kqShaMRQgghCtegQQO++OILLl26xLRp05g/fz7NmjWjQYMGLFy4EEVRLB1iuXD4YjJbtm0hUBtPnpUOt7Belg5JCCFEGSOJe0lw9iNT64y1xkhKzL+WjkYIIYQolF6vZ8WKFTzyyCO88sorNG3alPnz5/PEE08wadIknnrqKUuHWOZl6w1MWHGI7prdAFjX6gJ2ThaOSgghRFkjVeVLgkbDNedaOKQcwHDlX6C3pSMSQgghTA4cOMCiRYv48ccfsbKyYujQoXz++ecEBweb2nTr1o127dpZMMryYcbGE5yOT6OPbo+6oW5fi8YjhBCibJLEvYTovUIg5QAOUlleCCFEKdOsWTO6du3K3Llz6du3LzY2NgXa1K1bl0GDBlkguvIjW28gPOI8tTSXCOIyWNlCbVkmVgghxL2TxL2E2PmHwRnwzjyNoihShEYIIUSpcfbsWQIDA2/bxtHRkUWLFj2giMqno5dT0BsUBjjsByNQoxPoXCwdlhBCiDJI5riXEI/qDQGoocRwLVNv2WCEEEKIW8THx7N79+4C23fv3s2+ffssEFH5dOhCCgC9rG4Mk3/UgtEIIYQoyyRxLyF2/qEAVNJcI/rCBQtHI4QQQtz0v//9jwuFvDddunSJ//3vfxaIqHw6dDGZ6prLBOjPgdYa6vS0dEhCCCHKKEncS4qdM1etfAFIPh9p2ViEEEKIW0RFRdG4ceMC2xs1akRUVJQFIiqfDl9MoYf2em97tfZg727ZgIQQQpRZkriXoETHmgDkxR61cCRCCCHETXZ2dsTFxRXYHhsbi7W1lL8xh5RMPecSMugpw+SFEEKYgSTuJSjHU11WR5d03MKRCCGEEDd17dqViRMnkpKSYtqWnJzMW2+9RdeuXS0YWflx+FIylblKmPY8aLQQ/LClQxJCCFGGydfqJcjGLwzOgWfmaUuHIoQQQpjMmDGDdu3aERgYSKNGjQCIjIzE19eX77//3sLRlQ+HLiTT1Wq/eqdqa3D0smxAQgghyjRJ3EuQe7WGEAFV86IxGAxYWVlZOiQhhBCCypUrc/jwYZYsWcKhQ4ewt7fnmWeeYfDgwYWu6S7u3aGLKTyjvV6hX3rbhRBCFJMk7iXIN6geuYo1TppsLkWfonL1YEuHJIQQQgDqOu3PPfecpcMot87HXKC59vpUueBelg1GCCFEmSeJewnS2thywSqAGsZzJJ07KIm7EEKIUiUqKoqYmBhyc3PzbX/kkUcsFFH5cCUlm7DMXVjbGjH6hKJ1D7J0SEIIIcq4+0rcL1y4gEajoUqVKgDs2bOHpUuXUrduXfn2/j8SHGtQI+0cuZeOAIMtHY4QQgjB2bNn6devH0eOHEGj0aAoCgAajQYAg8FgyfDKvMgLyXS7Pr9dG9LbwtEIIYQoD+6rqvyTTz7JX3/9BcCVK1fo2rUre/bs4a233uLdd981a4BlXZa72stum3TMwpEIIYQQqpdeeolq1aoRFxeHg4MDR48eZfv27TRt2pStW7daOrwyLyr6Cu21h9Q7Mr9dCCGEGdxX4v7vv//SvHlzAFasWEFoaCgREREsXbqU8PBwc8ZX5llXCgXAI10qywshhCgddu7cybvvvou3tzdarRatVstDDz3E9OnTefHFFy0dXtl39k/sNbmk2/tDpTBLRyOEEKIcuK/EXa/XY2dnB8DmzZtNc+GCg4OJjY01X3TlgEtQAwB88y5CXo6FoxFCCCHUofBOTk4AeHl5cfnyZQACAwM5ceKEJUMr84xGheqJ2wDIrtEDrk8/EEIIIYrjvhL3evXq8fXXX7Njxw42bdpEjx49ALh8+TKenp5mDbCsqxpYgxTFAWuMZMVGWTocIYQQgtDQUA4fPgxAixYt+OSTT/jnn3949913qV69uoWjK9vOXU2hnaIuA+fWqJ+FoxFCCFFe3Ffi/vHHH/PNN9/QoUMHBg8eTIMGaq/ymjVrTEPohcrN0Y7TmkAAEs8ctHA0QgghBEyePBmj0QjA+++/T3R0NG3btmXt2rV8+eWXFo6ubLt0+C88NOmkaZyxDmpt6XCEEEKUE/dVVb5Dhw4kJCSQmpqKu7u7aftzzz2Hg4OD2YIrL+Lta0LWMXIuHbF0KEIIIQTdu3c3/Vy9enWioqJISkrC3d3dVFle3B+bU+sAOOPRloZWsuquEEII87ivHvesrCxycnJMSXt0dDQzZ87kxIkT+Pj4mDXA8iDTrTYANlePWjgSIYQQFV1eXh7W1tb8+++/+bZ7eHhI0l5cikL1hK0A5NToYdlYhBBClCv3lbg/+uijLF68GIDk5GRatGjBjBkz6Nu3L3PnzjVrgOVBXuWmAFRKiQR9lmWDEUIIUaFZW1sTGBgoa7WXAP3lw/ga48hSbPFt1MvS4QghhChH7itxP3DgAG3btgXgp59+wtfXl+joaBYvXixz4wrhFtSIy4oHtkoOnNth6XCEEEJUcJMnT2bixIkkJSVZOpRyJWn/KgB2auoTWMnLwtEIIYQoT+5r8lVmZibOzs4AbNy4kf79+6PVamnZsiXR0dFmDbA8qB/gxhZDI4ZYbyHv+Dqsa3ezdEhCCCEqsC+//JLTp0/j7+9PYGAgjo6O+R4/cOCAhSIr227Mbz/t0YFOMu1ACCGEGd1X4l6zZk1Wr15Nv3792LBhAy+//DIA8fHxuLi4mDXA8sDP1Z4DuhYMyduC4cR6rJXPZF1XIYQQFtO3b19Lh1D+ZF3DI+04ALnVu1o4GCGEEOXNfSXuU6dO5cknn+Tll1+mU6dOtGrVClB73xs1amTWAMsLfUBbss7aYp9xGeKOQqVQS4ckhBCigpo2bZpZjzdnzhw+/fRTYmNjqVevHjNnzjRNqfuv4cOH89133xXYXrduXY4eVYu4hoeH88wzzxRok5WVhU6nM2vsZnP1JACXFE9qVwuybCxCCCHKnfua4z5gwABiYmLYt28fGzZsMG3v3Lkzn3/+udmCK0/qBfnyj7GeeufUhts3FkIIIcqI5cuXM378eCZNmsTBgwdp27YtPXv2JCYmptD2X3zxBbGxsabbhQsX8PDw4PHHH8/XzsXFJV+72NjY0pu0A/q4YwCcMfrTIMDNssEIIYQod+4rcQeoVKkSjRo14vLly1y6dAmA5s2bExwcbLbgypNGAW78aWys3jkpibsQQgjL0Wq1WFlZFXm7F5999hkjR45k1KhRhISEMHPmTAICAopcZcbV1ZVKlSqZbvv27ePatWsFetg1Gk2+dpUqVbrv630QUi9GAXDBqgo+znYWjkYIIUR5c19D5Y1GI++//z4zZswgPT0dAGdnZ1555RUmTZqEVnvf3weUW2FVXJmgqNMIlAt70GQkgqOnhaMSQghREa1atSrffb1ez8GDB/nuu+9455137vo4ubm57N+/nzfffDPf9m7duhEREXFXx1iwYAFdunQhMDAw3/b09HTTsnUNGzbkvffeu+10vJycHHJyckz3U1NT7/o6zMEQp85vT3Wsjkbq2AghhDCz+0rcJ02axIIFC/joo49o06YNiqLwzz//8Pbbb5Odnc0HH3xg7jjLPAdba1x9gziaGEg9bTSc3gQNBlk6LCGEEBXQo48+WmDbgAEDqFevHsuXL2fkyJF3dZyEhAQMBgO+vr75tvv6+nLlypU77h8bG8u6detYunRpvu3BwcGEh4cTFhZGamoqX3zxBW3atOHQoUPUqlWr0GNNnz79nr50MDe75DMA5HnUtFgMQgghyq/76hr/7rvvmD9/Pi+88AL169enQYMGjBkzhm+//Zbw8HAzh1h+NKrqxhbj9d6Ck+stG4wQQgjxHy1atGDz5s33vN9/e5gVRbmrXufw8HDc3NwKVLlv2bIlQ4YMoUGDBrRt25YVK1ZQu3ZtZs2aVeSxJk6cSEpKiul24cKFe76O+6bPxjn7MgC2lUIe3HmFEEJUGPeVuCclJRU6lz04OJikpKRiB1VeNQxw40/D9Xnup7eAQW/ZgIQQQojrsrKymDVrFlWqVLnrfby8vLCysirQux4fH1+gF/6/FEVh4cKFDB06FFtb29u21Wq1NGvWjFOnThXZxs7ODhcXl3y3BybxNFqMJCuO+PoFPLjzCiGEqDDuK3Fv0KABs2fPLrB99uzZ1K9fv9hBlVeNqrpzSKlOouICOakQs9PSIQkhhKiA3N3d8fDwMN3c3d1xdnZm4cKFfPrpp3d9HFtbW5o0acKmTZvybd+0aROtW7e+7b7btm3j9OnTdzUsX1EUIiMj8fPzu+vYHqgEdSm4M4o/Qd5OFg5GCCFEeXRfc9w/+eQTHn74YTZv3kyrVq3QaDRERERw4cIF1q5da+4Yy43qXo446Wz5y9CQAVbb1ery1dpZOiwhhBAVzOeff55vKLtWq8Xb25sWLVrg7u5+T8eaMGECQ4cOpWnTprRq1Yp58+YRExPD6NGjAXUI+6VLl1i8eHG+/RYsWECLFi0IDQ0tcMx33nmHli1bUqtWLVJTU/nyyy+JjIzkq6++uo+rLXm5V45jC5w2VqaHp6OlwxFCCFEO3Vfi3r59e06ePMlXX33F8ePHURSF/v3789xzz/H222/Ttm1bc8dZLmi1GhoGuLHlTKPrift66C6F/IQQQjxYw4cPN9uxBg4cSGJiIu+++y6xsbGEhoaydu1aU5X42NjYAmu6p6Sk8PPPP/PFF18Ueszk5GSee+45rly5gqurK40aNWL79u00b97cbHGbU9blKGyBSzYBuDrYWDocIYQQ5ZBGURTFXAc7dOgQjRs3xmAwmOuQZpeamoqrqyspKSkPdv7bdZ9tPMHCPw9zSPc8Vhhg3AHwrPHA4xBCCFF6POj3pkWLFuHk5MTjjz+eb/vKlSvJzMzk6aefLvEYStqDfE5TPm+Ba8pxPnCdxqSXJ5TouYQQQpRNxX1fkgXXH7BGVd1Jx4FIq3rqhpMbLBuQEEKICuejjz7Cy8urwHYfHx8+/PBDC0RUhhkNOKadU3/2qmPZWIQQQpRbkrg/YA0C3AD4I7uBuuHkOssFI4QQokKKjo6mWrVqBbYHBgYWGNYu7iA5BmtjDjmKDS5+MoJOCCFEyZDE/QHzcLQlyNOBTcbry8Kd/wcyEi0blBBCiArFx8eHw4cPF9h+6NAhPD09LRBRGZagLlF3VqlEoPeDn4InhBCiYrin4nT9+/e/7ePJycnFiaXCaBjgxupEX+Id6+CTcQJO/AGNh1k6LCGEEBXEoEGDePHFF3F2dqZdO3V1k23btvHSSy8xaNAgC0dXxiScAOCMUplqUlFeCCFECbmnxN3V1fWOjw8bJgnonTSq6s7qyMvssGnFY5yAqF8lcRdCCPHAvP/++0RHR9O5c2esrdWPAkajkWHDhskc93uUG6cuBXdG8ae9l4OlwxFCCFFO3VPivmjRopKKo0JpeH2e++KUhjwGcHYbZF0D+3tbO1cIIYS4H7a2tixfvpz333+fyMhI7O3tCQsLMy3hJu6e/voa7nG2VXHWyVJwQgghSsZ9reMuiifEzwVbay2HsnzI9Q/GNuk4nFgPDQdbOjQhhBAVSK1atahVq5alwyi7FAWbJHWOe667PI9CCCFKjhSnswBbay1hldVpB2e8O6sbo361YERCCCEqkgEDBvDRRx8V2P7pp58WWNtd3EZmIrb6FIyKBhuf2paORgghRDkmibuF3Bguv1nTUt1wZgtkp1ouICGEEBXGtm3bePjhhwts79GjB9u3b7dARGXUVbUw3UXFiyo+HhYORgghRHkmibuFNKrqBsCqC84oXrXBkAsnN1g2KCGEEBVCeno6tra2Bbbb2NiQmipfIt+16xXlTyuVqeYlFeWFEEKUHEncLaRtLW9cdNacTczkuHsHdWPUagtGJIQQoqIIDQ1l+fLlBbYvW7aMunXrWiCiMur6Gu5nFH+CZCk4IYQQJcjiifucOXOoVq0aOp2OJk2asGPHjtu2z8nJYdKkSQQGBmJnZ0eNGjVYuHDhA4rWfFztbRjTsSYAH52vo248vRly0i0YlRBCiIpgypQpvPfeezz99NN89913fPfddwwbNoz333+fKVOmWDq8MkMfdwxQe9yDZCk4IYQQJciiVeWXL1/O+PHjmTNnDm3atOGbb76hZ8+eREVFUbVq1UL3eeKJJ4iLi2PBggXUrFmT+Ph48vLyHnDk5jG8dRDfRZxnW0olUtyr4Jp1EU5vgnr9LB2aEEKIcuyRRx5h9erVfPjhh/z000/Y29vToEED/vzzT1xcXCwdXpmhxJ8EIMk+CAdbWahHCCFEybFoj/tnn33GyJEjGTVqFCEhIcycOZOAgADmzp1baPv169ezbds21q5dS5cuXQgKCqJ58+a0bt36AUduHjobKyZ0rQ1o+CmribpRqssLIYR4AB5++GH++ecfMjIyOH36NP3792f8+PE0adLE0qGVDbkZ2GZcAsDoKUvBCSGEKFkWS9xzc3PZv38/3bp1y7e9W7duREREFLrPmjVraNq0KZ988gmVK1emdu3avPrqq2RlZRV5npycHFJTU/PdSpP+jatQx9eZ1TnN1A0nN0JupmWDEkIIUSH8+eefDBkyBH9/f2bPnk2vXr3Yt2+fpcMqG67Pb09QXPD28bNwMEIIIco7i43rSkhIwGAw4Ovrm2+7r68vV65cKXSfs2fP8vfff6PT6Vi1ahUJCQmMGTOGpKSkIue5T58+nXfeecfs8ZuLlVbDGz3rMCI8VV1ORp+gLg0X0sfSoQkhhCiHLl68SHh4OAsXLiQjI4MnnngCvV7Pzz//LIXp7sWthemkorwQQogSZvHidBqNJt99RVEKbLvBaDSi0WhYsmQJzZs3p1evXnz22WeEh4cX2es+ceJEUlJSTLcLFy6Y/RqKq2MdH1pU82Sdobm6Ye98SL1s2aCEEEKUO7169aJu3bpERUUxa9YsLl++zKxZsywdVtl0fSm4M0apKC+EEKLkWSxx9/LywsrKqkDvenx8fIFe+Bv8/PyoXLkyrq6upm0hISEoisLFixcL3cfOzg4XF5d8t9JGo9EwsVcIvxlaqRvOboXP68EPj8G/P4M+26LxCSGEKB82btzIqFGjeOedd3j44YexsrKydEhllnJV1nAXQgjx4Fgscbe1taVJkyZs2rQp3/ZNmzYVWWyuTZs2XL58mfT0m0umnTx5Eq1WS5UqVUo03pLWMMCNKqFteD53PCftQkExqsvD/TQC/q827PnW0iEKIYQo43bs2EFaWhpNmzalRYsWzJ49m6tXr1o6rDLJcL2i/BnFn0BPWQpOCCFEybLoUPkJEyYwf/58Fi5cyLFjx3j55ZeJiYlh9OjRgDrMfdiwYab2Tz75JJ6enjzzzDNERUWxfft2XnvtNUaMGIG9vb2lLsNsXusezBZa0C3lLQ48shnavQauAZCTAuteh4wES4cohBCiDGvVqhXffvstsbGxPP/88yxbtozKlStjNBrZtGkTaWlplg6xbDDkob12BoA0p2robGTkghBCiJJl0cR94MCBzJw5k3fffZeGDRuyfft21q5dS2BgIACxsbHExMSY2js5ObFp0yaSk5Np2rQpTz31FH369OHLL7+01CWYVTUvR55qoa5fP+XvbIwdJsFLh6FSfbUH/thvFo5QCCFEeeDg4MCIESP4+++/OXLkCK+88gofffQRPj4+PPLII5YOr/S7ehytUU+6osPeK9DS0QghhKgANIqiKJYO4kFKTU3F1dWVlJSUUjnfPTE9hw6fbiUtJ4/PBzagX6MqsOMz2PIOVO8Iw1ZbOkQhhBBmVhremwwGA7/99hsLFy5kzZo1FonBnEr0Od3zLax9le2GMDY0nssH/cLMe3whhBDlTnHflyxeVV7k5+lkxwsdawDw6foTZOsNUPdR9cFz2yEj0YLRCSGEKK+srKzo27dvuUjaS1zMTgD2GutIYTohhBAPhCTupdCINtXwd9VxOSWbRf+cB88aUCkMFAMc/93S4QkhhBAVl6JA9PXEXQmWpeCEEEI8EJK4l0I6Gyte6VYHgDl/nSYpIxfq9lUfjFptsbiEEEKICi85BtIuo1esiDTWIEh63IUQQjwAkriXUv0aVaaunwtpOXl8ueUU1OunPnB2G2QmWTY4IYQQoqK6Pkz+iFKNbOwI8Cj7q9oIIYQo/SRxL6W0Wg1v9QoB4Idd0ZxTKoHvjeHyf1g4OiGEEKKCumV+u4OtFXbWshScEEKIkieJeyn2UC0vOtTxJs+o8Mn64zeL1MlweSGEEMIybsxvNwbL+u1CCCEeGEncS7mJPUPQav6/vfuOj6rK/z/+mrRJDwlphBJCL6EmlICAWBDEgg1Qwa7Ligiy61rQr4qsWPbnWlawAa6CioooK0UCKl1Qem8JhISEEEIa6Zn7++OSQAwtkmQyyfv5eMwjkzv33pzPJHDmc8+5nwOLd6SwyWeAuTHuF02XFxERqWmnTkDaXgB+t7XBQ4m7iIjUECXutVzbUB/u7hUOwNMr8jCCO4CtGPYutnPLRERE6pkjvwKQ69eaDHywuupjlIiI1Az1OA7gb4Pa4O/pyr5jOWzyvtLcqOnyIiIiNev0/e0ZQVEAuOv+dhERqSFK3B1AA083nryuHQAvxbUyNx78GfIy7NcoERGR+ub0/e3H/bsD4OGmxF1ERGqGEncHMaJHUyIb+7ItP5QUawTYimDvIns3S0REpH4ozIXkLQAca9ANAHdNlRcRkRqiHsdBODtZeOmmSAC+yDGv9LPtK7CVXPjAvAyzmI6IiIj8eUm/mzVmfBtz0jUEQMXpRESkxihxdyBR4f7c2r0xC229zA1xP8N/esDG/0JxwZkdDQMS1sO8h+CNVvCvVvD1fZC0yS7tFhERcXgJZmE6mvUmr8gGgFWJu4iI1BAXezdAKufpIe24aucxni+6j+c8v8WafhD+9zj8/ArEjAV3P/jtI0jZXv7AnfPNR0R/6DseWl4NFot9ghAREXE0h9eaX5vFkJ9nJu4qTiciIjVFibuDCfZxZ/zVrfnnokH8VHI1S6+Mx2vj+5B9FGKfP7OjiztE3g49HwInV1j7DuyYB/ErzUfjaBj9rZnoi4iIyPmVFEPib+bzZjHkbTdvU/Nw08RFERGpGepxHNB9fZvTNsSHpFxn/pHUD8ZvgZv+AyGdIKgdXPsyTNwNw96DsG4QGgm3fgiPb4HeY8HVy7xXb8kz9g5FRESk9ju2HQpzwOoHwe3JLzYTd424i4hITVHi7oBcnZ341x1dcHaysHB7Mgt3pUP30fDX1TB2PfR9HDwDKh7YoCkMfgVGzQMssGUO7FFlehERkQsqu7+9Fzg5U3D6HnctByciIjVFibuD6tTEj7FXtgTg+e93kJZTcJEjzhIeA33Gmc//N15V50VERC6k7P723gDkFZ4ecVdxOhERqSFK3B3YY1e1pl2oD+mnCnn+ux0YhnHpBw+cBEHt4VQqLHzCrET/R7npkLqn6hosIiLiaAzjrBH3PgBlU+WtLvoYJSIiNUM9jgNzczGnzLs4WVi8I4UftiVf+sGu7nDLdHBygV3fm4XrShXlwao34a3OMK2XudyciIhIfZQeZ17kdnYz68ZwZsRdU+VFRKSmKHF3cJGN/Rg7sBUA//f9Do5nV2LKfFg36P+k+Xzh3yDrKGz7ylwbfvlLUJh9+rWJEPdL1TZcRETEEdhKoNMd0G6oedEbyC/WcnAiIlKzlLjXAWMHtqJ9I19O5hbx9Lxt5BeVXPrB/f4GjbpCfoaZsH/7MGQeAd/GcMsH5pJytmL46h44vq+6QhAREamdgtrAbR/DHZ+UbSrtZzXiLiIiNUWJex3g5uLE/zs9ZX75nlRufHc12xMzL+1gZ1czQXe2mkvduHnDVc/DuI3QZSTc/B406Qn5mfD5cPO+dxERkXqsNHF3d9XHKBERqRnqceqIDmG+fHRvNIHeVvan5nDLtDW8vWw/RSW2ix8c3A7u/ByufBYe3wz9/w6uHuZrru4w8nNo0AxOxsPcUVBcien4IiIidUxZ4q6p8iIiUkOUuNchA9sGs/SJ/lzfKZRim8G/l+3jtulrOZCaffGDW10DVz4F3sEVX/MOgru+AqsvHF4D/5sAJcVV3n4RERFHkFeauGuqvIiI1BAl7nVMgJcb793VnbdHdsXX3YVtiZnc/J817Dt2Ccn7hQS3hztmgcUZtn4O73aDddMgP6vivjYbJG+F3T+YRX1ERETqkPwiFacTEZGapcS9DrJYLNzctTFLnxhAVLg/pwpLGDN7IzkFlzlK3uoauOV98GwIGQnw4zPw747w4yRI/B1++9gsYvdGS/igP8y9G1b+q2qCEhGRWmvatGlERETg7u5OVFQUq1atOu++9913HxaLpcKjY8eO5fabN28eHTp0wGq10qFDB+bPn1/dYVwyFacTEZGapsS9Dgv1c+eD0VGE+roTd/wUT32zDcMwLu+knYfDEzvhxrchsA0UZMG6/8DHV5tLyu36HvLSwcVcMod1/1FBOxGROmzu3LlMmDCBSZMmsXnzZvr168eQIUNISEg45/5vv/02ycnJZY8jR44QEBDAHXfcUbbPunXrGDFiBKNHj2br1q2MHj2a4cOHs379+poK64JUnE5ERGqaxbjsTM6xZGVl4efnR2ZmJr6+vvZuTo3YlHCSER+so6jE4Lmh7XmoX4uqObHNBgeXm8l54kZo1AVaDICIAeYa8R8NhGM7zCXnrv6/qvmZIiJ1kCP3Tb169aJ79+5Mnz69bFv79u0ZNmwYU6dOvejx3333Hbfeeivx8fGEh4cDMGLECLKysli8eHHZfoMHD8bf358vvvjiktpVXe9pcYmNVpPMdm1+/lr8vdyq7NwiIlJ3XW6/pEvF9UD3Zv48N7QDAFMX72FDfBWNgDs5Qetr4Z7v4dlEuH8hDPgHNOsFLm4w8Flzv1/fh1NpVfMzRUSk1igsLGTjxo0MGjSo3PZBgwaxdu3aSzrHjBkzuOaaa8qSdjBH3P94zuuuu+6C5ywoKCArK6vcozrkF59ZrUVT5UVEpKYoca8n7okJ56YuYZTYDB77fBOp2fnV/0PbXm+OvBedgjVvVf/PExGRGpWWlkZJSQkhISHltoeEhJCSknLR45OTk1m8eDEPPfRQue0pKSmVPufUqVPx8/MrezRt2rQSkVy60mnyAFYXfYwSEZGaoR6nnrBYLLx6WyfahHiTml3AY3M2k5FbWN0/FAZOMp9v+Biyj114/6I8SI+HhPUQtwJKiqq3fSIiUiUsFku57w3DqLDtXD755BMaNGjAsGHDLvuczzzzDJmZmWWPI0eOXFrjK+ns+9svJUYREZGq4GLvBkjN8XRzYfqoKG56dzUbDqVzzZsrmTKsI4MjG1XfD211DTTpAYm/weo3YchrZ14zDNjyOax7DzKPmIXuzhbeF0bNA1eP6mufiIj8aYGBgTg7O1cYCU9NTa0wYv5HhmEwc+ZMRo8ejZtb+fvEQ0NDK31Oq9WK1WqtZASVdyZx1zR5ERGpORpxr2daBnnz+cO9aRXsTVpOAWNmb+LRORs5nl1QPT/w7FH332dCZpL5PDMJ5twB3z8KqTvPJO0u7tCgGbh6wuE18M2DUHKZy9g5gj2LYMHjqsAvIg7Fzc2NqKgoYmNjy22PjY2lT58+Fzx2xYoVHDhwgAcffLDCazExMRXOuXTp0ouesyZoDXcREbEHjbjXQ12aNmDh41fw7vIDTF9xkEXbU1h78AQv3tiRYd0aV/0PbHGlOXp+eA2s+pd53/uPk8xk3dkKVz4N7W4AnxCw+prJ/qHV8NmtsHch/DABbnrX3F4XFRfAgscg9wSkx8Gob83ifiIiDmDixImMHj2a6OhoYmJi+PDDD0lISGDMmDGAOYU9KSmJTz/9tNxxM2bMoFevXkRGRlY45/jx4+nfvz+vvfYaN998M99//z3Lli1j9erVNRLTheRpDXcREbEDjbjXU1YXZ/5+XVu+H9uXDo18ycgtYsLcLbyzfH/V/zCL5UyF+d9nwoJxZtLeOBrGrIJ+EyGoDbj7nUnOm18Bt88EixNs/gyWT676dtUWuxaYSTvAoVWw6O/mbQQiIg5gxIgRvPXWW0yePJmuXbuycuVKFi1aVFYlPjk5ucKa7pmZmcybN++co+0Affr04csvv2TWrFl07tyZTz75hLlz59KrV69qj+diSqfKqzCdiIjUJK3jLhSV2Hh72X7+8/MBAMZd1YqJ17ap+qI7/70J4leY0+EHToKYseB0kRGLjf+F/z1uPr9uKsQ8WrVtqg1mDoGEtdDyKoj7BQwbDH4Vev/V3i0TkRqivqnqVdd7unRnCo98tpFuzRow/9G+VXZeERGp27SOu1w2V2cn/n5dWyZd3x6Ad386wGtL9lLl13Ru+QCufBbGrIa+j188aQeIuheu/j/z+Y/PwOq3qq7afOqei1e6r27HdplJu8UZbn4Prn3Z3P7js7A/9sLHiohIjStdx133uIuISE1S4i5lHu7fghdv7ADA+ysOMmXh7qpN3n0bwZVPQWDryh13xUTofXqkfdkL8P4V5sj05Uj8Hab3gQ/6QfbF1xquNhtnmV/bDgHfMHMWQrfR5qj71/dD6m77tU1ERCo4ezk4ERGRmqJeR8q5r28EU4aZhYJmrI7nhQU7q37kvbIsFrjuFbjxHfBsCMf3wKc3w9xRcPJw5c9ns8Hip8AogZxjZoJsjzXjC0/B1i/N5z1O3+dpscDQN81ifoXZ8PkIOHWi5tsmIiLnlK/idCIiYgdK3KWCUb3Def22zlgs8Om6w3ywMs7eTTIT2qh7YdxG6DXGnFq++3/wXk+z2N2+pVCUf2nn2v41JP0Orl7g5mNOVV/+UvW2/5zt+MYs0ucfARFXntnu4gbDPwP/5pBxGL66B4oLa759IiJSQdmIu6bKi4hIDVLiLuc0vEdTJt9sjry/vmQP6w7WklFfD38Y8ppZjb55PyjOh02fwud3wBst4ev7zIS4MPfcxxeegmUvms/7TYRh75nP175rVne/VIYBRXmQlwE5qZBxBE4chLQD5iyA7BRzTfaCbHOE/1x+n2l+jX4AnP7wT9GrIdw517ywcHi1Ks2LiNQSeYWn73HXiLuIiNQgreMu5zWqVzO2JGQwb1Mi477YxMLH+xHi627vZplCOsK9/zOXT9u1APYshOyjsHO++QhqD6Pnm/fVn23N2+Z+DZpBzGPg6m5+Xfcf+O5RCO4Aga3O/3PTDsDWz2HrXMhKvLS2eofCjW9D28FntiVtguQt5jr2Xe8+93HB7cwl8T4fDpv+a8bc6y8V90veCnuXQJcR5ii9XJhhQEkhuFjt3RIRcUD5xRpxFxGRmqcRdzkvi8XClGGRtAv1IS2nkLFzNlFUcp7RY3uwWCCiPwz9FzyxEx76ySxk5xUEx3fDzOsgPf7M/hlHzMQdzOrtrqcvQlzzIjTrY95T/tVoc1S+VHEhZB2FjZ/AjEHwnyhY9f8qJu3OVrD6gtUPXDzMqfylclLgixGw5BkoLjC3/T7D/NpxmDm6fj5tBsGg05XmlzwNB5afeS03HX54Aj4YAL+8Au/1ghVvnPkZ1S37GMy+zYyrILtmfublKCkyL7hM7wNTm8JPU8xZEyIilaDidCIiYg9ax10u6lDaKW58dzXZBcU80DeC/ztdeb7WOnnILF538pA52j16PoR0gG8egB3zzMJv9y00E/9S2Snwfj84lQp+Tc1R2byTUHSq/LktTtDqGuh6F7QYCK6e4Oxa/lylbCVQlAs//RPWTze3NeoKN/wbZl0PxXnwwI/QrPeF4zEM+H4sbJljXhh4KBYOrYafXjbbCNCwFZw4YD4PaAHXv2G2szp9fZ85uwHArxnc9A60HFi9P/PPKMyFzbPN2yEyE8q/5t8crv8XtL624nHZKZCVBGHdz/37FcdhGOb/B8lbzX+Thu3MA6BpLwhub9cmqm+qetX1nj7z7Ta+2HCEv13bhnFXV3KVFBERqbcut19S4i6XZOnOFB75bCMA79zZjQ6NfNmelMH2xCy2J2WQfqqQ127rTHTzADu39LTsFPjsFkjdBe4N4MqnzRFrLPCXFdCoS8VjDq2B/95oVps/m8UJgtpBl5HQeQT4hFa+PXsWwfePnkm0AYI7wl/XXFpSWFxgtu3IenByAVvxmXNc/7p5MWLHPPhxkjnCD9D+JujxkHlhoDLTwguyzfv0g9qaFyXOZX8szLndnFngGwaZR8zt3e+BQVPA3e/i8Sx43Jyy3vNhaBZT9clxfhZs+BB+nQ65aeY2z0Do/VczYV/6vHnbBJjv1cBnzYsf8SshbgWk7TVfu+HfZh0CqVmGYRZnbBD+5/42Etabt9Ik/g6Jv535GziflldDn3HQ4kq7XKhR31T1qus9fWLuFuZvTmLS9e15uH+LKjuviIjUbUrcK0kfjv68Vxfv4f0VB8/7uo/Vhc8f7k2nJhdJ2mpKbrp5f3jib2e2db8Hbnr3/Mcc22Umcx7+Zx5Wv4rF4/6MzCSY95BZxR7Mkd6eD1/68Tmp8NFVZpJs9YOrJkH0g+B8VqmK/Cz45VVY//6ZCxCuXtBigDkCX5qU5GeahfXyMyD3hFlY7/geOL7XHGUGs/jf3d+cuaWgVGEuTOsFGQlmfYArnzGr8m/40HzdtzEMm27+zPP5cZJZV6BUo67mGvYdhplV9S+mIAfy0s3ZEX9MsvIyYP0H8Os0Mz4waxr0eRy6jQJXj9PnyDbfq1+nV7xYczafRvD4lorvg1SvxU+Zf8etroFh74N30KUf+/MrsOK18tucXCG0k7mkpMXp9MMChTnmDJbS0feQSPPvOvK2S/tbrCLqm6pedb2nYz7byJKdKbw8LJLRvcOr7LwiIlK3KXGvJH04+vOKS2zcN+s3Vh9Iw93ViY5hfnRqbD7m/n6EDfHp+Hu68tVfYmgd4mPv5poKcmDu3RD3i1mh/fFN4B1sv/aUFJsJZcZhc2S6NIm8VCcPw95FEHn7hROZlB2w7j04sMyc/l9ZFiczkWl3A9zx3/IXB2JfgDVvgW8TGLserN7m9kNrzCn9J+PBxR3uWQDNelU894HlMPtW83m7G8w2Fp9eys+nEfT/uzlT4Hzys+DDKyH9oHkBI7QTNOoMoZ0hPc5M9gqyzH0D20C/v5tJmPN5anGmbIeFfzNnMwS2NesmRPQ3p09/dJVZz2DIG9Drkcq8g/VHevzpopDfmn93GIDl9AUVi3nxq9vd5jKOvmGXds4Dy8z6CaW8Q+CWDy7tVozNs82/QzBnUjSLgSY9zL+T8118SY8zL+Bsnm1OpQdzls3tM82ikDVAfVPVq6739L5ZG/hl73HeuL0zd0Q3rbLziohI3abEvZL04ejyFBbbOJqRRxN/D1ycz4xCZ+cXMerj9WxNzCTYx8o3Y/rQrKGnHVt6luIC+H0WNI6Cpj3s3ZqaZbPBse1mIrR/GSRuMAvpeTQwp7O7NzCfB7Qwp8YHtTOT3ZTtZuJUUgBdR8HN/zETsWM74YP+5lT9kV9Au+vL/7zCXPjmfti3BDwC4KFl0LDlmddPpZnF4XKOmcn50P8Hp07Axpmw4eMz0/xvmwGdbj93TN+NhS2zLxx3UHsY8KQ5gu90iZWfi/IrJna/zYCFEx1/1L240LzFoipmjoBZmHDbXDNZP7r50o5xcoFOd5ij2aGR598v7yRMi4HsZOh4K6TuNotNYoG+4+Gq585/C8fBn81bOGzF5gWbq5+vXFy56bBxFqybZk6td3GH614xb5Wo5unz6puqXnW9pyM/XMevcem8e2c3buxyiRejRESk3lPiXkn6cFR9Tp4qZOSHv7L3WDZN/D34ekwMjfwqOaIs1cswLj0B2bMQ5o4yR977PA7XvGRW6k/cYI6Uj5xz7uMKT8EnN8DRTeAfAQ/GmrMDDAO+uBP2LTZHth/5BdzOurhTXAjLXoRf3zOn9z/yCwS1KX/u3f8z24QF7vnenPacsg2St5lfsZgj4+1urJoktbgA3unuWKPuWUchYR2k7jET3tQ95uyEoHZw/2LzQs355GealfbPV8ehIBvWvGPe5lA6Mm1xMm+riLzVvE/cxWr+rjHMr0c3m/sfXnPmPC0Gmhdtzr6oU2reQ7D9a7Pg4l9Wmdt+fNZMqMG8ADf4NWgSXf5v+dgu8++zIMu8QHDrR38+2c45Dt+NMS94gTlyf9M75uyBaqK+qepV13t683tr2Hokg4/vieaaDiFVdl4REanblLhXkj4cVa/U7HyGv7+OQydyaRHkxacP9KSJfy0ZeZfKO3vacYsrT99y4A1jN4Bf4/Mfl5MKH19j3hLQOBru/R9s/dycku7sBg//ZE5d/qOSYvhsmFlULLgDPLT8THKffQymx5j35PedANe+VLWxno8jjbrvW2pe2Cg5z5KAra+DO78890WN9DiYOcSc9dA4GjreYi5X6NfEXEpv03/NmgCnjpv7h3U3V1focPOl3X6StBHW/gd2fWdeDHLzgWHvmceX2vkdfH2veTHgwVgzOS+163tYMM68uABmgckeD5m3jeRnmn9vWYnm0o73fFe5goznYrOZF5GWvQS2IrOeQte7zFkjp1LNv/GcVLOtj2+6vJ+F+qbqUF3v6eC3VrInJZvZD/biitaBVXZeERGp25S4V5I+HFW/xJO5DH9/HUcz8/F0c+bJ69pyT0xznJ20pJZDWvsuLH3uzPeDXzUrs19M2n6Yca059TmiPxzZYN7Lft0rZiG68zl7ab6uo8zkzjDMQoP7l0JIJzPxr6nCYdU96l54ylxW8HKnYu9dAl+NNiv1B3cwk96g9hDcDrDAFyPN93/AU2YF/bNlp8CMQeaFlj9q0tP8HZ7Yb34f0AKuedEchf4zbU6Ph+8ePVOksddf4drJp6fI9zaLDp5vmnvGEbPw3I55Zy5OlN7ucfIQNGwNDy4Fzypc3SJpk7mU5Mn4c79ucYbn0y57hof6pqpXXe/pgDd+5vCJXOb9NYao8FqykoqIiNR6StwrSR+OakbCiVz+/vVWNhxKB6Br0wa8elsn2oXqPXdIy16C1W+ao6wPxp6/0NsfHV4Hn958JslqeRXcPe/iSU78SvM4wwY3v2cmoz88Yd6f/8gvENLhssKptOoYdU+PNyufb5trFkC78lloO+TPJcN7F8Pc0ebIcIebzRoBf7wPfOuXMP8v5vORn0O7oebzvJMw63pz6UT/CBjxGST8ahacO7wWs9gc5m0JA56G6PvPf4/5pSophp8mw5q3ze+b9DBncsT9bM7EeOgiF2Zy02HzZ+bvpfRig2dDs6ZCQDUsz5WfBWvfMUfYvUPMGQZeQWee+0coca+Fqus97fXKMo5lFfDDuCuIbFxLVlEREZFaT4l7JenDUc2x2Qy++C2BVxftIbugGBcnC2MGtOTBKyLw96q5ZZakCpTeqxzUrvx96Zdi53z4+n7wCoQxq89///QfrXgDfp5iFgizOJn3VF9stL66VOWoe9ZRWPkGbPrULKJ2trBuZgLf+tryCXxBNmQmmuvQ/3E1gT2L4Kt7Tiftw+C2j8+fWJcusebmY85a8GsMn91iVtT3DoUHfzTXuC9ra7JZV8AoMaeJu1dxkrJnkXkveen0d2e30xdmLrGSu812uvDijxB137lvv3AQ6puqXnW9p11eWkpmXhHLJg6gVbB3lZ1XRETqNiXulaQPRzUvJTOfFxbs4MedxwBwcbLQt1UgN3RuxKAOofh5XubondR+afvNwl5elbgf1GYzK4QfXG5+H9EfRn9fdZXRK+vsUffbZ5pT3AuyzKTasJlTxy8UX34mrHgdfvv4zPJ3La+GKybAwZ9g/YdQdMrc3jga/MPN5f9OHjIrnJfybQJhXaFRV3MpvqXPm0l7x1vNgmwXmg1RUmTOZDi8xpxW3qCZ+f66+8H9S2p+JgOY8X11LyRvsd+FmVpAfVPVq673tM1ziykstrHm6ato3EAFWEVE5NIoca8kfTiynyU7knln+QF2JWeVbXN1ttCvdRB3RDXhmg4huDrbKSmT2ulUmrmWelGuORLr18R+bTl71P1cPAJgyGtmRfM/Tnffu9ic6p+dbH7fLAaueh6a9z2zT85xWPu2uSxecV7F81v9Tq9Pf47/siNvg1s+vLRbGHJS4YMBkH3U/N7V06zQ37TnxY+tLiXF5vt69mh/PaO+qepVx3tqsxm0eHYRABufu4aG3pdZBFFEROoNJe6VpA9H9nfweA6LtiWzcHsye1Kyy7YH+VgZ2aMpI3s20yiGnFGYa45oW2vBlNTd/4Mlz5oJstUHrL7m1xMHIW2vuU+ra+GGf0ODpuaFh8VPwY5vzNcCWprJfatrzn8ve/YxswK/xdlMZP3DoUG4WYAtPwtStpu3LSRvgZQdEB5jTt+/1LoDAIkbYdZg8329cy60vuYy3hSpCuqbql51vKe5hcV0+L8fAdg1+To83Srx705EROo1Je6VpA9Htcv+Y9nM35zEV78nkpZjFjBzssDAtsE8cW0bFf4Rx1BSBGveMqfClxSahdai7oOtX5jL11mcoM84uPIZcK0lF6VOHDS/nmstdalx6puqXnW8p+mnCun+ciwAca9cj5NWSxERkUt0uf2S5iWLXbUO8eEfg9ux9umreO+u7vRp2RCbAcv3pDLsvTW8vWw/RSU2ezdT5MKcXaH/k2bxvaa9oDAH1v3HTNpDIs316K+dXHuSdjATdiXtIpWSX1QCgJuzk5J2ERGpUUrcpVZwc3FiaOdGfP5wb3762wCu7xRKsc3g38v2cfv0tRxIzbF3E0UuLqitWeRtyBtmBf6Bk+Dhn6Fxd3u3TESqQN7pxN3dVR+fRESkZqnnkVqnRZA3793VnbdHdsXX3YWtiZkMfWcVM1fHY7PVqzs7xBE5OZnLxY1dDwP+ceH1yEXEoeSXJe7Odm6JiIjUN0rcpVayWCzc3LUxS58YQP82QRQU25j8wy7u++Q3Tp4qtHfzRESkHipN3D3clLiLiEjNUuIutVqonzv/vb8HU4ZF4u7qxMp9x7nh3dVsPZJh76aJiEg9k19k1lxxd1HiLiIiNUuJu9R6FouFUb3D+W5sX5o39CQpI4873l/HnPWHqWeLIoiIiB3l6x53ERGxE/U84jDahfqyYNwVXNcxhMISG5Pm7+BvX23lQGo2aTkFqj4vIiLVKk/3uIuIiJ242LsBIpXh6+7K+6Oi+GhVHK8t2cu3m5P4dnNS2es+Vhf8PF1pHexNdPMAosP96dK0gT5kiYjIZSubKq8+RUREapgSd3E4FouFR/q3pHOTBrz8wy6OpOeSlV8MQHZBMdkFxSSezOPnvccBcHW20DHMj2s7hDCyR1Maelvt2XwREXFQpSPuHkrcRUSkhilxF4fVu0VDFj7eD4ASm0FWXhEZeUWcyClgW2ImGw+f5LdD6aRmF7DlSAZbjmTw9rL93NClEff1aU7nJg3sG4CIiDiUAt3jLiIidqLEXeoEZycL/l5u+Hu5ERHoRXTzAB64IgLDMEg8mcfag2l8vj6BrYmZfLspiW83JdG1aQNG9GjKVe2CCfF1t3cIIiJSy2kddxERsRcl7lKnWSwWmgZ4MiKgGSN6NGNzwkn+u/YQC7cnl43CA3Rq7MfAdsFc3S6YTo39cHKy2LfhIiJS66g4nYiI2IsSd6lXujXzp1szfyYN7cBXvx8hdtcxtiZmsD0pk+1JmbyzfD++7i50adqALk0amF+b+hHsoxF5EZH6TsXpRETEXpS4S70U5GNl7MBWjB3YiuPZBfyyN5Wf9qSyan8aWfnFrNqfxqr9aWX7d27ix5RhkbovXkSkHlNxOhERsRcl7lLvBflYuSO6KXdEN6WoxMbelGy2Jmaw9fRU+v2pOWxLzGTYe2t4oG8EEwe1wdNN/3REROqbfBWnExERO1H2IXIWV2cnIhv7EdnYj7t7hQNwPLuAKQt38f2Wo3y8Op4lO1N45ZZO9G8TZOfWiohITSrQVHkREbETuyfu06ZN44033iA5OZmOHTvy1ltv0a9fv4set2bNGgYMGEBkZCRbtmyp/oZKvRXkY+Xtkd0Y1q0xz83fQeLJPO6ZuYEuTfzAYqGgqITCYhsFxTZ83F2ICvenZ0QAPSMCaOTnYe/mi4hIFdFUeRERsRe7Ju5z585lwoQJTJs2jb59+/LBBx8wZMgQdu3aRbNmzc57XGZmJvfccw9XX301x44dq8EWS302sG0wS5/oz7+W7uWTtYfYmph5zv32pGQzZ30CAE0DPLi6XQhPD2mnERoREQdXOlXeqqnyIiJSwyyGYRj2+uG9evWie/fuTJ8+vWxb+/btGTZsGFOnTj3vcSNHjqR169Y4Ozvz3XffVWrEPSsrCz8/PzIzM/H19b2c5ks9diA1m70pObi7OmF1ccbq6oSbsxPJmfn8diidDfHp7Dyaie30v66YFg2ZcV+07o0XkXNS31T1quM9vWXaGjYnZPDh6CgGdQytknOKiEj9cLn9kt2yiMLCQjZu3MjTTz9dbvugQYNYu3bteY+bNWsWBw8eZPbs2UyZMuWiP6egoICCgoKy77Oysv58o0VOaxXsQ6tgnwrbuzSFwZHmh7ns/CJW7kvjH99sZV3cCe6b+Rsz7++Bt1XJu4iII9JycCIiYi92m+uVlpZGSUkJISEh5baHhISQkpJyzmP279/P008/zZw5c3BxubTkZ+rUqfj5+ZU9mjZtetltF7kUPu6uDO3ciE8f7IWP1YUNh9IZPWM9mXlF9m6aiIj8CQVlVeWVuIuISM2y+01aFoul3PeGYVTYBlBSUsJdd93FSy+9RJs2bS75/M888wyZmZlljyNHjlx2m0UqIyrcnzkP98LPw5XNCRmM+ng9GbmF9m6WiIhUkorTiYiIvdhtzm5gYCDOzs4VRtdTU1MrjMIDZGdn8/vvv7N582Yee+wxAGw2G4Zh4OLiwtKlS7nqqqsqHGe1WrFardUThMgl6tykAZ8/3IvRMzawPSmTG95dTatgbzxcnfFwdcbdzZkATzd6RATQo7m/7oUXEamFtI67iIjYi92yAzc3N6KiooiNjeWWW24p2x4bG8vNN99cYX9fX1+2b99ebtu0adP46aef+Oabb4iIiKj2Notcjo5hfnzxcG/u/ng9iSfzSDyZV3Gnn8HV2UK3pv7EtGxITMuGdG3aQNMyRURqgTxNlRcRETux67DexIkTGT16NNHR0cTExPDhhx+SkJDAmDFjAHOae1JSEp9++ilOTk5ERkaWOz44OBh3d/cK20Vqq7ahPix9oj/rDp4gt7CY/KIS8opKyCu0kZCey7qDaRzNzGfDoXQ2HErn7eX7cXW2ENnYj+hwf6LCA+jWrAGB3lacnSreUiIiItXDMAwVpxMREbuxa+I+YsQITpw4weTJk0lOTiYyMpJFixYRHh4OQHJyMgkJCfZsokiVC/ByY2jnRud8zTAMEtJzWXvwBGsPnuDXuBMczy5gc0IGmxMy+GhVPAAWCzTwcMXfyw1/TzeCfawMaBPEdR1D8fdyq8lwRETqhYJiW9lzTZUXEZGaZtd13O1Ba+WKIzEMgyPpefx+OJ3fD59k46GT7EvN5nz/al2cLPRtFcjQzo24rkMofp6uNdtgEflT1DdVvap+TzNyC+k6ORaA/f8cgquzkncREbl0DruOu4hcnMVioVlDT5o19OTW7k0AKCqxkZFbxMncQk6eKuRkbiEHUnNYtD2FXclZrNh3nBX7jjPJeTtR4f4MaBNM/zaBdGjke84VG0RELte0adN44403SE5OpmPHjrz11lv069fvvPsXFBQwefJkZs+eTUpKCk2aNGHSpEk88MADAHzyySfcf//9FY7Ly8vD3d292uK4kNJp8i5OFiXtIiJS45S4izgYV2cngnysBPmUXy3hsatac/B4Dou2JbNwezJ7UrL5NS6dX+PSeW0JBPlY6dc6kN4RDYlq7k+LQC8l8iJy2ebOncuECROYNm0affv25YMPPmDIkCHs2rWLZs2anfOY4cOHc+zYMWbMmEGrVq1ITU2luLi43D6+vr7s3bu33DZ7Je2gpeBERMS+lLiL1CEtg7wZd3Vrxl3dmkNpp1i5/zgr9h5n3el75b/dlMS3m5IA8177qHB/osP9uaK1RuRF5M958803efDBB3nooYcAeOutt/jxxx+ZPn06U6dOrbD/kiVLWLFiBXFxcQQEBADQvHnzCvtZLBZCQ0Orte2VUboUnFWJu4iI2IESd5E6qnmgF80DvbgnpjkFxSVsPHSSVQfS2HjoJFsSM0g/VUjsrmPE7joGiyHYx8qVbYMY2DaYvq0D8XXX/fEicmGFhYVs3LiRp59+utz2QYMGsXbt2nMes2DBAqKjo3n99df57LPP8PLy4qabbuLll1/Gw8OjbL+cnBzCw8MpKSmha9euvPzyy3Tr1u28bSkoKKCgoKDs+6ysrMuMrrw8reEuIiJ2pMRdpB6wujjTp1UgfVoFAlBQXMKOpCw2Hk5nQ3w6aw6cIDW7gK9+T+Sr3xNxcbIwoE0QI3o0ZWC7YN3PKSLnlJaWRklJCSEhIeW2h4SEkJKScs5j4uLiWL16Ne7u7syfP5+0tDQeffRR0tPTmTlzJgDt2rXjk08+oVOnTmRlZfH222/Tt29ftm7dSuvWrc953qlTp/LSSy9VbYBnyddUeRERsSMl7iL1kNXFmahwf6LC/Xmkf0sKikvYEJ/OL3uP8/PeVOKOn2L5nlSW70klyMfK7VFNGB7dlIhAL3s3XURqoT/eZmMYxnlvvbHZbFgsFubMmYOfnx9gTre//fbbee+99/Dw8KB379707t277Ji+ffvSvXt33n33Xd55551znveZZ55h4sSJZd9nZWXRtGnTyw2tTIHWcBcRETtS4i4iWF2c6dc6iH6tg3j+hg4cSM3m698T+WZjIsezC5j+y0Gm/3KQyMa+p/cLJDo8ADcXjcSL1GeBgYE4OztXGF1PTU2tMApfqlGjRjRu3LgsaQdo3749hmGQmJh4zhF1JycnevTowf79+8/bFqvVitVqPe/rl0vF6URExJ6UuItIBa2CfXjm+vb8bVBbftpzjC9/O8KKfcfZkZTFjqQspv9yEE83Z3pFBBDs405RiY3CEhtFJTaKSwxah/hwc9cw2jfSetQidZmbmxtRUVHExsZyyy23lG2PjY3l5ptvPucxffv25euvvyYnJwdvb28A9u3bh5OTE02aNDnnMYZhsGXLFjp16lT1QVyiM8XpdMFSRERqnhJ3ETkvNxcnBkc2YnBkI1Kz81lzII2V+9JYtf84aTmF/Lz3+DmPW74nlfdXHKRtiA83dwvj5q6NadzA45z7iohjmzhxIqNHjyY6OpqYmBg+/PBDEhISGDNmDGBOYU9KSuLTTz8F4K677uLll1/m/vvv56WXXiItLY0nn3ySBx54oKw43UsvvUTv3r1p3bo1WVlZvPPOO2zZsoX33nvPbnGeKU6nEXcREal5StxF5JIE+7hzS7cm3NKtCTabwZ6UbNbFnSC/qARXZwuuzk5lRexW7T/Oz3uOs/dYNq8v2cvrS/bStWkD+rcJYkCbILo08cPlDwXvsvOLiE87RbHNoJGfO0He1gr7iEjtM2LECE6cOMHkyZNJTk4mMjKSRYsWER4eDkBycjIJCQll+3t7exMbG8u4ceOIjo6mYcOGDB8+nClTppTtk5GRwSOPPEJKSgp+fn5069aNlStX0rNnzxqPr1T+6XvcNVVeRETswWIYhmHvRtSkrKws/Pz8yMzMxNdX03hFqktmbhGLdyTz3ZYk1senc/b/NL7uLvRrHYSvhwsHj58i7vgp0nIKyh3vZIEgHyuhvu60b+TLiB5N6dq0gdaalzpJfVPVq+r39L2fD/DGj3sZHt2E12/vUgUtFBGR+uRy+yWNuItItfDzdGVkz2aM7NmMlMx8Vu47zop9x1l9II3MvCIWbk+ucEygtxU3Zwup2QUU2wyOZRVwLKuArYmZfPnbETqG+XJ3r3Bu7hqGl1X/fYlIzdFycCIiYk/65Csi1S7Uz53hPZoyvEdTSmwGWxMzWL0/jeISGy2CvIkI9CIiyAtfd1cAbDaDtFMFHMssICkjj6W7UvhhWzI7j2bx7PztvLJoN8O6hXFHVFM6N/HTKLyIVLt83eMuIiJ2pMRdRGqUs5OF7s386d7M/7z7ODlZCPZxJ9jHnU5N/BgcGcrzQzswb1Mic9YnEJ92itm/JjD71wRaBXtze1QTbunWmBBf9xqMRETqk7yyqvJK3EVEpOYpcRcRh+Dv5cZD/VrwQN8I1sWd4Kvfj7BkRwoHUnN4dfEeXl+yh65NG9DQ20oDD1f8Tj+83V1wc3HCzdkJNxcnrC5O+Hm4ERXur3XoReSSqTidiIjYkxJ3EXEoTk4W+rYKpG+rQLLyi1i0LZl5mxL57dBJNiVkXPJ5fNxduKZ9CNd1DGVAmyA83PRhXETO78xUeV3wExGRmqfEXUQclq/7mQJ4h0+cYltiJpl5RWTmFZF1+mtOQTGFxTYKS2zm12Ibh07kkpZTwPzNSczfnISHqzMD2gRxVftgrmwbRLCPptyLSHkqTiciIvakxF1E6oTwhl6EN/S6pH1LbAabE06yeEcKS3akkJSRx5KdKSzZmQJAp8Z+DGwbxLUdQunUxK86my0iDqJ0qryK04mIiD0ocReResfZyUJ08wCimwfw3ND27DyaxdJdx/hlbyrbEjPZnmQ+3vnpAO/c2Y2buoTZu8kiYmd5miovIiJ2pMRdROo1i8VCZGM/Ihv7MfHaNhzPLmDFvuN8vyWJVfvTeOPHPQyJDMXVWR/WReozLQcnIiL2pE+iIiJnCfKxcntUEz4cHU2gtxtH0vOYvynJ3s0SETtT4i4iIvakxF1E5Bw83Jz5S/+WAPzn5wMUldjs3CIRsSctByciIvakxF1E5Dzu7t2MQG83EtJzmb9Zo+4i9ZlG3EVExJ6UuIuInIenmwuP9G8BwH9+0qi7SH2m4nQiImJP6n1ERC5gVO9wGnpp1F2kPjMMQ+u4i4iIXSlxFxG5gLNH3d/7+QDFGnUXqXeKSgxshvncqsRdRETsQIm7iMhFjI4JJ8DLjcMnNOouUh+VTpMHjbiLiIh9KHEXEbkITzcX/lJ6r7tG3UXqnYLTibuTBVydLXZujYiI1EdK3EVELsHZo+43/WcN7yzfz+7kLAzDsHfTRKSa5Z1VUd5iUeIuIiI1z8XeDRARcQSebi5Mur49T36zlV3JWexKzuLN2H008ffg2g4h3NWzGa1DfOzdTBGpBlrDXURE7E2Ju4jIJbotqgn92gTy0+5Ulu0+xqr9aSSezGPWmkPMWnOIa9oH85cBLenRPMDeTRWRKqQ13EVExN6UuIuIVEKwjzsjezZjZM9m5BYWs3p/Gt9uSuLHXSks253Kst2pdG/WgEf6t6BjmB8+7i54W11wcdadSSKOSmu4i4iIvSlxFxH5kzzdXBjUMZRBHUOJO57DR6vimbcpkU0JGYyZvekP+zrj7+nGjV3CGDuwJT7urnZqtYhUlkbcRUTE3nTpWESkCrQI8mbqrZ1Y/dRAxg5sSeMGHuVG53ILS0jKyOP9FQcZ+K8VzP0tgRKbCtuJOAIl7iIiYm8acRcRqULBPu48eV07nryuHQCFxTay84vIzi9md3IWr/+4l/i0Uzw1bzufrjvMc0M70MjPnf2pOew7ls3+Y9nEn8glzM+dXhEB9IxoSLtQH5ycVMlaxF5UnE5EROxNibuISDVyc3GiobeVht5Wmgd6cXX7ED5dd4i3l+9n59Es7vzo13Met/UILN6RAoCvuws9IwLo1syfjmG+RDb2I9DbWpNhiNRr+brHXURE7EyJu4hIDXJzceKhfi24pVtj/r1sH19sOIKLk4WWQd60CfGmdYgPEYFexKedYn18OhsPpZOVX1xW+K5UIz93Oob5Ed7Qk4bebgR6WQnwcqOhtxutQ3zwtuq/d5Gqkqep8iIiYmf6ZCciYgcNva1MGdaJ54Z2wNXZCedzTIUfOxCKS2zsPJrFhvh0tidlsuNoJvFpp0jOzCc5M/+c5/a2ujCiR1Pu79ucJv6e1R2KSJ1XOlVeibuIiNiLEncRETu6WCLg4uxEl6YN6NK0Qdm2nALzfvkdSZkkZ+ZzIqeQE6cKSD9VSHJmPsezC5ixOp5P1h5iSGQoD/drUe54EakcLQcnIiL2psRdRMTBeFtd6NE8gB7NAyq8ZhgGK/Yd5+NV8aw+kMYP25L5YVsywT7WspF9FycLTk4Wwhp4cGWbIAa2CyYi0KvCubLzi9h1NIuTuUUEeLkR4OVKgJeVBh6uKpYn9UrB6cRdxelERMRelLiLiNQhFouFK9sGc2XbYHYezWTGqngWbD1KanZBhX0PpOawct9xJv+wi4hALwa2DSbY18qOpEx2Hs0iPu3UOX+GkwUa+XlwZdsgrmkfQkzLhppCLHWaloMTERF7U+IuIlJHdQzz480RXZk0tD3JmfmU2AxKDIMSm0FRsY1dyVn8tCeVDfHpxKedIj4tvsI5wvzcCfZ1JyO3kPRThWTlF2MzICkjjznrE5izPgEPV2euaB3IVe2C6RURQESgFxaLRuSl7lBxOhERsTcl7iIidVzpcnR/1KdVIA/1a0F2fhFrDqTxy97jZBcU06GRueRcZJhvheOKSmyczC1kZ1IWy3YfY/nuVFKy8onddYzYXccACPR2Izo8gOjm/nQP96d1sDc+7q4Vfv6JnALWHDzBqn3HScrI447oJtzcpfF5p+GfyClg1f40rmgdqOXwpEapOJ2IiNibEncRkXrOx92VwZGNGBzZ6KL7ujo7EezjTnA7dwa2C2bKMIOdR7NYvjuVNQfS2JKYQVpOIUt2prBkZ0rZcUE+VloGedEiyBsvN2fWxZ1gR1JWuXOvPXiCT9Yc4v9u7EBU+Jn791Oz8vlwZRxz1ieQV1SCl5szD/dvwUP9WmjZO6kRKk4nIiL2pk88IiLyp1ksFnN0vrEf469pTUFxCTuSMtkQf5LfD6WzLSmT49kFZY9f49LLHd8u1Id+rQPxcHNhxqo4tiZmctv0ddzQuREPXBHBd5uT+PK3IxQWmyOe/p6unMwt4q1l+5n962HGXdWaO3s2w81FCZVUn3wVpxMRETtT4i4iIlXG6uJMVHjA6RHzlgBk5RcRd/wUccdzOHg8h4zcIqLC/bmiVSDBvu5lx47q3Yz/9+M+vtp4pKwafqmocH/GXdWK/q2DWLQjmX/9uJdDJ3J5YcFOPl4dR6+IhgT7WAnysRLs406Ir5XIxn6a2ixVQsXpRETE3pS4i4hItfJ1d6Vr0wZ0vcha8sE+7rx2e2fu6RPOlB92sy7uBDEtGjLu6lbEtGhYVvDuhs5hXNcxlC9/O8Lby/ZzJD2PI+mJFc7nbXVhUMcQbuoSxhWtAnFxPjMqn1tYzO7kLPam5NC7RQAtgryrNGapW0rvcdeIu4iI2IsSdxERqVU6hvnx+cO9yMwrooGn2zn3cXV2YnTvcG7r3pilO4+RlJHH8ewCUrPzOZ5dwKETuRzPLuDbTUl8uymJAC83rusYSn6ROZX/4PEcbIZ5rgaernz9lxhah/jUYJTiSEpH3K26x11EROxEibuIiNQ6FovlvEn72TzdXBjWrXGF7TabwaaEkyzYepSF25I5caqQLzYklNsn2MeKq7MTSRl5jJqxnm/G9KFpgGeVxSB1h5aDExERe1PiLiIidY6Tk4Xo5gFENw/g/27owJqDJ/h5TyoBXm5ENvYlMsyPYF93Tp4qZMSH69h3LIdRM9bz9V9iyt13LwKaKi8iIvanxF1EROo0F2cnBrQJYkCboAqv+Xu58dmDvbj9/bUcPpHLPTM38OUjvS9ptF/qDxWnExERe9PNWiIiUq+F+Loz+8FeBPlY2ZOSzf2f/MapgmJ7N0tqES0HJyIi9qbEXURE6r3whl7MfrAXfh6ubE7I4NZpa1m4LZmS0gp2Um8VldgoPv134K7idCIiYifqgURERIC2oT58cn8PfN1d2Hssm7Gfb2LQv1fw7aZEikts9m6e2EnpaDtoqryIiNiP7nEXERE5rVszf1b+YyCz1hxi1pp4Dh4/xcSvtvLWsv0M6xpGy2BvIgK9aBHkjbdVXWh9UFqYDsDqovEOERGxD33qEBEROUsDTzeeuLYND/WL4NN1h5mxOp6E9Fze+elAuf2Cfay0CPLik/t7aiS2DjtTmM4Ji8Vi59aIiEh9pcRdRETkHHzcXRk7sBX3923Ot5uS2J6YSXzaKeLSckjLKSQ1u4DCEpuS9jpOhelERKQ2UOIuIiJyAZ5uLozqHV5uW2ZeEfFppziZW2inVklNaezvwRcP98YwVKhQRETsR4m7iIhIJfl5uNK1aQN7N0NqgKebCzEtG9q7GSIiUs+pyoqIiIhclmnTphEREYG7uztRUVGsWrXqgvsXFBQwadIkwsPDsVqttGzZkpkzZ5bbZ968eXTo0AGr1UqHDh2YP39+dYYgIiJSqylxFxERkT9t7ty5TJgwgUmTJrF582b69evHkCFDSEhIOO8xw4cPZ/ny5cyYMYO9e/fyxRdf0K5du7LX161bx4gRIxg9ejRbt25l9OjRDB8+nPXr19dESCIiIrWOxahnN21lZWXh5+dHZmYmvr6+9m6OiIiIQ/dNvXr1onv37kyfPr1sW/v27Rk2bBhTp06tsP+SJUsYOXIkcXFxBAQEnPOcI0aMICsri8WLF5dtGzx4MP7+/nzxxReX1C5Hfk9FRKTuudx+SSPuIiIi8qcUFhayceNGBg0aVG77oEGDWLt27TmPWbBgAdHR0bz++us0btyYNm3a8Pe//528vLyyfdatW1fhnNddd915zwnm9PusrKxyDxERkbpCxelERETkT0lLS6OkpISQkJBy20NCQkhJSTnnMXFxcaxevRp3d3fmz59PWloajz76KOnp6WX3uaekpFTqnABTp07lpZdeusyIREREaieNuIuIiMhlsVgs5b43DKPCtlI2mw2LxcKcOXPo2bMn119/PW+++SaffPJJuVH3ypwT4JlnniEzM7PsceTIkcuISEREpHbRiLuIiIj8KYGBgTg7O1cYCU9NTa0wYl6qUaNGNG7cGD8/v7Jt7du3xzAMEhMTad26NaGhoZU6J4DVasVqtV5GNCIiIrWXRtxFRETkT3FzcyMqKorY2Nhy22NjY+nTp885j+nbty9Hjx4lJyenbNu+fftwcnKiSZMmAMTExFQ459KlS897ThERkbpOibuIiIj8aRMnTuTjjz9m5syZ7N69myeeeIKEhATGjBkDmFPY77nnnrL977rrLho2bMj999/Prl27WLlyJU8++SQPPPAAHh4eAIwfP56lS5fy2muvsWfPHl577TWWLVvGhAkT7BGiiIiI3WmqvIiIiPxpI0aM4MSJE0yePJnk5GQiIyNZtGgR4eHhACQnJ5db093b25vY2FjGjRtHdHQ0DRs2ZPjw4UyZMqVsnz59+vDll1/y3HPP8fzzz9OyZUvmzp1Lr169ajw+ERGR2kDruIuIiNiZ+qaqp/dURERqE63jLiIiIiIiIlKHKXEXERERERERqcWUuIuIiIiIiIjUYvWuOF3pLf1ZWVl2bomIiIiptE+qZ2VnqpX6exERqU0ut6+vd4l7dnY2AE2bNrVzS0RERMrLzs7Gz8/P3s2oE9Tfi4hIbfRn+/p6V1XeZrNx9OhRfHx8sFgslTo2KyuLpk2bcuTIEYeuUFtX4oC6E0tdiQPqTix1JQ5QLLXRH+MwDIPs7GzCwsJwctJdbFXhz/b3dfVvzJEpltqnrsQBdSeWuhIH1J1Yqrqvr3cj7k5OTjRp0uSyzuHr6+vQf0Sl6kocUHdiqStxQN2Jpa7EAYqlNjo7Do20V63L7e/r4t+Yo1MstU9diQPqTix1JQ6oO7FUVV+vy/oiIiIiIiIitZgSdxEREREREZFaTIl7JVitVl544QWsVqu9m3JZ6kocUHdiqStxQN2Jpa7EAYqlNqorcdRFdeV3U1fiAMVSG9WVOKDuxFJX4oC6E0tVx1HvitOJiIiIiIiIOBKNuIuIiIiIiIjUYkrcRURERERERGoxJe4iIiIiIiIitZgSdxEREREREZFaTIn7JZo2bRoRERG4u7sTFRXFqlWr7N2ki1q5ciU33ngjYWFhWCwWvvvuu3KvG4bBiy++SFhYGB4eHlx55ZXs3LnTPo29gKlTp9KjRw98fHwIDg5m2LBh7N27t9w+jhDL9OnT6dy5M76+vvj6+hITE8PixYvLXneEGM5n6tSpWCwWJkyYULbNUeJ58cUXsVgs5R6hoaFlrztKHABJSUmMGjWKhg0b4unpSdeuXdm4cWPZ644SS/PmzSv8TiwWC2PHjgUcJ47i4mKee+45IiIi8PDwoEWLFkyePBmbzVa2j6PEUl+or7efutLXQ93t79XX1w7q62uXGu3rDbmoL7/80nB1dTU++ugjY9euXcb48eMNLy8v4/Dhw/Zu2gUtWrTImDRpkjFv3jwDMObPn1/u9VdffdXw8fEx5s2bZ2zfvt0YMWKE0ahRIyMrK8s+DT6P6667zpg1a5axY8cOY8uWLcbQoUONZs2aGTk5OWX7OEIsCxYsMBYuXGjs3bvX2Lt3r/Hss88arq6uxo4dOwzDcIwYzmXDhg1G8+bNjc6dOxvjx48v2+4o8bzwwgtGx44djeTk5LJHampq2euOEkd6eroRHh5u3Hfffcb69euN+Ph4Y9myZcaBAwfK9nGUWFJTU8v9PmJjYw3A+Pnnnw3DcJw4pkyZYjRs2ND44YcfjPj4eOPrr782vL29jbfeeqtsH0eJpT5QX29fdaWvN4y62d+rr68d1NfXvjhqsq9X4n4JevbsaYwZM6bctnbt2hlPP/20nVpUeX/szG02mxEaGmq8+uqrZdvy8/MNPz8/4/3337dDCy9damqqARgrVqwwDMOxY/H39zc+/vhjh40hOzvbaN26tREbG2sMGDCgrDN3pHheeOEFo0uXLud8zZHieOqpp4wrrrjivK87Uix/NH78eKNly5aGzWZzqDiGDh1qPPDAA+W23XrrrcaoUaMMw3Ds30ldpL6+dqlLfb1hOHZ/r76+9lBfX/viqMm+XlPlL6KwsJCNGzcyaNCgctsHDRrE2rVr7dSqyxcfH09KSkq5uKxWKwMGDKj1cWVmZgIQEBAAOGYsJSUlfPnll5w6dYqYmBiHjAFg7NixDB06lGuuuabcdkeLZ//+/YSFhREREcHIkSOJi4sDHCuOBQsWEB0dzR133EFwcDDdunXjo48+KnvdkWI5W2FhIbNnz+aBBx7AYrE4VBxXXHEFy5cvZ9++fQBs3bqV1atXc/311wOO+zupi9TX1z51oa+HutHfq6+vPdTX1744arKvd6m6ZtdNaWlplJSUEBISUm57SEgIKSkpdmrV5Stt+7niOnz4sD2adEkMw2DixIlcccUVREZGAo4Vy/bt24mJiSE/Px9vb2/mz59Phw4dyv7hOkIMpb788ks2bdrEb7/9VuE1R/qd9OrVi08//ZQ2bdpw7NgxpkyZQp8+fdi5c6dDxREXF8f06dOZOHEizz77LBs2bODxxx/HarVyzz33OFQsZ/vuu+/IyMjgvvvuAxzrb+upp54iMzOTdu3a4ezsTElJCf/85z+58847AceKpa5TX1+7OHpfD3Wnv1dfX7viUF9f++Koyb5eifslslgs5b43DKPCNkfkaHE99thjbNu2jdWrV1d4zRFiadu2LVu2bCEjI4N58+Zx7733smLFirLXHSEGgCNHjjB+/HiWLl2Ku7v7efdzhHiGDBlS9rxTp07ExMTQsmVL/vvf/9K7d2/AMeKw2WxER0fzyiuvANCtWzd27tzJ9OnTueeee8r2c4RYzjZjxgyGDBlCWFhYue2OEMfcuXOZPXs2n3/+OR07dmTLli1MmDCBsLAw7r333rL9HCGW+qKu/i4cLS5H7+uhbvT36utrXxzq62tfHDXZ12uq/EUEBgbi7Oxc4Yp7ampqhSsnjqS0kqYjxTVu3DgWLFjAzz//TJMmTcq2O1Isbm5utGrViujoaKZOnUqXLl14++23HSoGgI0bN5KamkpUVBQuLi64uLiwYsUK3nnnHVxcXMra7CjxnM3Ly4tOnTqxf/9+h/q9NGrUiA4dOpTb1r59exISEgDH+ndS6vDhwyxbtoyHHnqobJsjxfHkk0/y9NNPM3LkSDp16sTo0aN54oknmDp1KuBYsdR16utrj7rQ10Pd6O/V19e+ONTX1744arKvV+J+EW5ubkRFRREbG1tue2xsLH369LFTqy5fREQEoaGh5eIqLCxkxYoVtS4uwzB47LHH+Pbbb/npp5+IiIgo97ojxfJHhmFQUFDgcDFcffXVbN++nS1btpQ9oqOjufvuu9myZQstWrRwqHjOVlBQwO7du2nUqJFD/V769u1bYemkffv2ER4eDjjmv5NZs2YRHBzM0KFDy7Y5Uhy5ubk4OZXvZp2dncuWiHGkWOo69fX2V5f7enDM/l59fe2LQ3197YujRvv6SpWyq6dKl4iZMWOGsWvXLmPChAmGl5eXcejQIXs37YKys7ONzZs3G5s3bzYA48033zQ2b95ctrTNq6++avj5+RnffvutsX37duPOO++slcss/PWvfzX8/PyMX375pdyyEbm5uWX7OEIszzzzjLFy5UojPj7e2LZtm/Hss88aTk5OxtKlSw3DcIwYLuTsSrOG4Tjx/O1vfzN++eUXIy4uzvj111+NG264wfDx8Sn79+0ocWzYsMFwcXEx/vnPfxr79+835syZY3h6ehqzZ88u28dRYjEMwygpKTGaNWtmPPXUUxVec5Q47r33XqNx48ZlS8R8++23RmBgoPGPf/yjbB9HiaU+UF9vX3WlrzeMut3fq6+3L/X1tS+Omuzrlbhfovfee88IDw833NzcjO7du5ctT1Kb/fzzzwZQ4XHvvfcahmEuT/DCCy8YoaGhhtVqNfr3729s377dvo0+h3PFABizZs0q28cRYnnggQfK/oaCgoKMq6++uqwTNwzHiOFC/tiZO0o8pWtpurq6GmFhYcatt95q7Ny5s+x1R4nDMAzjf//7nxEZGWlYrVajXbt2xocffljudUeK5ccffzQAY+/evRVec5Q4srKyjPHjxxvNmjUz3N3djRYtWhiTJk0yCgoKyvZxlFjqC/X19lNX+nrDqNv9vfp6+1NfX7vUZF9vMQzDqNwYvYiIiIiIiIjUFN3jLiIiIiIiIlKLKXEXERERERERqcWUuIuIiIiIiIjUYkrcRURERERERGoxJe4iIiIiIiIitZgSdxEREREREZFaTIm7iIiIiIiISC2mxF1ERERERESkFlPiLiI1zmKx8N1339m7GSIiIlJN1NeLVC0l7iL1zH333YfFYqnwGDx4sL2bJiIiIlVAfb1I3eNi7waISM0bPHgws2bNKrfNarXaqTUiIiJS1dTXi9QtGnEXqYesViuhoaHlHv7+/oA5tW369OkMGTIEDw8PIiIi+Prrr8sdv337dq666io8PDxo2LAhjzzyCDk5OeX2mTlzJh07dsRqtdKoUSMee+yxcq+npaVxyy234OnpSevWrVmwYEH1Bi0iIlKPqK8XqVuUuItIBc8//zy33XYbW7duZdSoUdx5553s3r0bgNzcXAYPHoy/vz+//fYbX3/9NcuWLSvXWU+fPp2xY8fyyCOPsH37dhYsWECrVq3K/YyXXnqJ4cOHs23bNq6//nruvvtu0tPTazROERGR+kp9vYiDMUSkXrn33nsNZ2dnw8vLq9xj8uTJhmEYBmCMGTOm3DG9evUy/vrXvxqGYRgffvih4e/vb+Tk5JS9vnDhQsPJyclISUkxDMMwwsLCjEmTJp23DYDx3HPPlX2fk5NjWCwWY/HixVUWp4iISH2lvl6k7tE97iL10MCBA5k+fXq5bQEBAWXPY2Jiyr0WExPDli1bANi9ezddunTBy8ur7PW+fftis9nYu3cvFouFo0ePcvXVV1+wDZ07dy577uXlhY+PD6mpqX82JBERETmL+nqRukWJu0g95OXlVWE628VYLBYADMMoe36ufTw8PC7pfK6urhWOtdlslWqTiIiInJv6epG6Rfe4i0gFv/76a4Xv27VrB0CHDh3YsmULp06dKnt9zZo1ODk50aZNG3x8fGjevDnLly+v0TaLiIjIpVNfL+JYNOIuUg8VFBSQkpJSbpuLiwuBgYEAfP3110RHR3PFFVcwZ84cNmzYwIwZMwC4++67eeGFF7j33nt58cUXOX78OOPGjWP06NGEhIQA8OKLLzJmzBiCg4MZMmQI2dnZrFmzhnHjxtVsoCIiIvWU+nqRukWJu0g9tGTJEho1alRuW9u2bdmzZw9gVoH98ssvefTRRwkNDWXOnDl06NABAE9PT3788UfGjx9Pjx498PT05LbbbuPNN98sO9e9995Lfn4+//73v/n73/9OYGAgt99+e80FKCIiUs+prxepWyyGYRj2boSI1B4Wi4X58+czbNgwezdFREREqoH6ehHHo3vcRURERERERGoxJe4iIiIiIiIitZimyouIiIiIiIjUYhpxFxEREREREanFlLiLiIiIiIiI1GJK3EVERERERERqMSXuIiIiIiIiIrWYEncRERERERGRWkyJu4iIiIiIiEgtpsRdREREREREpBZT4i4iIiIiIiJSi/1/h7MABq0aiAUAAAAASUVORK5CYII="
      },
      "metadata": {},
      "output_type": "display_data"
     }
    ],
-   "execution_count": 10
+   "execution_count": 25
   },
   {
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2025-06-25T06:54:11.966429Z",
-     "start_time": "2025-06-25T06:54:11.961222Z"
+     "end_time": "2025-06-26T02:23:28.719481Z",
+     "start_time": "2025-06-26T02:23:28.713924Z"
     }
    },
    "cell_type": "code",
@@ -535,9 +221439,9 @@
     "\n",
     "    def circuit(self, nqubit):\n",
     "        cir = dq.QubitCircuit(nqubit)\n",
-    "        cir.rxlayer(encode=True)  #对原论文的量子线路结构并无影响，只是做了一个数据编码的操作\n",
+    "        cir.rxlayer(encode=True)  # 数据编码\n",
     "        cir.barrier()\n",
-    "        for iter in range(4): #对应原论文中一个量子卷积线路上的深度为4，可控参数一共16个\n",
+    "        for iter in range(5):  # 将线路深度从4增加到5\n",
     "            cir.rylayer()\n",
     "            cir.cnot_ring()\n",
     "            cir.barrier()\n",
@@ -546,31 +221450,41 @@
     "        return cir\n",
     "\n",
     "    def forward(self, x):\n",
-    "        kernel_size, stride = 2, 2\n",
-    "        # [64, 1, 18, 18] -> [64, 1, 9, 18, 2] -> [64, 1, 9, 9, 2, 2]\n",
+    "        kernel_size, stride = 3, 3  # 使用3x3数据块\n",
     "        x_unflod = x.unfold(2, kernel_size, stride).unfold(3, kernel_size, stride)\n",
-    "        w = int((x.shape[-1] - kernel_size) / stride + 1)\n",
-    "        x_reshape = x_unflod.reshape(-1, self.nqubit)\n",
+    "        print(f\"Input shape: {x.shape}\")  # 添加输入形状检查\n",
+    "        print(f\"Unfolded shape: {x_unflod.shape}\")  # 添加展开后形状检查\n",
+    "        \n",
+    "        # 动态计算w值并验证特征图尺寸\n",
+    "        w = x_unflod.shape[2]  # 使用实际展开后的尺寸\n",
+    "        # 确保展平后的总元素数与量子线路输入匹配\n",
+    "        x_reshape = x_unflod.reshape(-1, kernel_size * kernel_size)  # 将每个3x3块展平为9维\n",
     "\n",
     "        exps = []\n",
-    "        for cir in self.cirs: # out_channels\n",
+    "        for cir in self.cirs:\n",
     "            cir(x_reshape)\n",
     "            exp = cir.expectation()\n",
     "            exps.append(exp)\n",
     "\n",
     "        exps = torch.stack(exps, dim=1)\n",
-    "        exps = exps.reshape(x.shape[0], 3, w, w)\n",
+    "        out_channels = len(self.cirs)  # 使用动态计算而非硬编码值\n",
+    "        # 验证总元素数一致性\n",
+    "        assert exps.numel() == x.shape[0] * out_channels * w * w, \\\n",
+    "            f\"Element count mismatch: {exps.numel()} vs {x.shape[0] * out_channels * w * w}\"\n",
+    "        # 确保展平后的总元素数与量子线路输出匹配\n",
+    "        exps = exps.reshape(x.shape[0], out_channels, w, w)\n",
+    "        print(f\"Reshaped shape: {exps.shape}\")  # 添加最终形状检查\n",
     "        return exps"
    ],
-   "id": "7694f4aa38f91ef2",
+   "id": "4b12ed743e5d80d7",
    "outputs": [],
-   "execution_count": 11
+   "execution_count": 26
   },
   {
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2025-06-25T06:54:12.225014Z",
-     "start_time": "2025-06-25T06:54:12.053468Z"
+     "end_time": "2025-06-26T02:23:29.063164Z",
+     "start_time": "2025-06-26T02:23:28.846340Z"
     }
    },
    "cell_type": "code",
@@ -579,27 +221493,27 @@
     "net = ParameterizedQuantumConvolutionalLayer(nqubit=4, num_circuits=3)\n",
     "net.cirs[0].draw()"
    ],
-   "id": "26eb9fed6938a56b",
+   "id": "f22bb2f92a377eed",
    "outputs": [
     {
      "data": {
       "text/plain": [
        "<Figure size 2210.55x785.944 with 1 Axes>"
       ],
-      "image/png": "iVBORw0KGgoAAAANSUhEUgAABroAAAJxCAYAAAAdC2LsAAAAOnRFWHRTb2Z0d2FyZQBNYXRwbG90bGliIHZlcnNpb24zLjEwLjAsIGh0dHBzOi8vbWF0cGxvdGxpYi5vcmcvlHJYcgAAAAlwSFlzAAAPYQAAD2EBqD+naQAAyiJJREFUeJzs3Xd4FFXbBvB7N5veSQIJJRBqQui9IyC9qVQFLIgFfAEFQUVRKRZsqBQBwYIgIr13EEKH0AJJIEAI6SG9b7Ll+4PPKJLA7mZ3Znfm/l3Xe+m7O2fmiZnsvTPnzDkKvV6vBxEREREREREREREREZGNUYpdABEREREREREREREREZEp2NFFRERERERERERERERENokdXURERERERERERERERGST2NFFRERERERERERERERENokdXURERERERERERERERGST2NFFRERERERERERERERENokdXURERERERERERERERGST2NFFRERERERERERERERENokdXURERERERERERERERGST2NFFRERERERERERERERENokdXURERERERERERERERGST2NFFRERERERERERERERENokdXURERERERERERERERGST2NFFRERERERERERERERENokdXURERERERERERERERGST2NFFRERERERERERERERENokdXURERERERERERERERGST2NFFRERERERERERERERENokdXURERERERERERERERGST2NFFRERERERERERERERENokdXURERERERERERERERGST2NFFRERERERERERERERENokdXURERERERERERERERGST2NFFRERERERERERERERENokdXURERERERERERERERGST2NFFRERERERERERERERENokdXURERERERERERERERGST2NFFRERERERERERERERENkkldgFElXHu3Dmjtk9PT8fmzZvxzDPPwNfX16A2bdu2NaU0IiISiDFZYEoOAMwCIiJrxhwgIpI33hsiIiI+0UWykp6ejpUrVyI9PV3sUoiISATMASIieWMOEBERs4CISHrY0UVEREREREREREREREQ2iR1dREREREREREREREREZJPY0UVEREREREREREREREQ2iR1dJCvu7u7o168f3N3dxS6FiIhEwBwgIpI35gARETELiIikR6HX6/ViF0FkqnPnzln8GG3btrX4MYiIyHTMAiIieWMOEBHJG3OAiIj4RBfJilqtRnx8PNRqtdilEBGRCJgDRETyxhwgIiJmARGR9LCji2QlNjYWw4YNQ2xsrNilEBGRCJgDRETyxhwgIiJmARGR9KjELoDKp9froSmynZElKmdHKBQKscuQDFv7/QM8B4iIzM3WsoA5QERkXraWAwCzgIjInJgDRESGY0eXldIUqbG23lixyzDYmFtrYO/iJHYZkmFrv3+A5wARkbnZWhYwB4iIzMvWcgBgFhARmRNzgIjIcJy6kIiIiIiIiIiIiIiIiGwSO7qIiIiIiIiIiIiIiIjIJnHqQpKV4OBgnD17VuwyiIhIJMwBIiJ5Yw4QERGzgIhIevhEFxEREREREREREREREdkkdnSRrMTFxWH8+PGIi4sTuxQiIhIBc4CISN6YA0RExCwgIpIeTl1IslJUVISrV6+iqKhI7FKIiAQRl5SH89fSER6ZgZi7OShSa2GnVMDbwxEtGlVB68a+aBniAzcXe7FLFQRzgIjkJitX/f85kI6ImCzkFZRCr9fD1UWFkCAvtAn1RZtQP1TzcRa7VEEwB4hIbtQlWly+nonwyHRciEpHerYapRodHO3tUKeGG1qH+KJ1Yx80rOMJhUIhdrmCYBYQkZzodHrciMtBeOT9a4K4pAKoS7WwVynh6+WI1o190bqxL5o1rAJHBzuxyzUZO7qIiIgkpqhYgz/23sbS9VE4fy29wu1+/f9/OjvZ4dn+9fDGqBC0auwrTJFERGQxer0eh88kY+n6KGz7Kw5arf6xbZ7sUB1vjA7BoG6BUKk48QcRka27cScHP/wZhV+2xSA7r+Sx2zeq44mJI4PxwpAG8PJwFKBCIiKypKxcNX7ZFoMf/oxCTFxuhdut3HwDAODt4YCXnmqIiSNDUD/QQ6gyzYYdXRLi3zEU/TbPeeC10oIi5N5Oxq2NxxC1ajf0Wp1I1ZEQeA4QyZter8dvO25i2ldnkJGtNrhdUbEWP225gZ+23MCTHapjxYddEFTT3YKVkqUwB4goPDIdEz4Ow6XoTKPaHTydhIOnk1CnuhuWf9gZfTrVtFCFZGnMAiJ5u5dZhCmfn8Yfe28b1e76nRy8+cUZzPo+HB9PbIlpzzeBnR0HPtgi5gCRvGk0Onz1awTmLr+IomKtwe2yckvwzeqr+Gb1VYwZWA/fvdMBPl5OFqzUvNjRJUGxW08g/uB5QKGAs58X6o/ojnZzXoRngxo4NWO52OWRAHgOEMlPUloBXp17AruOxVdqPwdPJ6HpsM344q22mDgqRDbTl0gNc4BIfkpKtZi3/BI+W3XZoCe4KnInKR99X9+HV4Y1wtdvt4O7q4MZqyQhMQuI5GfzwTt4fd4J3MsqNnkfhcUazFx4DpsP3cHP87ohOMjLfAWSoJgDRPITeSsLL84+hnNXK57dxxBrd93CwdNJWP5hZwztUdtM1VkWh2ZIUOa1WNzeFIbbG4/h2g/bsWvgLOQn3kPD53rB0cf2Hjs0p4CAAMyZMwcBAQFil2JRPAeI5OV6bDbaj9lR6U6uvxUUafDGp6cwcf5JaCU20o85wBwgkqLCIg2GTjmI+SsuVaqT699+3HQd3V/ajbQMaa1fIpccAJgFRHLz2crLGDbtUKU6uf7t9JV7aD9mO45fSDHL/qyJXLKAOUAkL0fPJ6PD2B2V7uT6W2pGEZ6aehBf/nzFLPuzNHZ0yYCmSI30CzehUCrhUbua2OWIytPTE/3794enp6fYpQiK5wCRdN2Kz8UTL+9GQmqB2fe9fEM0Js4/Cb3ePDdNrQFzgDlAJDXqEi2GTj2AvScSzL7vi9EZePLVPcjKNXw6XGsn1xwAmAVEUvb5qsuY9f15s+83N78U/Sbuw+nLaWbft5jkmgXMASLpOn4hBf0n7UNeQanZ9z1z4Tl89UuE2fdrbuzokgn3OvcDrDgrT+RKxJWVlYUNGzYgKytL7FIEx3OASHqKijUY+MZ+pKRbbrT9j5uu45vVVy22f6ExB5gDRFIz5fNTOHg6yWL7j4jJwqgZhyUz6EHOOQAwC4ikaMuhO3jvO/N3cv2toEiDQZP3IyW90GLHEJqcs4A5QCQ9SWkFGDLlgFHrcRlrxjdnsfPoXYvt3xzY0SVBds6OcKziDkcfD3gFB6L9pxPg07Qu7l2MQV6s9B45N0Zqaiq+/PJLpKamil2KRfEcIJKHDxaH4/qdHKPanFs3BPEHRuPcuiHGHSc228jqrBNzgDlAJCX7TyZgxcbrRrUxJQcOnErCj5uMO461kksOAMwCIjlIzyrG6/NOGNXGlBzIyFZLaqYHuWQBc4BI+vR6PV6dewJZuSVGtTMlC+4fx3pnelCJXYAQ0tPT8fnnn2PLli1ITEyEn58fRo4cifnz5+OVV17B2rVr8eOPP2LChAlil2oWLaaNQItpIx54LW7PGZx+50eRKiKh8Rwgkr5Tl1Ox8Dfjn7Ty93VBzWquRrUpVmvx0odhOLF6EBQKhdHHJOExB4ikL7+wFBM+Pm50O1NyAACmf3UW/bvURC1/N6PbkjiYBUTSN3XBaaRlGrcml6k5sPVwHNbvvY3R/esZ3ZbEwRwgkr61u26ZtF67KVmQfK8Qby44jV8/6W708YQg+Y6uK1euoE+fPkhNTYWrqysaN26MxMREfPPNN7hz5w7u3r3/yF3z5s1FrtR8bqw9iNhtJ6BU2cGrUSCaTn4Kzr5e0BT/07OrdFBh8L4vcHvLcUR8v7ns9S7fvgEnPy8cHPOJGKWTmRhyDjyxYjr0ej2OvvZN2WsOXm546q+FOD93NW5vDhOjdLPT6/U4dzUdS9dHYVdYPHLySuDirELrEB9MHBmCoT1qw96eD7eS7fn0x8sQckDlqctpOHwmGb06VBfuoGQy5sCDElIKsGJTNNbsvIWke4WwUypQp4YbXn66IV4c2hBVPB3FLpHIaKu3xyA+xfzrM1Ykv7AU36+NxJfT2wl2TKocZsE/CgpLsW7PbSzfGI3o2ByUlGrh5+2E4b2DMHFkMBoFeYldIpHRbt7Nxe+7bwl6zPkrLmFUv7oc/GYjmAP/0On0OHg6ET/8GY2j55ORX6iBm4sKT7QNwKRRIejVvjrPa7I5er0e85ZfEvSYv+28iTmTWqFODXdBj2sISd/dTU9Px6BBg5CamoqZM2ciJSUFFy5cQGpqKubNm4fNmzfj0qVLUCqVaNKkidjlmk3enRQkh0Ug8cglXFu2HYfGfQbfVvXR8fNXyrbRlWgQNmURmk15Gt6NawMAAvu1Rc3ebXBi2lKxSiczMeQcOPXOclRrH4KgpzqXvdbh0wlIOxstmS8yOXkl6D9xH9qP2Y5ft8cgPasYpRodcvJKcPhsMka8fRgNBm/ApegMsUslMsqdxDzsCjN+xE5lLV0fJfgxyTTMgfv0ej0+WnIBdfqvx7zllxCbmAd1iRaFxRpE3srG9K/OosaT6/Dz1htil0pkFL1eL8pn8k9bb6CoWCP4cck0zIL79h5PQK0+f+CVOcdx/lo68gtLUVKqQ2JaIb5bew3BQzfh5Y/CUFJquXUtiCxh+YZowY957VY2wsI55Z2tYA7cdzc5H61Hb0Xf1/dh6+E4ZOWWoFSjQ1ZuCbYcikPvV/eizehtSBBwABGRORw+k4wbccYtZ1FZej2MnjpdKJLu6JoyZQri4+Mxbdo0LFiwAG5u/0yz8f777yM4OBgajQYNGzaEs7OziJVa1r0LMbi9KQx1n+kKv1YNyl7PjIjF1SXb0PX7yXAJqIKOX7yGM7NWoihVuotxuri4oH379nBxcRG7FEGVdw6os/JxYtpStP9kApyreaP2wA7w7xSKU+8sF7la88gvLEWvV/Zg38nER24Xl5SP7uN34fJ1dnaR7Vi15YagT3P9bdtfcUi+Z9uLUDMH5JMDADD9qzOYu/witNqK/2CK1VqM/zAMy/5kRy7ZjpOX0nDtVrbgx83MUWPTwTuCH9ec5JoDgDyzYNexuxg0ef9j1634acsNjJ55BFqtTqDKiCqntFSHn0QaqLPcSm9wGkOuWSDHHEhMLUCXF3biUnTmI7e7EJWBLi/utPnrXZKX5RuFH/AAACs3X4dGY33fmSTb0RUZGYn169ejatWqmDdv3kPvKxQKtG7dGsDD0xbGxsZiyJAhcHd3h7e3N55//nlkZNj2TfDLCzdCp9Gi5czRD7x+5fvN0JVqMOTAl0g5eQ2x24xbxNTWBAYGYtGiRQgMDBS7FMGVdw4kHr6IuJ2n0G3xFHT4/BWcnP4D1Fn5IlZpPm9/fRbhkekGbZubX4qn3zxklR/SROU5ej5ZlONqtXqcvGTbCzYzB+STA9uOxGHhb9cM3v6NT0/hasyjL4CJrIVYOSD2sc1BzjkAyCsL0rOKMWrGkUcOdvi3LYfi8N1aw3ODSExXb2YiM0ctyrFtPQcAeWeBnHIAAMbNOmrwVM9xSfl44YOjFq6IyDz0er1on8f3sooRHZstyrEfRbIdXWvXroVOp8PYsWMrHKHh6Hh/PYZ/d3Tl5eWhR48eSEhIwLp167BixQqEhYVh0KBB0Ols9yZ43p0UxG47gerdm6Nq+5Cy1/VaHVLPRsPJxxM31x8RsUJhaLVa5OfnQ6uV37QUFZ0D5z7+Fe5B/kg8fBEJhy6IWKH5ZOWqsXpHjFFtYhPzsFuEqeCIjKXV6nDxMaPRLCk80rYHfjAH5JEDAPDdGuNuVup0eiz5g091kW0Q87OYOWDb5JQFP2+9gYIi46baXLwuCjqdCI/NExlJzM/ixLRCpGYUiXZ8c5BzFsgpByJuZOLIOeM6Ag6cSkLU7WzLFERkRomphUjLLBbt+NZ4TSDZjq5Dhw4BAHr06FHhNgkJCQAe7OhasWIFEhMTsXXrVgwaNAgjRozA2rVrcfr0aWzfvt2yRVvYle82QafVouWMUWWvVW3bCA2e7YmoVbvRbu5LsHNyELFCy4uJiUHPnj0RE2NcJ4hUlHcOaIrUyI9LQ1bUXRErM69ft8WgqNj4L6xcf4hswc27ucgvLBXt+BeiDHtS0loxB+SRA9Gx2UZf1AL3F9bNK3j09FZE1uCiiOuLRtzMtOm1jOSeA4A8skCv12OZCesXxSbmYd+JBAtURGReYn8nv2Dg7CnWSu5ZIIccAIAfTJya3NR2REIS83oAAMLNmEN6vR4FBQUoKCiAvhLrdCj0lWltxQICApCSkoKIiAg0adLkoffVajWqVauGnJwcJCQkoEaNGgD+6Rg7cuTBp5uCgoLQs2dPrFq1yuha2rRpg5QU4xbrtNcr8ZGundHHMobK2RFDDn2FyBW7EP3rPvTfMhcZEbdxdvbPRu9rjvIsShXCP/E2fPhwo7ZPS0vDunXr8Oyzz6Jq1aoGtdm4caMppVWKEL//f+u3aQ7iD4Tj2jLTO3PFOgfKk+n6NIocWxjdTqnLQ0D2V+YviMiM1KpaSPeYUO5759YNgb/vo+eZ9/d1hspOCY1Wh5T0ikdipqQXou2zD38m2GsSUTV3hXFFW5gxWWBKDgDSzwKp5UChQxNkuY0wqa1fzg9w0HKRdbJuSd7vQa9wKve9x2VBZXMAAPyzPoed3npG8zMHzENKWaBTOCLZe5ZJbT0KD8K9OMzMFRGZV6brCBQ5PnyvCzBfDgAVZ4F3/ia4lFwxrmgL4r0h85BSDgBAmscElKpqGd3OofQO/PKMvzdKJKQChxbIdnu63PeEuDfkrL6CKgWbjCu6AjqdDsnJ9weqtmjRAhcvXjRpPyqzVGOFCgruz79aXFz+I3xr1qxBTk4OfHx8yjq5gPtre40Y8fCNkcaNGyMyMtKkWlJSUpCYmGhUGweFHVDNpMMZrO2cF5Affw/Rv+wFAByfuhhDDn6Fu7vPIuWUcdP9JCUnoUQv/MjOv3/PhioqKir7p6Ftjf3dmYMQv39zE+scKFetUsDR+GY6vUqU3zeRUVxdAY/y3/L3dUHNaq4G7UZlpzR4238rLdVa3d+JMVlgSg4AzAJDWFUOeNcB3Exrei89Gyi0rnOc6CFeCkBR/luGZoGpOQAAKSlpgCbXpLaWwBywHlaTBSpPwNu0prn5RchNZQ6QlQssqfCaV4gcyMrORVaW9fyd8N6Q9bCaHAAAZ71Jd75LSvVWd81L9BDvuhVe8wpxb6ioSG2Rv5PUVNPXhZdsR5e/vz/y8vJw/vx5tGnT5oH34uPj8d577wEAmjVr9sB7WVlZ8PLyemh/3t7euHnzpsm1GMterwQsOACiRo8WCBrSGdt6TS97LS8uFeGfrEXnbydhW8/p0BQYPs9n9YDqoozYcHU17g/x7y8wzs7OBrf9d0eoUCz9+7cEsc6B8mQ5KVBoQjs7qOEvwu+byBgldt64V8F7KemPP/ONGbVTHnt7Bapa2d+JMVlgSg4AzAJDWFMOFNk7weiV7PR6QKFAVR932Htb1zlO9F9J0KGiaTkelwWVzQEACPD3g1LvbkipgmAOWA9ryQI9VEgysa2Hmz3cVcwBsm6ZTvao6BPcXDnwqH15e7nBxcV6/k54b8h6WEsOAMA9lRamTEruaK+Dr5Vd8xL9V6GDO7IqeE+Ie0POzvaoYqa/k38/0VWtmum9+5KdunDSpEn44YcfUKtWLRw8eBANGzYEAJw+fRrjxo1DXFwcSktL8eabb2LhwoVl7RwcHPDOO+9g3rx5D+xvzJgxOH/+PK5fvy5I/aWFxVhbb6wgxzKHMbfWwN6l/OlTLOncuXNGbR8dHY3nn38eq1evRnBwsEFt2rZta0pplWJrv39AvHOgPDuP3sXgyQeMbve/Zxtj0XsdLVARkfmkZhTBv8fvJrePPzAaNau5IiG1ALV6/2F0+9H96mLdFxWvfykGY7LAlBwAmAWGsKYcyMkrQfVe61BYrDGqXf1AD1zfPhxKZQWPyhBZiVYjt5o8L39lc8DL3QGZx8dCobCevxPmgPWwpizo89oeHDhlfHfX9e3D0bCOpwUqIjKfWd+dx2erLpvUtrI5AACnfhuMDs0Nn/7V0nhvyHpYUw58szoC0786a3S779/tgMnPhVqgIiLzCQtPQbeXdpncvrJZ8OFrLTHnjVYmH//fCgoK4OZ2//G0/Px8owcv/E1plmqs0Lvvvgtvb2/Ex8cjNDQUTZs2Rf369dGxY0eEhISgS5cuAIDmzZs/0M7b2xtZWQ/3h2ZnZ6NKlSqC1E6WU79+fezbtw/169cXuxSyoP5daqJ2dePnrJo40vCbHURiqebjjBpVHz3XsiW1buwr2rHNgTkgD57uDhg7qJ7R7SaODGYnF9mE1o19RDt2qxAfq+rkMhZzQD4mjQoxuk3vjtXZyUU2QcwcsLNToHkj274/xiyQhxeHNoSTo51RbVycVHh+cAMLVURkPi1DfCDmV3Ixc6giku3oCgwMRFhYGAYMGAAnJyfExcXBz88PK1euxLZt2xAdHQ3g4Y6ukJCQctfiioyMREiI8V+UybqoVCp4e3tDpZLsrJ0EwM5OiQ9fa2lUmxF9gtC4nokT+RMJrE2oeJ1N1vhlxhjMAfmYNq4J3F3tDd4+MMAVLz3V0IIVEZmPmIMObH3AA3NAPgZ1C0SrEMO/t9jZKfD+Ky0sVxCRGYl5PdC4rhecnWz7M5RZIA9VPB0xdYxxT2ZNe74JPN0dLFQRkfm4udijkYiDc6zxmkCyHV0AEBoail27diEvLw+5ubk4deoUXn75ZWRlZSE5ORkqlQqNGzd+oM2gQYNw/PhxxMfHl712+vRp3LlzB4MHDxb6RyAzS0hIwPTp05GQkCB2KWRh459uiNmvtTBo226t/fHLvG6WLYjIjIY8UVuU4/p6O6GjFU1RYgrmgHw0CvLC5oW94GLAjZhqPs7YvaQvvD0qWNWdyMoM7FZLtKcPh/YQJ4PMhTkgHyqVEjsX9zHoJpBSqcCqj7uie5sAASojqrzAADc0ayjOU1VDnggU5bjmxCyQj08mt8az/esatO24QfUxZ5J5pmIjEoJY38tbN/ZFdRFnGqqIpDu6KhIREQEACA4OhqPjgzc0Xn31VQQEBOCpp57Czp07sXHjRjz33HNo164dhg4dKka5ZEb5+fkICwtDfn6+2KWQAOa+0Rq/zOuG+oEe5b7v6e6A6c83wf7l/eDizJFcZDtG96sLLxFGmb38dEM4Odr23wpzQF6e7FADR38egB5ty79xqVQq8FTP2ji9ZjBC6/OpXrIdtfzdRLnR2LxRFXRqYdsDHpgD8hLg54KTvw3Gi0MbwNGh/OmrWoX4YPeSPnhhKKeqItuhUChMmp6zspRKBV4bYftT/jML5MPOTok1nz2Bz99sg2o+zuVuE+Dngi/eaotf5nfjNOZkU14bHizK9IWTRoVY5VTmsu7o+u+0hQDg4eGBw4cPIyAgAKNHj8aECRPQqVMn7Ny5E0qlLP9zEdm0F4Y2wPXtw7FvWV9Mfq4xnJ3uX+B6uTsg8cBofPV2+woveomslYuzCi89JezNGIXi/pcoIlvTJtQPh1cNwLUtz2DWhOZw+f8ccHe1x529I7Hl2ydRp4a7yFUSGU+MG5zWelFL9ChVPB3x87xuSDw4Gl+/3a4sB9ycVTizdgjO/zEUfTvXFLlKIuONGVgPHm6GT9NsDkOeCEQtf+PXwyYSk1KpwDvjm+Pu/lFY/2UPuP7/QGdXZxX+/Kon4vaOwoyXmrGTi2xOUE13DOhaS9Bjerk7YHQ/w56SFJptD8s20aM6ugCgXr162Llzp5AlmdXws0uhVZdCW1wCALiyaAvubD/5wDb+nZug9awxsHdzgl6nx91953Dx83VilEtmUv2J5mj13nNQKJRQ2tvh6tJtuLXhaLnb2jnaY/D+L6ApKsHOfu8AALyCA9Hhswlw9vWErlSDtPAbOPP+KuhKNEL+GBahVCrQp1NN9OlUE5sP3kFicSFcnVVwdRH2ooDInGZNaIG1u24hLbNYkOO9Na4JgmqyM8Ca9f5jNpz9vACdDqUFxTjz/ipkXrvzwDZuNf3Q5bv/oUqTOsiNTSnLgL/fe+bUYmRH3y177ciEr5AXlyrQT2BZjet545MpbfDr9hgUFhfCw9WeN2rIpj3ZoToGdquFXcfiH7+xGTRvVAUv8okXq+Xg4YJ+m+aU/X87Z0e4166GP5q+jJLsf55YqDeiO0JfHVT2/12q+yD1dBSOvPwlVC5O6LHqbfg0u3/z4o/Q8cL9AALw8XLCtOeb4pvVV1FYXAhPdwe0a+ondllEJnNzsccnk9tg8menBDmek6MdPp/aRpBjkWnsHO3Rfdlb8GxQE9oiNYruZePUzBXIT7hX4fZyujfkYG+HkX3rYtqXZ1BQpIGXuwNG9AkSuyyiSvnirbY4cCoRJaU6QY73+ZttrXZWLOusysL+7uhq1qyZyJVYzpEJXyH7esUXvSU5BTg6cSHy76bBztEeff78EEFPd0HsluMCVknm1G3Jm9g3/GNkRcXBraYfng77DnG7z0BT8PBN8JbvPou08zdQJbRO2WtadQnOzFqFrKg4KJRKdFs6FaGvD0HE95sF/CmIyFC+3k744YPOGDbtkMWP1bC2J+b/r7XFj0OVc/TVr1GSWwgACOzXFp0XTsKOPjMf2KYkvwgXFqyDg7sLWswY9dA+SnILsL33DEHqJaLKUSgUWPFhZ4Q+vRnZeSUWPZZKpcAv87rBwZ5PwVurktzCBz6/Q18fAv+OjR/o5AKAWxuOPjAYbujhr3F78zEAgE6jwdUl26DOykOfPz8UpnAiqpRJo0Kw6eAd/HUu2eLHmv+/1mgU5GXx41DlXP/tABIPXwQABL/UDx2/eh0HRs8rd1veGyKyfY3reWPOpFZ477vzFj/Wkx2q49XhjSx+HFPJci6+kydPQq/Xo2/fvmKXIprMq7HIv5sGANCqS5F59Q7ca1cTuSrL8/Pzw9SpU+HnJ82Rew6e9xcCtHd3QXFWXrkjbqq2D4FHUABubzz2wOt5sSnIiooDAOh1OqRfugX3QNteg4FI6p55so7RUximpBciIbUAKemFBm3v6GCHX+d3g7OTNMbGSDkH/u7kAu7ngF6nf3ib7HyknY2GplAtZGlEZCHVq7pi2ezORrUxNgeA+zc3WwT7GFueVZJyDvxb/dE9ELPu0YNhfFvUh5OfF+7uu39jRFeiQfLxCJTkFAhRIhGZgVKpwE9zusLHy/HxG/8/U3KgR9sAvDk21JQSrZJUs0CrLi3r5AKAexdiKryvw3tDRNLx9gtN0a21v1FtjM0CP28nrPy4i1VPYy6Nu1b0kG5LpkKhVODehRiEf/Y71Bm5FW7r7OeFOoM64MDYTwWsUBw+Pj4YM2aM2GVYxNHXvkGPVTOgKVTDwdMVR17+ErrSBzu6VM6OaDf3RRx+YQE8ggIq3JfK2RENx/TC+flrLF02EVXS8tldkJGtxva/7j5+YwBtn91u8L7tVUps+KonOjSXzoWNlHMAALp8PxkBne7fhDjw3Hyj29u7OWPQns+hsFPi7p6zuPLdZuh1wkyBQESmGdWvLlIzijB1wWmDtjcmBwDgrXGhmPmSdGbCkHoOAIBf64Zw8nZD/IHwR27X4LmeuLXxKPQarUCVEZElBNV0x94f+qLXK3uQm1/62O2NzYG2TXyx9bsnYWcnnbHycsgCAAh5eUC5WcB7Q0TSolIpsf373ug5YTcuRGUY1MaYLPByd8C+ZX1Ru7p1L2chnZSiMnue/hDbn3wb2/vMhDorH12/+1+F29q7OaPX6ncRsXQbMiNiBaxSHLm5uTh48CBycyvu+LNFCjslmk5+GodfXICNbSdi34g56PLd/+Dg9eDaI20+HIfon/ehMCWz4n2p7NB9+TQk/nUZ8fvOWbp0Iqoke3slNn7dC2MH1TPrft1c7LFjUW8MfiLQrPsVm1Rz4G/HpyzChjav48KCdWjz4fNGtS1My8KGVq9hZ/93sW/kXFRrH4LQ1wdbqFIiMqcpY0Kx4sPOUKnMO8Lyg1db4Ou321v1yE1jST0HAKDBsz1xc8NR6LUVD1Swc3ZA0JDOiFl3WMDKiMhS2oT64a9VA+Dv62zW/fZsF4CDK/rDw83BrPsVmxyyoOmUZ+AR5I8Ln//+0Hu8N0QkPZ7uDji8cgC6tzHuya7HqV7VBUd/HoiWIb5m3a8lsKNLggoS0wEAeo0WkT/uRLX2IeVup3J1Qu/fP8Dd/ecRuXynkCWKJikpCbNmzUJSUpLYpZhVlSZBcPH3Rtq56wCAjMu3UJSahSqNaz+wXdV2IWgxbTiGn12K7svehHdwIIb+tbDsfYXKDk+smIaitCycnf2ToD8DEZnO3l6J1Z90x5rPusPbo/IXoU92qI6rm59G3841zVCddZFqDvzXrQ1H4d8pFI7ebo/f+P/pSjQo/v8nwEuy8xHzx+EKv0MQkfV5ZXgwzv0+FM0bVan0vmpXd8PBFf0x73+tJdXJBUg/B1TOjqgzpBNu/vHoDqw6gzoiOyYBOTcSBKqMiCytZYgvrm5+Bs8NqPwAOCdHO3wzoz32L+8nuU4uQPpZEPr6ENQe0B4Hx3wCbdHD63jy3hCRNHm6O+DQj/3xxVtt4ehQ+bV1xw2qj4hNz6BZw8pfXwiBUxdKjMrZEUp7u7J1OoKe7oKMqw8/qaVyud/JlfjXJVxZuFHoMsnMChLT4RLgA4961ZF7KwnudfzhXrsacm8/uCDt9l7Ty/7dv2Mo2nz0PHb2ewfA/afCui97C+qsfJx8e5mg9RNR5SkUCowZWB8921XHB4vD8fvuWyhWGzcVUcPannhnfDO89FQDyd3YlDp7N2eoXJ1QlJoFAAjs3w7qrHyos/IN3oeTjwfUOQXQa7RQOqhQe0CHcr9DEJH1ahHsg7O/D8E3q6/i+98jkXzP8PVXgPvTkrwyrBFmv9YC7q7Su7EpB3UGd0RWVBxybj765m2D0T35NBeRBPl4OWHt509gZN8gzFt+CeGR6Ua1t7NT4KketfHplDZoWMfTQlWSJTV+bRCCnu6M/SPnPrCG77/x3hCRdNnZKTHjpWYY/EQgZn1/Htv/ugut9uH1ux+lbRNffPhaSwzqblsz/LCjS2Kc/DzRY+UMKOyUUCiAvLg0HJ+8CADQ6avXEb//POL3n0fjVwbAr2V92Ls4onb/dgCAOztP4cp3m8Usn0xUnJ6DUzOXo8eP06HX6QGFAqffW4nClMwHfu+PEjS0M+oM7IDMa3cw5MCXAIDUc9dxZtZKIX4EIjKTAD8XrJrTFV9Oa4dftt3Ahv2xuHQ9s8JOrxpVXdC5ZTW8MqwRerWvzg4uG2Xv4YIeK9+GyskBep0exRm5OPT8ZwAezH+lgwrDTi2BnaMK9u4uGBG+HLc2HcWFT39H1fYhaDljFPRaHRQqO6Qcj8CV7zaJ/JMRkbEc7O3w7svNMf35pth2JA6/bI/BmYh7SM8qLnd7Dzd7tGnsizED62N0v7pwceYloi1r8FwvxPz+YAfWf68H3GtXg0+zujj4/znxb4P3fwHnqt5w8HTFiPDlSDl5FWH/fz1JRLZjaI/aGPJEIM5dTceyDVE4ej4FtxPyyt3WXqVEk/reGNw9EK8Ob4Qa1VwFrpbMxSWgCtp9/CJy76Sg38aPAQDaEg12DXyP94aIZCY4yAubFz6JhJQCLN8YjV3H4hFxMxMaTfmdXvVqueOJNgF4bUQw2jbxE7ha8+BVjMTk303Djj4zyn3v3yMxrny3mZ1aEhO79QRit5546PWKRuCknLpWNmIHAG5vDsPtzWEWq4+IhFXF0xHTnm+Kac83hUajQ9TtbMTczcWEj8OQlVsCH09HXNs6DNV8zDuPP4mjMCkDuwa8V+57/84BXYkGG1q/Vu52d3efwd3dZyxSHxEJz95eieF9gjC8TxD0ej3iUwoQEZOJcbOOIiu3BFU8HXB6zRDUq+UBpZKDHKRiz1OzH3rtv9cDeXGpWNtgXLntd/SZaZG6iEh4CoUC7Zr6oV3T+zcss3LVuBiVgRHTDyEztwRVPBywb3k/NG1QxSxTXJH4CpMz8UvA8HLf470hInmq6e+Kef9rjXn/aw11iRYRMZmIS8rHK3OOIyu3BL5ejojZOQJeHo5il1ppXKOLZMXR0RGNGjWCo6Pt//ESERlKpVKiacMqeObJOnBxuj/GxcnRTpadXMwBIpIjhUKBwAA3DOwWWJYDzo4qNKjtKbtOLuYAEcmVt4cjeravDue/c8BJhTahfrLs5GIWEJEcOTrYoU2oH4b1Diq7JnB0sJNEJxfAJ7pIZoKCgvDbb7+JXQYREYmEOUBEJG/MASIiYhYQEUkPn+giIiIiIiIiIiIiIiIim8SOLpKV69evo3Pnzrh+/brYpRARkQiYA0RE8sYcICIiZgERkfSwo4tkRa/Xo7S0FHq9XuxSiIhIBMwBIiJ5Yw4QERGzgIhIerhGl5VSOTtizK01YpdhMJWzNBatsxa29vsHeA4QEZmbrWUBc4CIyLxsLQcAZgERkTkxB4iIDMeOLiulUChg7+IkdhkkEv7+iYiIWUBEJG/MASIieWMOEBEZjlMXEhERERERERERERERkU3iE10kK3Xq1MG6detQo0YNsUshIiIRMAeIiOSNOUBERMwCIiLpYUcXyYqTkxPq1asndhlERCQS5gARkbwxB4iIiFlARCQ9nLqQZCU5ORnz589HcnKy2KUQEZEImANERPLGHCAiImYBEZH0sKOLZCUnJwfbt29HTk6O2KUQEZEImANERPLGHCAiImYBEZH0sKOLiIiIiIiIiIiIiIiIbBI7uoiIiIiIiIiIiIiIiMgmsaOLiIiIiIiIiIiIiIiIbBI7ukhWqlSpghdeeAFVqlQRuxQiIhIBc4CISN6YA0RExCwgIpIednSRrCiVStjb20Op5KlPRCRHzAEiInljDhAREbOAiEh6+IlOspKeno6VK1ciPT1d7FKIiEgEzAEiInljDhAREbOAiEh62NFFRERERERERERERERENokdXURERERERERERERERGST2NFFRERERERERERERERENokdXSQr7u7u6NevH9zd3cUuhYiIRMAcICKSN+YAERExC4iIpEcldgFEQqpRowbmzp0rdhlERCQS5gARkbwxB4iIiFlARCQ9fKKLZEWtViM+Ph5qtVrsUoiISATMASIieWMOEBERs4CISHrY0UWyEhsbi2HDhiE2NlbsUoiISATMASIieWMOEBERs4CISHo4daGV0uv10BTZzsgSlbMjFAqF2GVIhq39/gGeA0RE5mZrWcAcICIyL1vLAYBZQERkTswBIiLDsaPLSmmK1Fhbb6zYZRhszK01sHdxErsMybC13z/Ac4CIyNxsLQuYA0RE5mVrOQAwC4iIzIk5QERkOE5dSERERERERERERERERDaJHV1ERERERERERERERERkkzh1IclKcHAwzp49K3YZREQkEuYAEZG8MQeIiIhZQEQkPXyii4iIiIiIiIiIiIiIiGwSO7pIVuLi4jB+/HjExcWJXQoJSKPR4WpMJnYdu4uiYg0AQF2qRWGRRuTKiEhozAF50uv1uJucj/0nE1Ckvv/ZX1yiRVpGkciVEZHQmAPylZ2rxrHzyf/kgFqLm3dzodfrRa6MiITGLJAndYkW4ZHpKFJrAQBFai0uRKajpFQrcmVEZA6cupBkpaioCFevXkVREW9sSV1aRhFWbbmBHUfv4tL1DBQVP/jFJT1LDfeOqxES5InubQLw+ohgNG1YRaRqiUgozAH5UJdosfFALNbuuoXz19JxL6v4gfczstWo1uN31Kzmig7N/PDi0Abo17km7Ow4DoxIypgD8qHX63H8QipWbr6OE5dScSs+74H3M3LUaDBoAzzdHdAq2Acj+gRh7KB6cHd1EKliIhIKs0A+YhPysHxjNPafTMTVm1ko1ejK3svMUaP16G2wVynRtIE3+naqiVeHN0KdGu4iVkxEpmJHFxFJSmxCHmYvCceG/bEoKdU9cludTo9rt7Jx7VY2lq6PQtdW1TD7tZbo3bGGQNUSEZG5FRZp8Nmqy1i+Ifqhzq3yJKQWYOOBAmw8cAdBNdzx5thQvDE6hB1eREQ2Sq/X4/fdt7DgpyuIiMl67PY5eSU4ci4ZR84l451vz+GFIQ3w8cSW8PFyEqBaIiKyhPDIdHy4JBx7jifgcQ/ulmp0uBCVgQtRGfj8p8sY2LUW5r7RCi1DfIUplojMgh1dEuLfMRT9Ns954LXSgiLk3k7GrY3HELVqN/TaR9/4J9sm53NAp9Pjhz+j8M7CcygwcUrCsAup6PPaXkx4piG+frs9PNw4mpOIbIuccwAAwsJT8NKHxx4atW+o2MQ8TF1wGr/vvoVf5ndDcJCXeQskIhKAnLMgMbUAr807gV3H4k1qn1dQisXrIvHnvttYNrsznu5Vx7wFEhEJQM45oC7RYu6yi1jw8xVotcZPTavXAzuPxWPvyQTMerkF3n+1ORzs7SxQKRGZGzu6JCh26wnEHzwPKBRw9vNC/RHd0W7Oi/BsUAOnZiwXuzwSgNzOgfzCUoyYfhh7TySYZX8rN9/A/lOJ2LO0LxrX8zbLPomIhCS3HNDr9fjkx0v4cMmFx47YNMSZiHtoMWIrVs3pgjED61d+h0REIpBbFhw+k4Rnph1CTl5JpfeVllmMZ946hFeHN8LS9zvxKV8isklyy4Hke4UY8MY+XIrOrPS+NBo95i6/iJ3H7mL30r6o5uNshgqJyJL4bU2CMq/F4vamMNzeeAzXftiOXQNnIT/xHho+1wuOPh5ilyeqgIAAzJkzBwEBAWKXYlFyOgfyC0vR9/W9Zuvk+tvd5AJ0e2kXIm5U/gsSEVkP5oD0ckCv1+O9785j9mLzdHL9TV2ixdj3jmLlpuvm2ykRiU4uOQDIKwv2Hk/AgDf2m6WT699WbLyOse8dhVaiTz4QyZVcskBOOZCYWoAuL+w0SyfXv12IykDXF3ci+V6hWfdLRObHji4Z0BSpkX7hJhRKJTxqVxO7HFF5enqif//+8PT0FLsUQUn1HNBqdRg27RBOXkqzyP4zstXo8/peJKQUWGT/RCQ85oC0cgAAvll9FQt+umKx/b869zi2HYmz2P6JSFhyzQFAullw7uo9PDPtINQlWovs/4+9tzF1wWmL7JuIxCHXLJBqDuQVlKDP63txO8G06csfJyYuF31e24uCwlKL7J+IzIMdXTLhXud+gBVnWeZD31ZkZWVhw4YNyMp6/KLEUiPFc2DR75HYfzLRqDbn1g1B/IHROLduiEHbp6QXYcLHYdCb8zEBIhINc0BaOXDlRibe++68UW2MzQG9Hpjw8XGkZRSZUiIRWRk55wAgvSwoKtZg3KyjKCo2vJPL2BwAgCV/RGF3mGnrfhGR9ZFzFkgtBwDgnYXnEHkr2+DtTcmBqzezMOt74647iEhYsujoSk9Px9tvv4169erByckJtWrVwvTp01FUVISxY8dCoVBg5cqVYpdpNnbOjnCs4g5HHw94BQei/acT4NO0Lu5djEFebIrY5YkqNTUVX375JVJTU8UuxaLkcA7ExOVg1iLjv2T4+7qgZjVX+Pu6GNxm38lE/LTlhtHHIiLrwxyQTg6Ulurw4gfHUKoxbjopU3IgPasYb3x60tgSicgKySUHAHlkwUdLL+D6nRyj2piSAwDwypzjyM5VG9WGiKyTXLJADjlw+EwSfvgz2qg2pubA979H4tj5ZKPaEJFwVGIXYGlXrlxBnz59kJqaCldXVzRu3BiJiYn45ptvcOfOHdy9excA0Lx5c5ErNZ8W00agxbQRD7wWt+cMTr/zo0gVkdDkcA688+05o0ZuVtbMhefw3IB6cHayzY/NxNQCrNl1E3eTC6DX61GzmivGDqqPwAA3sUsjIguQQw78su0GLkZnCHa8jQfu4Nj5ZHRrY5trORQWabB+321cvp6JIrUG3h6OGNojEB2aVYVCoRC7PCKyAKlnwZ3EPHy9+qpgx0tKK8SCn6/gs6ltBTumOel0ehw+m4R9JxKRk18CV2cVOjSriqd71YaDvZ3Y5RGRBUg9B/R6veBTy0794jQurH/KZr8/X4/Nxro9t5GaUQSVSol6Nd0xbnB9+Hg5iV0aUaXZ5h1bA6Wnp2PQoEFITU3FzJkzMXv2bLi53b+pO3/+fMyePRsqlQpKpRJNmjQRuVrzubH2IGK3nYBSZQevRoFoOvkpOPt6QVP8z8K8SgcVBu/7Are3HEfE95vLXu/y7Rtw8vPCwTGfiFE6mYkh58ATK6ZDr9fj6GvflL3m4OWGp/5aiPNzV+P25jAxSjdIfEo+th25K+gxM3PUWL/vNl4c2lDQ41bW7YRczPzmHLYeiYNW++D0i7OXXMDg7rWw4M22aBTkJU6BRGQRUs8BvV6PJeujBD/ukvVRNtfRVazW4MMlF/DjpuvIzit54L0FP11By2AffDyxJYb0qC1ShURkKVLPguUbo6HTCTu9+MrNN/DxxFZwdLCtjqFft8Xg05WXcSPuv0+/XYO/rzP+N7ox3n25GezsZDHpD5FsSD0HjoWn4OpNYaefvBSdiVOX09CphW2tcXbu6j289915HDqT9NB77353HqP71cWCt9qimo+zCNURmYekv8VMmTIF8fHxmDZtGhYsWFDWyQUA77//PoKDg6HRaNCwYUM4O0vnDznvTgqSwyKQeOQSri3bjkPjPoNvq/ro+PkrZdvoSjQIm7IIzaY8De/G929sBPZri5q92+DEtKVilU5mYsg5cOqd5ajWPgRBT3Uue63DpxOQdjbaqr/IAMCKjdcFv6gFgKUi3FStjCs3MtFhzA5sOnjnoU4u4P6ozm1H7qLjuB04d/WeCBUSkaVIPQdOXU7D5euZgh9386E7SL5XKPhxTZVfWIrer+7Fl79EPNTJ9beL0RkYOvUgFv1+TeDqiMjSpJwF6hItVm4Wfmrx9KxibDwQK/hxK2PWd+fx4uxj5XRy3ZeSXoQPFodj+PTDKC01bjpgIrJuUs4BQLx7NLZ2b2jv8QR0f2lXuZ1cwP1M/XV7DDqO3YG4JOms3UbyI9mOrsjISKxfvx5Vq1bFvHnzHnpfoVCgdevWAB6ctjAhIQFTpkxBp06d4OLiAoVCgTt37ghVtkXcuxCD25vCUPeZrvBr1aDs9cyIWFxdsg1dv58Ml4Aq6PjFazgzayWKUqW7GKeLiwvat28PFxfj5uG1deWdA+qsfJyYthTtP5kA52reqD2wA/w7heLUO8tFrvbxthy6I8pxz11NR2JqgSjHNlZaRhH6T9qHe1nFj902K7cEA9/YbzM/G1FlMAekkgNxohxXo9Fj17F4UY5tijHv/oXjFw1be2LK56ex9fAdyxZEZAXkmgOAtLLgxMVUpBvwPdcSxMogUyz7Mwqfrbps0LZbD8dh6oJTFq6IyDrINQuklAMajQ47jgo708/fth25K8rga1NcjcnEsGmHUKR+/NIfsYl56D9pPwqLNAJURmR+ku3oWrt2LXQ6HcaOHVthcDk6OgJ4sKPr5s2bWL9+Pby8vNC9e3dBahXC5YUbodNo0XLm6Adev/L9ZuhKNRhy4EuknLyG2G0nRKpQGIGBgVi0aBECAwPFLkVw5Z0DiYcvIm7nKXRbPAUdPn8FJ6f/AHVWvohVPl5BYSmiYo1bcNqcwiPTRTu2MZZvjEZSmuFPHdzLKsbiPyItWBGRdWAO2H4OAOJ+FttKDpy7eg/b/zLu4v+jpReh19vGRTuRqeScA4B0soA58HilpTrMWXbRqDbLN15HQgoHv5H0yTkLpJIDUbezBV23/d/yC0srfErW2iz4+QoKiw3vuIq6nY31+25bsCIiy5FsR9ehQ4cAAD169Khwm4SEBAAPdnR169YNqamp2L17N0aNGmXZIgWUdycFsdtOoHr35qjaPqTsdb1Wh9Sz0XDy8cTN9UdErFAYWq0W+fn50GrFCUMxVXQOnPv4V7gH+SPx8EUkHLogYoWGuXwjU9SRM+GRGaId21AajQ7LN0Qb3W7l5htQl8jvb4PkhTlg+zmg1+txIVq8z+LwKNu4wfnDn8ZPqXLlxv01B4ikTM45AEgnC8T8LL6TlI+MbHGeJjPG1iNxSEkvMqqNTqfHik3GX0cQ2Ro5ZwFzwEzHt4FBD/cyi/DnPuOn27W1qRmJ/ibZjq64uPvTCdSpU6fc99VqNc6cOQPgwY4upVKy/0lw5btN0Gm1aDnjnw68qm0bocGzPRG1ajfazX0Jdk4OIlZoeTExMejZsydiYmLELkUU5Z0DmiI18uPSkBUlziPfxopNFHe+4NuJuaIe3xAXojKQaMTTXH9LzyrGCQOnuCKyVcwB28+BzBw1cipYb0oIsYnWPbr1b1sPmza1lqntiGyF3HMAkEYW3E4Q95pA7GsSQ5j6eW5LUzMSmUruWcAcsP3jG2LviQSUmLD24vlrtrNsB9G/qcQuwFIKCu7/QRYXlz/Sas2aNcjJyYGPjw9q1Khh0VratGmDlJQUo9rY65X4CO2MapNy6hp+CRhe4fs5MYlYXfOfEFM5O6LLd//DhU9/R/Sv+9B/y1y0fn8Mzs7+2ajjAkDDBg1RqhB+4drhwyv+ecuTlnZ/lPKePXsQHh5uUJunn37a6Loqy5TfP2D8OWBOQp0DBQ6tALeh5b53bt0Q+Ps+eo5tf1/nsn/GHxhd4XYp6YVo++z2h17fuHk7jqwea0TFwiu2rw+4jzOp7YjRL8K5lFMYSlmy1zRA6YnklGTUrFlT7HLMwpgsMCUHANvJAjnkgFbhDni/XeH7j8uCyuZARmaO1f/t6KFAVpWPTWq7ZNkv+P2bZ8xaD1kX5oC0cwCQRxakekwEVP7lvmeuHAAqzoL+A4fCUWPdN4PT3cYCDg0ev+F/RN6Ik8xnA5VP7jkA8N6QFHIgx7k34Nyl3PeEyIEvvvoWy+cdMqJi4eU7dgBc+5vUtlW7rrDXcqYHKbOmLNDp/vnM6NKlCy5eNG7q5b9JtqPL398feXl5OH/+PNq0afPAe/Hx8XjvvfcAAM2aNbN4LSkpKUhMTDSqjYPCDqhmoYL+X9s5LyA//h6if9kLADg+dTGGHPwKd3efRcqpa0btKyk5CSV64R/5/rtD01BFRUVl/zS0rbG/O3MQ4vdvboKdA161Abfy3/L3dUHNaq4G7UZlpzR4238rLswX5Zwwiqsb4G5a08yMFCDPyn8+qhx3LaAEdFqt9Z/LBjImC0zJAYBZYAjBcsDOHfCu+G1Ds8DUHNDrNLbxt+NVCijtjW5WmJ+NwmQb+PnIdMyBsn8yB8xPsCxwVld4N8PSOQAA6WmpQJGV//0E5gEmTNiiKy2WzGcDVUDmOQDw3pAlCZYD1XIA5/LfEiIH8nOzkZ9m5X8/Ve4Bpv14SEtJBErY0SVpVpoFqammzzQl2Y6uJ598EjExMfj000/Rs2dPNGzYEABw+vRpjBs3DtnZ2QAenLbQUvz9yx9p9ij2eiVgwQEQNXq0QNCQztjWa3rZa3lxqQj/ZC06fzsJ23pOh6bA8HnHqwdUF+WJLldX4z6x//4C4+zsbHBbSz/xVx5L//4tQahzoMjeEZkVvJeS/vjp+vx9naGyU0Kj1T1yzvqK9uXqpICXCOeEMbQKPVL0WkBhZ1xDvR7VvLRQeVj3z0eVk2xnBx0ApZ0dAqz8XDaUMVlgSg4AzAJDCJUDeqiQpNcBivKnm35cFlQ2B+yghr8N/O2k6VJRqjR+ZJ6XcwFcbeDnI9MxB5gDliRUFqSrNFBX8J65cuBR+6rq6wZ7nXX//eQ65MCUibWcFOnwkchnA5VP7jkA8N6QJQmVA3lOKlS0sIQQOeDpZg83e+v++ymxU+MeAOj1gEJhcDuFrhABfs5QwLp/Pqoca8oCnU6H5ORkAEC1aqb37iv0er3eXEVZk7t376JFixbIysqCSqVCcHAwioqKcOvWLQwePBj5+fk4cuQIfv75Z7z44ovl7uOXX37BSy+9hNjY2ArX+rKU0sJirK1n3dOj/duYW2tg7+Ik+HHPnTtn1PbR0dF4/vnnsXr1agQHBxvUpm3btqaUVim29vsHhDsHElIKUKvPHya3jz8wGjWruSIhtQC1ehu/n5Ufd8HLzzQy+fhCGTXjsNGLjg7qVgs7FvexUEVkLWo+uQ6JaYWoUdUFCQefFbscszAmC0zJAYBZYAghvws0fWYzrt7MMqltZXNg2JN1sPGbXiYdW0i/bLuBl2aHGdXGw80eiQefhZuL8U+Cke1gDjAHLEmoLJj5zVl8+UuESW0rmwNuLvbIOTkOSqXhNw3FkJBSgDr910OrNe6Wz4EV/fBkB97clDK55wDAe0OWJFQOHD6ThF6v7DGpbWVzAACO/TwQXVsb/2CDkPR6PVqN2opL0RUNFy/ftOeb4Ou321uoKrIW1pQFBQUFcHO7P31Xfn6+0YMX/lb+UFgJCAwMRFhYGAYMGAAnJyfExcXBz88PK1euxLZt2xAdHQ1AmCe6yHrUr18f+/btQ/369cUuhUxUo5oLqlYRvlP1b60b+4p2bGP8b3Rj49s8a3wbIlvDHJAGMT+LbSUHRvWtCz9v4/LypaEN2clFkscckAYxP4tbhfhYfScXANT0d8UzveoY1SY4yBM921W3TEFEVoRZYPtahfiIdmyFAmgp4vENpVAoMPnZUKPa2Nkp8PoIwwcCEVkTyXZ0AUBoaCh27dqFvLw85Obm4tSpU3j55ZeRlZWF5ORkqFQqNG7MG7tyolKp4O3tDZVKsrN2Sp5CoUA3kUbNeHs4ILTeIxaGsSJdW/tjzqRWBm8/86Wm6NtZGgsREz0Kc0AaxMoBAOjayjYWSnB2UmHTN73g6GDY1/32Tf3w6ZQ2j9+QyMYxB6Shc4tqxszCZFa2kgMAsPT9TmhY29Ogbb3cHbDx61420YlHVFnMAtvn5eGIZg2riHLslsE+NjM47KWnGmDsoHoGb7/sg85oYGBuEFkbSXd0VSQi4v4UB8HBwXB0dHzo/Y0bN2Ljxo04f/48AGDPnj0P/H+yXQkJCZg+fToSEhLELoUq4dXh4owuGf9UQ9jb287H5uzXWuCLt9rCzq7ii1WlUoGPJ7bE528KPw0DkRiYA9Iwsm8QPNyEv7hsXM8LnVvazg3Orq39cWB5f/g+5smufp1rYv/yfnBx5s0ekj7mgDTU9HfFwK61BD+uQgFMsIFpzP/m6+2Eoz8PQLsmfo/crnZ1Nxz7eSBC69vGoD6iymIWSMOrw8X5PH7Nhp54UigU+HluN0waFfLI7Rwd7PDLvG6YMMx2Mo7ov2znjq0Z/d3RVdG0hSNGjMCIESOwZMkSAMCkSZMwYsQILF68WLAayTLy8/MRFhaG/Px8sUuhSujVvjoa1PYQ/Livj3z0FwNro1AoMOOlZojbOwofvd4Stau7lb1np1Tg/VeaI3bPSHw0sRUUYg2JJRIYc0Aa3Fzs8cKQBoIfd9KoEJv7vOza2h9xe0dh1ZyuaNvkn6m+FApg3KD6OPXbYOxe2gcebg4iVkkkHOaAdDzupp0lDOxaC3VquAt+3Mrw93XBqTWDcWBFPzzTqw5cnP4Z1OBgr8T6L3vgxo7haCrSkxFEYmAWSMO4QfXhKvBALQ83ezw3wPAnpKyBSqXEkvc7IXrbMLw5NvSB5UBUdgp8Oa0dEg+OxgtDhb++IjInWQ7bfFxHl15v3GKtYms3bzwC+7aBW62q2PrEW8i+Hg8AcPLxQNdFk+Fe2x/aklKcemcF0s5GP9TeKzgQHT6bAGdfT+hKNUgLv4Ez76+CrkQj9I9CldD7j9lw9vMCdDqUFhTjzPurkHntzoMbKRRoM3scavRoAaXKDqlno3H63R+hK9XY1HmgVCowa0JzvDQ7TLBjjuoXhPqBwneumUONaq74eFIrfDypFWr0+h1J94rg7+uM+ZM5RRWR1FX0HaHzN5NQtW0jaItLUJJfhLMf/PRwZli5N8eGYuXm6ygq1gpyvBpVXfD8YNtcx8HFWYXxTzfE+KcbluVAdT8XrP60u9ilEZEZVfa6UOXihB6r3oZPs7oAgD9Cxwtav7H6dq6JViE+uBCVIcjxFArg3Zdtc41vpVKBJzvUwJMdagBAWRb4eTthZN+6IldHRObi4OGCfpvmlP1/O2dHuNeuhj+avoyS7H916j3i3hAAhE4cgvojn4BCqUDOrSSceHMJSnILhf5xHsvDzQFTngvFZ6suC3bMN8c0sZlpC/+rUZAXFs7sgIUzO5TlQDUfZ7z9YlOxSyMyC1k/0dWsWTORKzGPuF2nsHvoB8iPT3vg9dbvj8W98Bhs7jwZx99cgm5LpkJh9/CvXKsuwZlZq7Cl61Rsf3IGHNxdEPr6EKHKJzM5+urX2N5rOrb3noFry7aj88JJD23T4Lle8G5cGzv6zMSWrlMBACETBgCwvfPghSEN0E+gNaX8vJ2w6N2OghzL0mztSQQiqpyKviPc3XsWW594C9t7z8DVJdvQffk0kSo0Xd2aHvhMwDWlfvyoC9xdbf+pJ+YAkXRV9rpQp9Hg6pJt2D9yrlAlV4pSqcAv87rBXiXMbY0pz4Xa1PS1j8IsIJKmktxCbO89o+x/N9YcROLhiw92cuHR94YCujVD/RHdsWvQLGzt/hYyr95By3efE/xnMdSHr7dASF0vQY7VpL43Zr1imwMe/os5QFIky46ukydPQq/Xo2/fvmKXYhapp6NQmJz50Ot1hnRE1C97AQAZl2+h6F42qrV7eHqHvNgUZEXFAQD0Oh3SL92Ce2BVyxZNZvfv0TX27i7Q6x5+MrFK49pIPnalbJRO4uELqDf8/mhuWzsPFAoFfvyoCzzdjbvpmJJeiITUAqSkGz4aaen7neBXxdnYEomIRFfRd4T4/eeh1+oAAPcu3IBbTb/7Q9VtzOTnQtG1lXE3HU3JgfFPN0R/EdaCISIyRmWvC3UlGiQfj0BJToHFazWXpg2r4KPXWxrVxpQcqB/ogU8FHFxBRGQO9Uf3QMy6Qw+9/qh7Q1Ua10HqmShoCooBAAkHw1FveDfhijaSk6MKv8zr9si1yf/LlBxQqRT4ZX43ODrYmVImEQlAllMXyoGjtxsUSiXUGbllr+XH34NrTd9HtAJUzo5oOKYXzs9fY+kSReHn54epU6fCz+/Ri/Haqi7fT0ZAp1AAwIHn5j/0fvqlm2g4tjeif90HXYkGQUO7wK3Ww/8tbOU8qOnvim3fPYl+E/ehWG3Y1FVtn91u1DHmTGqF4X2CTCmPiKyQ1HPAFI0nDETC4QuAjU3dDNwfzb/x617o+uIu3IjLMaiNsTnwRNsALH5PGk/1EpH8csDU60Jb8t6E5oi+k401O28ZtL2xOeDn7YRdi/vAReB1YIjIcuSQBX6tG8LJ2w3xB8Ifeu9R94YyLt9Co3G94eTrieL0HNQd1g0O7i5w8HJ76Mkwa9GuqR9WfdwVL84+ZtD2xuaAQgH8Oq87WjeWTnYSSRG/qUnYf9cae9xjqQqVHbovn4bEvy4jft85S5YmGh8fH4wZM0bsMizm+JRFAIB6I7qjzYfP4+DYTx94/9aGo3Cr6Yf+W+ZBU1SM5GMR8O/c5IFtbO086N4mADsX9cHQqQdQUGTe9cQ+ntgSs19rYdZ9EpG4pJ4Dxqo7rCvqDO6IPU/NFrsUk1X1ccaRVf3R+7W9iLyVbdZ992pfHVu/exLOTvzKTCQVcswBY68LbY1SqcDPc7tBAQV+23nTrPsO8HPBgeX90LCOp1n3S0TikkMWNHi2J25uOFo2i8O/PereUMqpa7i2fAeeXDMLeo0WcbvPAEDZ01/W6oWhDaDT6zHh4+PQlTPDkans7BT4eW5XPDewntn2SUSWIcupC+VAnXV/lIWjj0fZa641fVGQkF7u9gqVHZ5YMQ1FaVk4O/snQWoUQ25uLg4ePIjc3NzHb2zDbm04Cv9OoXD0dnvovcsLN2JHnxnYM3Q2cm4mIicmoew9Wz0PenWojpO/DUaL4Cpm2V8VT0f8/vkT+GhiK8ndCCCSO7nkgCHqDOmEFtNGYv/oeSjOsO3/HtWruuL4r4MwxkwXoEqlAjNfaopdS/rY7GLTRFQ+ueWAsdeFtkqlUuKX+d3w1fR2cHI0z7RSvTtWx5m1gxFa39ss+yMi6yH1LFA5O6LOkE64+cfhCrd51L2h66v3Y2e/d7Br0Cykno5EQWJ62VSG1uylpxpi/7J+CAxwNcv+6lR3w8EV/TFucAOz7I+ILIsdXRIWt+MUQl7sBwDwaV4PzlW9kHo26qHtFHZKdF/2FtRZ+Tj59jKhyxRUUlISZs2ahaSkJLFLMSt7N2c4V/vnAiywfzuos/LLLmz/ZudoDwcPFwCAYxV3NP3fU4hYshWA7Z8HzRpWwdm1QzFnUis42Jv+0fZ0r9q4tuUZPDuAo3WIpEiqOWCsOoM7otU7z2LfqDkoSJTGzU5vD0es+ewJbP3uSQT4uZi8n5C6Xjjx6yAseKsd5+AnkiA55oCh14W2TqlUYPoLTXHpz6fQqYXpaw17uNljxYedsW9ZP9Tyf3jgIBHZPqlnQZ3BHZEVFYecm+X/fI+6NwQAzlW97m/n7IAWM0cjYuk2S5dsNr06VEfEpmfw6vBGJi9BrFAAE0cGI2LzM3iibYB5CyQii+E8LBLQ/tMJCOzbFs5VvdD3zw9RWlCMzZ0m4/z8Nei6eAqeObEI2lINwiYvKntkucWMUShKzcL11fsRNLQz6gzsgMxrdzDkwJcAgNRz13Fm1koxfywygr2HC3qsfBsqJwfodXoUZ+Ti0POfAQA6ffU64vefR/z+87B3d0H/LXOh0+qgtFMi8sddSPj/+ZqlcB7Y2yvx4est8fqIYPy09QaWbYhGXNLj55B2c7HHuEH1MHFkCJo2NM9TYURE1qCi7wjdlkxFUVo2ev38Ttm2+0bOeWiAhC0a2qM2+nWuiU0H7mDpn1E4cTH1sW0UCmBA11p4Y1QI+nauCaWST/MSke2p7HUhAAze/wWcq3rDwdMVI8KXI+XkVYRNXiTmj2W0RkFeOP7rIJy4mIql66Ow8cAdlGoenrrrv5o28MakUSEYM7Ae3F0dBKiUiMgyGjzXCzG/P/g0l6H3hgCgzx+zAaUSdvYq3Np4FNE/7RH6R6gUDzcHLP+wC94d3xzLN0Zj1ZYbSM96/BNpft5OmPBMI7w6vBHq1HAXoFIiMid2dEnAmVkry+2MKE7PwYHR88ptc+nL9WX/fntzGG5vDrNYfWR5hUkZ2DXgvXLf+/fTWcXpOdjSdWq520npPKjq44x3X26OGS82xdWbWQiPzMD5a/cQl5wPdYkO9iolfL0d0SrEF61DfNC6sS9cOTUVEUlQRd8RVgeOFqEa4Tg62OG5gfXw3MB6uJOYh/PX0hEelY7IW9koLNZAqVTAzdkeTRt4o02oL9o28UM1H2exyyYiqpTKXhcCwI4+My1Sm9AUCgW6tPJHl1b+WPK+GuevpeP8tXRcup6BnLwSaLR6ODvaoUFtD7Ru7IvWjX3RqI4npy0nIkkob/1dQ+8NAcC2ntMtUpfQgmq64/M322LOpFa4FJ2B8Mh0hEdlIPleIUpKdXB0UMLf1wVt/j8HWgRXgYM9Z3QgslXs6CIiybKzU6J5Ix80b+SD8U83FLscIiISQZ0a7qhTwx3D+wSJXQoREYnA28MRvTvWQO+ONcQuhYiIRODoYIf2zaqifTPTp7UlIuvHNbpIVhwdHdGoUSM4OjqKXQoREYmAOUBEJG/MASIiYhYQEUkPn+giWQkKCsJvv/0mdhlERCQS5gARkbwxB4iIiFlARCQ9fKKLiIiIiIiIiIiIiIiIbBI7ukhWrl+/js6dO+P69etil0JERCJgDhARyRtzgIiImAVERNLDji6SFb1ej9LSUuj1erFLISIiETAHiIjkjTlARETMAiIi6eEaXVZK5eyIMbfWiF2GwVTOXMDTnGzt9w/wHCAiMjdbywLmABGRedlaDgDMAiIic2IOEBEZjh1dVkqhUMDexUnsMkgk/P0TERGzgIhI3pgDRETyxhwgIjIcpy4kIiIiIiIiIiIiIiIim8QnukhW6tSpg3Xr1qFGjRpil0JERCJgDhARyRtzgIiImAVERNLDji6SFScnJ9SrV0/sMoiISCTMASIieWMOEBERs4CISHo4dSHJSnJyMubPn4/k5GSxSyEiIhEwB4iI5I05QEREzAIiIulhRxfJSk5ODrZv346cnByxSyEiIhEwB4iI5I05QEREzAIiIulhRxcRERERERERERERERHZJHZ0ERERERERERERERERkU1iRxcRERERERERERERERHZJHZ0kawolUq0bNkSSiVPfSIiOWIOEBHJG3OAiIiYBURE0sNPdJIVnU6HixcvQqfTiV0KERGJgDlARCRvzAEiImIWEBFJDzu6iIiIiIiIiIiIiIiIyCaxo4uIiIiIiIiIiIiIiIhsEju6iIiIiIiIiIiIiIiIyCaxo4tkxd3dHf369YO7u7vYpRARkQiYA0RE8sYcICIiZgERkfSoxC6ASEg1atTA3LlzxS6DiIhEwhwgIpI35gARETELiIikh090kayo1WrEx8dDrVaLXQoREYmAOUBEJG/MASIiYhYQEUkPO7pIVmJjYzFs2DDExsaKXQoREYmAOUBEJG/MASIiYhYQEUkPpy60Unq9Hpoi2xlZonJ2hEKhELsMybC13z/Ac4CIyNxsLQuYA0RE5mVrOQAwC4iIzIk5QERkOHZ0WSlNkRpr640VuwyDjbm1BvYuTmKXIRm29vsHeA4QEZmbrWUBc4CIyLxsLQcAZgERkTkxB4iIDMepC4mIiIiIiIiIiIiIiMgmsaOLiIiIiIiIiIiIiIiIbBKnLiRZCQ4OxtmzZ8Uug4iIRMIcICKSN+YAERExC4iIpIdPdBEREREREREREREREZFNYkcXyUpcXBzGjx+PuLg4sUshIiIRMAeIiOSNOUBERMwCIiLp4dSFJCtFRUW4evUqioqKxC6FyOJKSrXYdyIRZyLSEB6Zgas3s5B0rxAAkJxehL6v70XrEF+0a+qLfp1rwsmRkUDSxxwgOdHr9TgbcQ/HwlNwPjIdF6Mz/smBe4XoMGY7Wjf2RZtQX/TvUhP+vi4iV0xkecwBkpu4pDzsPZGI8Mh0hEemIz6lAOnZxQCA1IwijJv1F1qH+KJn++po1rCKyNUSCYNZQHKSV1CC3WEJOHf1HsKjMnD9Tg5S0u9fE6SkF2HI5ANo3dgHHZtXRa/21WFnx+diyDbxriYRkcTcTc7H8g3RWLn5OtIyi8vdRqfTY//JROw/mQgA8PFyxPinGmLiyBAE1XQXslwiIjKzgsJSrNl1C0vXR+HKjcxyt9HpgTMR93Am4h4AQKVSYNiTdTBpZAi6tQkQslwiIjIzvV6PvccTsHR9FHaFxUOvL387jVaPNTtvYc3OWwCATi2qYuLIEIzoEwRHBzsBKyYiInO7djMLS9dHYfWOm8gvLC13G61Ojx1H72LH0bsAgMAAV7w2PBgTnmmEqj7OQpZLVGns6JIQ/46h6Ld5zgOvlRYUIfd2Mm5tPIaoVbuh1+pEqo6EwHNA3rRaHb5dcw0fLA5HsVprVNuMbDW+/CUC3629ho9eb4mZLzWDSsVRPES2hjlAB08n4uWPwnA3ucCodhqNHuv3xmL93lgM710HS2Z14sUtkY1iFsjbncQ8vPxRGA6fTTa67clLaTh5KQ2fr7qMX+Z3Q5tQPwtUSESWxhyQt6JiDT5YHI6Fv12tcKBDRe4mF+D9ReH44pcILJzRHi8ObQCFQmGZQonMjB1dEhS79QTiD54HFAo4+3mh/ojuaDfnRXg2qIFTM5aLXR4JgOeA/NxJzMNz7/6FU5fTKrWfklId3l8Ujs2H4rBuwRNoUNvTTBUSkZCYA/KjLtHizQWnsWxDdKX3tfHAHfx1PgWrPu6CIT1qm6E6IhIDs0B+ftpyA1MXnK5w5L6hrt3KRoexO/Dey80xZ1IrKJW8yUlki5gD8hMemY7n3vkLN+JyKrWfnLwSjP8wDBsPxOKXed3gV4UD4Mj6cbi+BGVei8XtTWG4vfEYrv2wHbsGzkJ+4j00fK4XHH08xC5PVAEBAZgzZw4CAqQ9JQ/PAXmJjs1G5xd2VrqT69/CI9PR5YWduHw9w2z7JLIGzAHmgBQVFmkwePJ+s3Ry/S09qxhPvXkQqzZfN9s+iayBXHIAYBbIzfwVF/HyR2GV7uT6m1arx/wVlzBu1lFoNHzyg6RFLlnAHJCXv84l44nxuyvdyfVvu8MS0O2lXUhMNW62CCIxsKNLBjRFaqRfuAmFUgmP2tXELkdUnp6e6N+/Pzw95fWUCs8B6YpNyEOvV/YgKa3Q7PtOyyxG71f34npsttn3TSQW5gBzQGpKSrV45q2DOHAqyez71uuBV+Ycx9pdN82+byKxyDUHAGaBlH31SwRmL75gkX3/vvsWXv4oDDqdkfNfEVkxuWYBc0C6Tl9Ow6D/7TfbYId/i47NQe/X9iI9q/w14ImsBTu6ZMK9zv0AK87KE7kScWVlZWHDhg3IysoSuxTB8RyQHo1Gh5EzDlukk+tv97KKMXz6YahLjFvzi8haMQeYA1Lz8dKL2Hcy0WL71+uBlz86jqjb2RY7BpGQ5JwDALNAio6eT8aMb85a9Bird9zEsj+jLHoMIiHJOQuYA9KTlavGsOmHUFCksdgxom5nY/yHYdAbu+gXkYDY0SVBds6OcKziDkcfD3gFB6L9pxPg07Qu7l2MQV5sitjliSo1NRVffvklUlNTxS7FongOyMNXv0bg/LV0o9qcWzcE8QdG49y6IQa3uXozC/NXXDKyOiLrxBxgDkjJuav3sODnK8a1MSEH1CVavPjBMU5dRZIglxwAmAVyUFBYivEfhhndzpQsmLnwHGITeGOcpEEuWcAckIdpX54xegC0KTmw4+hdrNnJmR7IeqnELkAI6enp+Pzzz7FlyxYkJibCz88PI0eOxPz58/HKK69g7dq1+PHHHzFhwgSxSzWLFtNGoMW0EQ+8FrfnDE6/86NIFZHQeA5I3634XHy01PjpSfx9XVCzmqvR7T5bdRmj+gahSYMqRre1FkXFGty8m4uCIg083R3QsLYH7Ow43oOkiTkgfTqd3qSppEzNgbNX72Hxuki8Oa6J0W2thV6vR0xcLjJz1HBxVqFuTXe4udiLXRaRxTALpG/Osou4bULnkylZUFCkwcT5J7B3WT+jj2dNktIKkHSvEAqFAjWruaKaj7PYJRFZDHNA+g6dTsIv22KMbmfqNcGUz09jQNda8PFyMrqttcjNL0FsYh6K1Vr4eDmhXi13KBQKscsiM5B8R9eVK1fQp08fpKamwtXVFY0bN0ZiYiK++eYb3LlzB3fv3gUANG/eXORKzefG2oOI3XYCSpUdvBoFounkp+Ds6wVNcUnZNkoHFQbv+wK3txxHxPeby17v8u0bcPLzwsExn4hROpmJIefAEyumQ6/X4+hr35S95uDlhqf+Wojzc1fj9mbjRwaScJb8EYmSUuFG1mu1eixaF4nlH3YR7Jjmcis+Fz+sj8JPW28gK/efv4E61d3w2ohgvPx0Q/hV4QUuSQtzQPoOnk5ERIyw0+18u/YaJj/X2OYGCeTkleDX7TH44c8oRMf+szi3m4sKzw9ugEmjQhBa31vECoksg1kgbfmFpVi2IVrQY+47mYhrN7Ns7jNTq9Vhx9G7WLo+6oE1LRUKYGDXWpg4KgT9OteEUskbnSQtzAHp++a3CEGPl51Xgl+2xWD6C00FPa45XIhMx9L1Ufh9zy0UFf+zPEezhlUwcWQwxg6qz0FwNs62rlKNlJ6ejkGDBiE1NRUzZ85ESkoKLly4gNTUVMybNw+bN2/GpUuXoFQq0aSJ7Y5O/a+8OylIDotA4pFLuLZsOw6N+wy+reqj4+evlG2jK9EgbMoiNJvyNLwb1wYABPZri5q92+DEtKVilU5mYsg5cOqd5ajWPgRBT3Uue63DpxOQdjaaX2SsXGGRBj9vNX7ETmWt2XkL2blqwY9bGev33kbjpzbh69VXH+jkAoA7Sfl477vzaPLMZpy/dk+kCoksgzkgfUv+EH6tlLikfOw5niD4cSsjOjYbLUZswdQFpx/o5AKA/EINlq6PQrPhW/DDeq49Q9LDLJC2NTtvIq+gVPDjLrWxz8uCwlI8NfUgnn7z0AOdXMD9dSh3HovHwDf2Y8y7f3FdYpIc5oC03U7IFeW7+Q9/Rhk9q4SY9Ho9Fvx0Ga1Hb8OqLTce6OQCgCs3MjFx/km0e2477iRyil5bJumOrilTpiA+Ph7Tpk3DggUL4ObmVvbe+++/j+DgYGg0GjRs2BDOztIdzX/vQgxubwpD3We6wq9Vg7LXMyNicXXJNnT9fjJcAqqg4xev4cyslShKle5inC4uLmjfvj1cXFzELkVQ5Z0D6qx8nJi2FO0/mQDnat6oPbAD/DuF4tQ7y0Wulh5n6+E4ZOeVPH5DMyss1mDD/ljBj2uq7Ufi8Ny7fz32ybe0zGL0fnUvIm9J97OP/sEcYA5Iwb3MIuw8Fi/KscUYaGGqu8n56DlhD+4k5T9yO51Oj0mfnMRPW24IVBmJSa45ADALpEasz+Pfdt60mTUbNRodhk8/bFBm/rH3Np5//6hN3bwl08k1C5gD0rJ6+03oRfjIuhWfh+MXbGeNt2/XXMO7355/7HZRt7PR65U9SMsoEqAqsgTJdnRFRkZi/fr1qFq1KubNm/fQ+wqFAq1btwbw4LSFGzduxLBhw1C7dm24uLggODgY77//PvLzH32BbO0uL9wInUaLljNHP/D6le83Q1eqwZADXyLl5DXEbjshUoXCCAwMxKJFixAYGCh2KYIr7xxIPHwRcTtPodviKejw+Ss4Of0HqLNs+1yXg9MRaaId+0yEbTz5VKzWYLwRa9dk55Vg4vyTFq6KrAFzgDkgBeevpYt2I07MDDLW9K/OIPme4Qtzv/HpSWRkF1uwIrIGcs4BgFkgFeoSLS5GZ4hy7LyCUkTdzhbl2MZaveMm9p4w/GmHP/fFYuvhOAtWRNZCzlnAHJCO01fE+15++opt3BuKT8nH21+fNXj72wl5+GBxuAUrIkuSbEfX2rVrodPpMHbs2ApHaDg6OgJ4sKPrq6++gp2dHT799FPs2bMHr7/+OpYuXYq+fftCp7ONUUvlybuTgthtJ1C9e3NUbR9S9rpeq0Pq2Wg4+Xji5vojIlYoDK1Wi/z8fGi18puSoKJz4NzHv8I9yB+Jhy8i4dAFESskQ4VHpot37Cjxjm2MDftjkZFt3DSLx8JTcO0mn+qSOuYAc0AKxPwsTkorREq64Z1HYklKKzD6ZmWxWmvSYt5kW+ScAwCzQCoiYjJRKuJTVbZwTaDX67Hkj0ij2/3wp21NzUimkXMWMAekQa/XIzxKnAEPgG3kAACs2Hjd6AGCa3fZ3rIddJ9kO7oOHToEAOjRo0eF2yQk3B/Z8++Orh07duDPP//EmDFj0L17d7z55ptYtGgRTp48iePHj1u2aAu78t0m6LRatJwxquy1qm0bocGzPRG1ajfazX0Jdk4OIlZoeTExMejZsydiYuR5E6O8c0BTpEZ+XBqyou6KWBkZIyJGvM6YqzezbGI6j1UmTj+1ast1M1dC1oY5wByQAjFzALg/j721+23nTWi0xufVys3MAamTew4AzAIpYA483pUbmbhgwk3gg6eTuEaLDMg9C5gDti81owjpWeLNRGALOQCYdm+osFiDP/betkA1ZGkKvV6M2TwtLyAgACkpKYiIiECTJk0eel+tVqNatWrIyclBQkICatSoUeG+oqKi0LhxY/z+++949tlnja6lTZs2SEkxbu5Se70SH+naGX0sY6icHTHk0FeIXLEL0b/uQ/8tc5ERcRtnZ/9s9L7mKM+iVCH8iLLhw4cbtX1aWhrWrVuHZ599FlWrVjWozcaNG00prVKE+P3/W79NcxB/IBzXlm03eR9inQNyoweQVGVOhe+fWzcE/r4VzzPu7+sMlZ0SGq0OKekVzzuckl6Its+Wfz4EZM6HEsIvfG2MFM+3oLXzMrqdU0kUfPL/MH9BViTZaxp0Sk8odTkIyP5G7HLMwpgsMCUHAOlnAXPAtqS7jYXaoUG575krB4CKs6BK3h9wLrXuEe9ZLoNR6NTG+Ib6UtTImm/+gqwIc4A5UBFmge3Id2yPHNcB5b73uBwAKn9N4FIcDu9C088TIRTZByPT3fj7NwDgm/szHDV3zFuQFZF7DgC8N1QR5oDtKFX6IM1rSoXvW/rekC18fuihQlKV2Sa1dSs6Ds+iA2auyLpYUxbodDokJycDAFq0aIGLFy+atB+VOYuyJgUFBQCA4uLye7fXrFmDnJwc+Pj4PLKTCwCOHTsGAAgODjaplpSUFCQmJhrVxkFhB1Qz6XAGazvnBeTH30P0L3sBAMenLsaQg1/h7u6zSDl1zah9JSUnoUQv/CPff/+eDVVUVFT2T0PbGvu7Mwchfv/mJtY5IEtVKn7L39cFNau5PnYXKjulQduVJzk5GdBZ+WPc7nrAzvhmxcVqUf7mBeWuBZSATquVzM9qTBaYkgMAs8AQzAEB1VEDFTyEL0QOZGZmAblW/vlRowhwMqWhQjKfjRViDpT9kzlgfswCgfjkABV8hBuaA4DpWVBYWIhCa//88PAH3E1rmp6eARRY+c9XGTLPAYD3hiyJOSAQBw3gVfHblr4m0Gn11v/5oXB45P2zR8nPL0B+ipX/fJVlpVmQmppqclvJdnT5+/sjLy8P58+fR5s2D47mjI+Px3vvvQcAaNas2SP3k5iYiA8//BB9+vRBy5YtTa7FWPZ6JWDBARA1erRA0JDO2NZretlreXGpCP9kLTp/Ownbek6HpsDwR2CrB1QXZcSGq6txH8Z/f4FxdnY2uO3jOkItwdK/f0sQ6xyQoyR9CfSK8u9wPm7dFGNG7ZRLr0P1AD8oYN0PA6cpClEKb6PbuTho4C3C37yQku3soAOgtLNDgER+VmOywJQcAJgFhmAOCCfDUYmKvqWZKwcetS+fKu5wcrfuz49cZx1MmXjKTpcPf4l8NlaEOcAcsCRmgTAKHF2RXcF7hqyjWNlrAlcXe3hZ+eeHWmUPo1eQ0esBhQJVqzjB3su6f77KkHsOALw3ZEnMAWFolB54VHeApe8N2Sm1Vv+dWQ8gWV8MvcL40W8eroC7lf98lWVNWfDvJ7qqVTO9d1+yUxdOmjQJP/zwA2rVqoWDBw+iYcOGAIDTp09j3LhxiIuLQ2lpKd58800sXLiw3H3k5+fjiSeeQEpKCs6ePYvq1asLVn9pYTHW1hsr2PEqa8ytNbB3MWnYbKWcO3fOqO2jo6Px/PPPY/Xq1QY/ode2bVtTSqsUW/v9A+KdA3LUauRWXIw2bdHR+AOjUbOaKxJSC1Crt/FT9DWq44no7cZNCyGGb3+7ire+PGN0u2M/D0TX1sYPTrAlNZ9ch8S0QtSo6oKEg6ZN52JtjMkCU3IAYBYYgjkgnPe/P49PV142qW1lcwAAbu8eiaCaJg6TF0h0bDZChm4yut17LzfHp1NNmPLQhjAHmAOWxCwQRlh4Crq9tMvk9pXNgqXvd8LEUSEmH18IOp0e9QduQKyR6221DPZB+PqhUCgUFqpMfHLPAYD3hiyJOSAMnU4Pr86/Ia/AtGUlKpsDg7sHYvui3iYdW0iT5p/AD39GG9XGzk6BO3tGoaa/abNf2ApryoKCggK4ubkBuN8fY+zghb8pzVmUNXn33Xfh7e2N+Ph4hIaGomnTpqhfvz46duyIkJAQdOnSBQDQvHnzctsXFxdj6NChiI2Nxf79+wXt5CLLqV+/Pvbt24f69euLXQqRyVo39hHx2L6iHdsYLwxtAGcn4+YubFLfG11a2di8EGQ05gBJgZifxd4eDqhTw0204xsqOMgLPdsFGNVGqVTgtRGNLFQRWQvmAElByxAfiNkPYwvXBEqlAhNHGr/8xKRRIZLu5KL7mAVk65RKBVoGi3lvSLxjG2PiSOMHZQx9orbkO7mkSrIdXYGBgQgLC8OAAQPg5OSEuLg4+Pn5YeXKldi2bRuio+/35pbX0VVaWoqRI0fi7Nmz2LNnDxo3bix0+WQhKpUK3t7eUKkkO2snyUDbJn6iHbtNqPVf1AKAt4cjPpls+Ih8lZ0CC2e250WtDDAHSArE/Cxu28TPZj4rv5jWDi5Ohv+tz3ypKWpXt+4n1ajymAMkBW4u9gip6yXKse1VSjRraPwU4WJ4bUQwmjYwvNb2Tf0wZmA9C1ZE1oJZQFLQVuRrAlvQtGEVvD7C8EEPnm4OmD+5tQUrIkuSbEcXAISGhmLXrl3Iy8tDbm4uTp06hZdffhlZWVlITk6GSqV6qBNLp9Nh3LhxOHDgALZv34527dqJVD1ZQkJCAqZPn46EhASxSyEy2TO96sDRwbinlczBzk6BkX2CBD+uqd4cG4qPXn/82ooO9kr8vqAHnuwg7fmX6T7mAElBYIAburQU5wnUMQNs5wZg68a+2Pb9k3BzefxNrEmjQowaIEG2izlAUiHW5/GwJ+vAydE2Ogc83BywZ2lfhNbzeuy2bUJ9sWNRbzgbMUCCbBezgKTgOZFyoGoVJ/Rqbzszny16ryPGDXr805tVPB2xe2kf0QaSUOVJuqOrIhEREQCA4OBgODo6PvDe//73P6xfvx7Tp0+Hs7MzTp8+XfY/BqDty8/PR1hYGPLz88Uuhchkvt5OGNVX+A6np3rURo1qtvP4tkKhwMeTWmHP0r7o0+nhTiwHeyXGDKyH02sGY4QNdeBR5TAHSComibA2ShVPR4wUIX8q48kONXBu3VC8MqxRuU93dW1VDX9+1ROLZ3WEUmkbT6pR5TAHSCpefqYR7FXC39IRI38qo0Y1V5xYPRjz/tcKNcu5lgmq4Y4Fb7bFX6sGwK+KswgVkhiYBSQFrRr7on1T4Z+sevnpRqIMvjaVSqXEr590w+pPupX738vd1R5vjA7BuXVD0KkFl7OwZbLu6Cpv2sLdu3cDAD755BN07Njxgf+tXLlS0DqJiCoy+Tnhp1QV45jm0K9LTexb1g83dgyHt4cDgPtrzMTvH401nz2BliG2MR0jEdG/DetdB9Wrugh6zFeGNbKZUfz/FhzkhRUfdUHiwdHY+HVPeLnfz4KqVZxw7JdBGNEnyGamYyQi+ls1H2eM7ldX0GO2CK5ik2vaero74INXWyJ2z0jsW9a3LAd8vRwRs3M4Zo5vBlcXe5GrJCIy3pTnQgU9nr1KaZNr2ioUCowb3ACn1w7BhfVDy+4NVfF0QOLB0Vg8qxPq1vQQuUqqLNu7UjWDR3V03blzR+BqzKP3H7Ph7OcF6HQoLSjGmfdXIfPanYe28woORIdPXoaTnycUCgXCP/sdd3efAQCEThyC+iOfgEKpQM6tJJx4cwlKcguF/UHIZMPPLoVWXQptcQkA4MqiLbiz/WS529o52mPw/i+gKSrBzn7vAPj/c+OzCXD29YSuVIO08Bs48/4q6Eo0gv0MZLg2oX6YODIYP/wZLcjxnh9cH93bBAhyLEtpUNsTLk4qZOWWwMVJhao+HLFJ0tFu3ngE9m0Dt1pVsfWJt5B9PR4A0PmbSajathG0xSUoyS/C2Q9+Kvf7wb/1+fMjVAmtjT9CxwtQOZnKwd4OP7zfCUOnHhTkeEE13PHBqy0EOZaleHk4YljvIExdcBrZeSWiPAlBZEnMAvn5Ylpb7Dx2F1m5JRY/llKpwPLZXWx6YIBKpUSfTjXh6qxCdl4JHB3sYGfHLCDpqGwOuNX0wzOnFiM7+m7Za0cmfIW8uFShfgQy0rMD6mLVlus4fDZZkON99HpLm1/TtmWIb9m9IWdHFdxdHcQuicxE1h1dzZo1E7kS8zn66tdlnVKB/dqi88JJ2NFn5gPb2Dk7oOfPM3F86mKknY2Gwk4JBy83AEBAt2aoP6I7dg2aBU1BMZpPG4GW7z6HM7P4FJstOTLhq7IvMo/S8t1nkXb+BqqE1il7TasuwZlZq5AVFQeFUoluS6ci9PUhiPh+swUrpspY8FZb7D6egLgkw6dbSEkvfOCfhgjwc8G373Qwuj4iEk7crlO4unQrBmyb/8Drd/eexckZy6DX6lDzydbovnwatnSZUuF+gsf3R358GqqE1rZ0yWQGQ3rUxpiB9bB21y2D25iSAwDw09yucONodyKrxiyQH39fFyx6ryPGvnfUqHamZMGMF5uinQhTZBGR4cyRAyW5Bdjee4YQ5ZIZKBQKrJrTFU2HbUF+YanB7UzJgdaNffHOeOncSyfpkWVH18mT5T/lYsv+/eSVvbsL9Dr9Q9vUfbor7oXfQNrZ+0+A6LU6qDNyAQBVGtdB6pkoaAqKAQAJB8PRd+PHkuvo8vPzw9SpU+HnJ98v6FXbh8AjKACRy3c+0NGVF5tS9u96nQ7pl27Bs77tLC4pR+6uDvhjQQ/0fGU3ioq1BrVp++x2o47hYK/E2s+6w9vD8fEbE9kAqeZA6umocl+P33++7N/vXbgBt5p+gEIB6B/+nuAe5I+goZ1x/M3FCOzX1mK1knkteq8jLkZnIPJWtkHbG5sDAPDhay3xRFvbfqqX6G9SzQGAWSBXzw2oh8Nnk/HTlhsGtzE2C7q0rIaPJ7Y0tjQiqyXVLDBHDpDtqVPDHSs/7oJn3zli8K/U2Bzw8XLE2s+6Q8UZEciKybKjS6q6fD8ZAZ3uz8164Ln5D73v1bAmtMUl6LX6PbgGVEFm1F2cm/Mr1Bm5yLh8C43G9YaTryeK03NQd1g3OLi7wMHLDSXZ0lmc08fHB2PGjBG7DIvptmQqFEoF7l2IQfhnv5d1ZP5N5eyIdnNfxOEXFsAjqOIbVipnRzQc0wvn56+xdMlUSR2aV8W273pj6NQDBnd2GcrBXokNX/VEj3bs8CTpkHoOPErjCQORcPhC+Re0CgU6fzURZ2athL7UvJ8lZFneHo44sLwfery8Bzficsy+/7fGheLjSby5SdIh5xwAmAVSpFAosHx2ZxQVa7Buz22z7799Uz/sXNzHJtdoJKqInLPgkTkAwN7NGYP2fA6FnRJ395zFle82Q6/TCVwlGWtUv7rILyzFK3OOm73/soqnI/Yv64dGQV7m3TGRmbEbVkKOT1mEDW1ex4UF69Dmw+cfel+pskONJ1rg1Mzl2N57BgqS0tHh0wkAgJRT13Bt+Q48uWYWBu78FEWpWQAAXam01mfKzc3FwYMHkZub+/iNbcyepz/E9iffxvY+M6HOykfX7/730DZtPhyH6J/3oTAls8L9KFR26L58GhL/uoz4fecsWTKZSe+ONbB/WT9Ur+pitn1WreKE3Uv6YkgPTllD0iLlHHiUusO6os7gjjg1Y3m57zeZOAQppyMfu2YLWafqVV0R9stAdG1VzWz7tLNT4NMpbfD12+1tej0Wov+Saw4AzAIpU6mU+O3T7nhzbKhZ9zu0RyAO/tgfnu5cv4SkRa5Z8LgcKEzLwoZWr2Fn/3exb+RcVGsfgtDXBwtcJZnq5WcaYf2XPeHuar7pxhvU9sDRnwagVWNfs+2TyFLY0SVBtzYchX+nUDh6uz3wen5COpJPXivr5Li9OQx+LeuXvX999X7s7PcOdg2ahdTTkShITC+bylAqkpKSMGvWLCQlJYlditkVJKYDAPQaLSJ/3Ilq7UMe2qZquxC0mDYcw88uRfdlb8I7OBBD/1pY9r5CZYcnVkxDUVoWzs7+SbDaqfK6tPLH1c3P4IUhDSq9r9H96uLalmHo1YFPcpH0SDkHKlJnSCe0mDYS+0fPQ3FG+Rfz1To0Rv1RT2D42aXov20+HDxdMfzsUjh4ugpcLZmqqo8z/vppIL6d2R7OTnaV2lfTBt44u3YI3pvQnJ1cJDlyzAGAWSAHdnZKLJzZAYdX9ked6m6Pb/AIXu4OWP1JN2z59kmuz0iSJMcsMCQHdCWasvdKsvMR88fhcu8tkfUa0ScIVzc/gz6dalRqPwoFMO35Jrj059No0qCKmaojsiw+ey4B9m7OULk6lT2FFdi/HdRZ+VBnPTjl4J0dJ9HguZ6wd3NGaX4RavRogcxrcWXvO1f1QlFaNuycHdBi5mhELN0m6M9BplM5O0Jpb1e2VlvQ012QcTX2oe2295pe9u/+HUPR5qPnsbPfOwAAhZ0S3Ze9BXVWPk6+vUyYwsmsvD0c8cv8bnh+cH18t/Yadh6Lh66c9frKo1AA/TrXxNQxoejbuaaFKyUiodQZ3BGt3nkW+0bNKRsQUZ5Dz39W9u9uNf0waN8CbGw3SYgSyYyUSgWmjm2CwU8EYuFv1/Dr9hjkFRi+KHVoPS9MGhWCCcMawcG+cp1lRGQ9mAXy0qNddURsfgZL/4jCsg3RiE3MM7itr7cTXn66IaaOCUWAn/lmiyAicRmaA04+HlDnFECv0ULpoELtAR3KvbdE1i0wwA17f+iLjQfuYPG6SBwLTzG4rb1KieG96+CtcU3Qtom01rAj6WNHlwTYe7igx8q3oXJygF6nR3FGbtlFSqevXkf8/vOI338eBYnpiFi0BQN2fgq9RovClEycnPFPh0afP2YDSiXs7FW4tfEoon/aI9aPREZy8vNEj5UzoLBTQqEA8uLScHzyIgAPngOPEjS0M+oM7IDMa3cw5MCXAIDUc9dxZtZKi9dP5tWzfXX0bF8dcUl5WLPzFs5evYfwyHQkphU+sF2Anwtah/igbRM/jBlYD/VqeYhUMRFVVvtPJyCwb1s4V/VC3z8/RGlBMTZ3moxuS6aiKC0bvX5+p2zbfSPnQJ2VjxYzRqEoNQvXV+8XsXKyhLo1PbDovY74dEprrN8bi2PhKQiPSkd0bM4DAyDcXOzRKsQHrRv74KketdG1tT+f4CKyYcwC+pubiz1mjm+Gt19sir0nErA7LB7hkem4dD0Txep/1l9TqRQIreeN1o190bNdAIY9WYdrcRHZsMrmQNX2IWg5YxT0Wh0UKjukHI/Ale82ifgTkakUCgVG9Am6/4RXTCb+2Hsb56+lIzwqA+lZD87eFVTDHa0b+6BDs6oYO6g+qvk4i1Q1UeXwG4wEFCZlYNeA98p9779P5tzacBS3Nhwtd9ttPaeX+zpZv/y7adjRZ0a571X0dFbKqWtlT3MB96eyvL05zCL1kThqV3fH+6+2KPv/2blqFBRpoNcDri4qeHs4ilccEZnVmVkryx2YsDpwdIVtLn25vtzX8xPu4Y/Q8WarjcTj7uqACcMaYcKwRgCAomINsvNKUKrRwcnBDr7eTlAq2bFFJBXMAvovpVKBAV1rYUDXWgAAjUaHzBw1iku0cLBXwtvDEY4OfIKXSCoqmwN3d5/B3d1nLFIbiadJgyqY///TD+r1emTmqFFYrIGdUgF3V3u4u3IdRpIGdnSRrDg6OqJRo0ZwdOQNfpIfLw9HeLFzi2SOOUBy5uykgrMTv/6TvDEHSM5UKiWqcqQ+EbOAZEuhUMDHywk+YhdCZAG80iVZCQoKwm+//SZ2GUREJBLmABGRvDEHiIiIWUBEJD1KsQsgIiIiIiIiIiIiIiIiMgU7ukhWrl+/js6dO+P69etil0JERCJgDhARyRtzgIiImAVERNLDji6SFb1ej9LSUuj1erFLISIiETAHiIjkjTlARETMAiIi6eEaXVZK5eyIMbfWiF2GwVTOXMDTnGzt9w/wHCAiMjdbywLmABGRedlaDgDMAiIic2IOEBEZjh1dVkqhUMDexUnsMkgk/P0TERGzgIhI3pgDRETyxhwgIjIcpy4kIiIiIiIiIiIiIiIim8QnukhW6tSpg3Xr1qFGjRpil0JERCJgDhARyRtzgIiImAVERNLDji6SFScnJ9SrV0/sMoiISCTMASIieWMOEBERs4CISHo4dSHJSnJyMubPn4/k5GSxSyEiIhEwB4iI5I05QEREzAIiIulhRxfJSk5ODrZv346cnByxSyEiIhEwB4iI5I05QEREzAIiIulhRxcRERERERERERERERHZJHZ0ERERERERERERERERkU1iRxcRERERERERERERERHZJHZ0kawolUq0bNkSSiVPfSIiOWIOEBHJG3OAiIiYBURE0sNPdJIVnU6HixcvQqfTiV0KERGJgDlARCRvzAEiImIWEBFJDzu6iIiIiIiIiIiIiIiIyCaxo4uIiIiIiIiIiIiIiIhsEju6iIiIiIiIiIiIiIiIyCaxo4tkxd3dHf369YO7u7vYpRARkQiYA0RE8sYcICIiZgERkfSoxC6ASEg1atTA3LlzxS6DiIhEwhwgIpI35gARETELiIikh090kayo1WrEx8dDrVaLXQoREYmAOUBEJG/MASIiYhYQEUkPO7pIVmJjYzFs2DDExsaKXQoREYmAOUBEJG/MASIiYhYQEUkPO7qIiIiIiIiIiIiIiIjIJnGNLiul1+uhKbKdR6hVzo5QKBRilyEZtvb7B3gOEBGZm61lAXOAiMi8bC0HAGYBEZE5MQeIiAzHji4rpSlSY229sWKXYbAxt9bA3sVJ7DIkw9Z+/wDPASIic7O1LGAOEBGZl63lAMAsICIyJ+YAEZHhOHUhERERERERERERERER2SQ+0UWyEhwcjLNnz4pdBhERiYQ5QEQkb8wBIiJiFhARSQ+f6CIiIiIiIiIiIiIiIiKbxI4ukpW4uDiMHz8ecXFxYpdCREQiYA4QEckbc4CIiJgFRETSw6kLSVaKiopw9epVFBUViV0KEQkkK1eNC5EZuBCVjpz8EgBATn4JFq+LROvGPmje0AcuzoxDuWAOEMmPukSLiJhMhEdmIDYxD9l597Mgr6AU+04koHVjX/h6c9F0uWAOEMmPTqfHzbu5CI9MR+Tt7LIcyM0vwZ/7bqN1Y1/UrekOhUIhcqUkFGYBkfykZhQhPDIdF6MyHrg3tHLTdbRu7IPQ+t5wsLcTuUqqDN7ZIyIiySkoLMXvu29h2YZoXIjKeOj9/EINJn92CgBgZ6dAz3YBmDQqBIO6BUKl4sPORES2TqfT48CpRCxdH4U9xxNQqtE9tE1uQSn6TdwHAGhczwuvDQ/G84Prw8vDUehyiYjIAiJvZeGHP6OxdtdNZOWWPPR+XqEGo2YcAQD4ejvh+cH1MXFkCOoHeghdKhERWUB6VjF+3noDP26+jpi43Ifezy/U4JU5xwEAjg52GNy9Ft4Y3Rjd2/hz8IMNYkeXhPh3DEW/zXMeeK20oAi5t5Nxa+MxRK3aDb324Yt8kg6eAyR3paU6fPnLFXzxSwRy8h6+mC2PVqvHgVNJOHAqCTWruWLBm23x7IC6/FJDNok5QATsOnYXb35xBjfvPnwxW5HIW9mYuuA03vvuPCY/1xgfT2wJJ0deKpFtYhaQ3MXE5eB/n53C/pOJBrdJzyrGN6uv4pvVVzHkiUAseq8jAgPcLFglkeUwB0ju8gtLMev781ix8TrUJVqD2qhLtNh44A42HriD0HpeWPReR/RoV93ClZI58epNgmK3nkD8wfOAQgFnPy/UH9Ed7ea8CM8GNXBqxnKxyyMB8BwgOYq4kYkXZx8r9wkuQyWkFmDMe39hw4FY/PBBJ/j7upixQiLhMAdIjrJy1XhzwWms3nHT5H0UFmuw4Kcr2HYkDj/P7YYOzauasUIiYTELSG50Oj2+X3sNsxadR1GxYTc2y7P9r7s4ci4ZX09vhwnDGnEAHNks5gDJ0ZGzSRj/YRjuJOWbvI9rt7LRc8IeTBoVggVvtYWbi70ZKyRL4fxMEpR5LRa3N4Xh9sZjuPbDduwaOAv5iffQ8LlecPSR9yP4AQEBmDNnDgICAsQuxaJ4DpDc7PjrLto+t71SnVz/tvVwHFqO3IqrMZlm2R9ZD+YAc4Ck6U5iHto+u61SnVz/Fh2bg84v7MSv22LMsj+yHnLJAYBZQPKiLtFixPTDeOvLM5Xq5PpbXkEpXp17Ai/NDoOmnOlvybbJJQuYAyQ3i36/hp4T9lSqk+vflq6PQsexO5B8r9As+yPLYkeXDGiK1Ei/cBMKpRIetauJXY6oPD090b9/f3h6eopdiqB4DpCUbT8Sh6ffOmjw4+iGSkkvQvfxu3HtZpZZ90viYg4wB0h64pLy0O2lXbgVn2fW/ep0erw4+xh+3nrDrPslcck1BwBmAUlXaakOw6YdwuZDd8y+71+3x+D5949Cy2neJEWuWcAcICn79rermPL5abPv9+rNLHQfvwupGUVm3zeZFzu6ZMK9zv0AK84y7w0AW5OVlYUNGzYgK0t+N655DpAUXYxKx8gZR6DV6i2y/8wcNfq+vheZOWqL7J+ExxxgDpC0FBZp0H/SfsSnFFjsGBM+Po7DZ5Istn8SlpxzAGAWkDRN+fwUdh2Lt9j+1+25jQ8Wh1ts/yQ8OWcBc4CkaOvhO3jryzMW239MXC4G/W8/Sks56MGasaNLguycHeFYxR2OPh7wCg5E+08nwKdpXdy7GIO82BSxyxNVamoqvvzyS6SmpopdikXxHCA5KCnV4sXZYUY9yXVu3RDEHxiNc+uGGNwmMa0QUz4/ZUqJZIWYA8wBkpbZS8IRdTvbqDbGZoFOp8f4j8KQV1BiQoVkbeSSAwCzgORh/8kELNsQbVQbU64Jvvg5AqcvpxlbHlkpuWQBc4DkID2rGK/NPWFUG1Ny4Py1dHz+02VjyyMBqcQuQAjp6en4/PPPsWXLFiQmJsLPzw8jR47E/Pnz8corr2Dt2rX48ccfMWHCBLFLNYsW00agxbQRD7wWt+cMTr/zo0gVkdB4DpAcfLLiMq7cMG4NLX9fF9Ss5mr0sdbuuoURfYIwtEdto9sSiYE5QHJw8lIqFv521eh2pmRBXFI+Zn5zDj/M7mz08YjEwiwgqcvNL8GEj48b3c6UHPh7OtuLfz4FZydZ3EojCWAOkBxM/uwU0jKLjWpj6r2hecsvYWiP2mjWsIrRbcnyJP9E15UrV9CkSRN8/fXXSE1NRePGjVFSUoJvvvkGY8eOxfXr1wEAzZs3F7lS87mx9iD2jZyDA8/Nx7k5q1GcmQtnXy9oiv8Zhap0UGHokW/QdMozD7Tt8u0beHLt+0KXTGZmyDnwxIrp6L582gPtHLzcMPLSj6j7TFehSyYySnauGl/+ekXQY36wKBx6vWWmSLS0nLwSLF4Xie4v/TOvdGaOGsfOJ9vsz0SPxhwgOfhwyQUI+RG2YtN1xCXZ5jQ/Wq0O24/EYeiUA0jNvJ8D6dnFWLPzJorVGpGrI0thFpDUrdh43aJT1/7X9Ts5WLvrlmDHM7dL0Rl4dc5xpP1/DtzLLMYnKy5x3RkJYw6Q1F2NycQfe28LdrxSjQ5zl10U7HjmlpJeiPkrLqLj2O1ln/1ZuSW4fD1D5MrMQ9IdXenp6Rg0aBBSU1Mxc+ZMpKSk4MKFC0hNTcW8efOwefNmXLp0CUqlEk2aNBG7XLPJu5OC5LAIJB65hGvLtuPQuM/g26o+On7+Stk2uhINwqYsQrMpT8O78f0nFAL7tUXN3m1wYtpSsUonMzHkHDj1znJUax+CoKf+GZnc4dMJSDsbjdubw8Qom8hgv26PQVGx4VMWmsPVm1k4fsG2prbQ6/X4bOVl1HhyHSZ/dgrHwlOg+f/1zIrUWnQfvxvNh2/B1Rjjnowj68ccIKmLjs3GIYHXzdLp9Fix8bqgxzSHg6cTUXfAnxg69SC2/3UXGs39HFCX6DBu1lHU6rMea3beFLlKsgRmAUmZTqfHD39GCX7cpeujbG6gWEJKAbq/tAstR27Fj5uuo/T/c6BEo8MHi8NRq/cfmPL5Ka49I0HMAZK6H/40bupac9h6JA6JqcINsjCHklIt3vjkJGr1/gOzF1/A6Sv3yu4NFRZr0GLEVvR4eTeS0mzr5/ovSXd0TZkyBfHx8Zg2bRoWLFgANze3svfef/99BAcHQ6PRoGHDhnB2dhaxUsu6dyEGtzeFoe4zXeHXqkHZ65kRsbi6ZBu6fj8ZLgFV0PGL13Bm1koUpUp3MU4XFxe0b98eLi4uYpciqPLOAXVWPk5MW4r2n0yAczVv1B7YAf6dQnHqneUiV0v0aHq9XpQvMwBEuZg2lV6vx5sLTmPW9+dRUFTxaP2ImCx0fmEnLkSmC1ideJgDzAGShuVGrsdiLj9uuo6SUmEHWlTG9iNx6D9xH+4mV3zRmp5VjHGzjmLJH5ECViYeueYAwCwgaTlwKhG3E4R/yvZidAbORtwT/Limupucj47jduBYeMXrMZVqdFj0eyRGvH0IGo08OrvkmgXMAZKS/MJSrN4h/GAtrVaPHzfZzuA3jUaHYW8dwtL1UWWdW+X561wyOo7bgQQBn5Q2N8l2dEVGRmL9+vWoWrUq5s2b99D7CoUCrVu3BvDgtIVhYWHo3bs3AgIC4OjoiJo1a2LUqFGIirKdm5vlubxwI3QaLVrOHP3A61e+3wxdqQZDDnyJlJPXELvNuMX7bE1gYCAWLVqEwMBAsUsRXHnnQOLhi4jbeQrdFk9Bh89fwcnpP0CdlS9ilUSPl5BagOt3ckQ59sHTSTYzgnP19pv4/nfDblrm5pdi0OQDKCgstXBV4mMOMAdIGg6cShTluPeyihERYxuDwmIT8jB65pFHXtD+2+TPTuH4BekvTC/nHACYBSQdYuUAcP+awBbo9Xo8NfUgEgx88mDbkbuYt/ySZYuyEnLOAuYAScWZK/eQL9I9jIMCzyxRGR8tvYCdx+IN2vZucgGemXbQZu57/ZdkO7rWrl0LnU6HsWPHVjhCw9HREcCDHV1ZWVlo0qQJvv/+e+zfvx8LFizAtWvX0LFjRyQkJAhSuyXk3UlB7LYTqN69Oaq2Dyl7Xa/VIfVsNJx8PHFz/RERKxSGVqtFfn4+tFrbGYlrLhWdA+c+/hXuQf5IPHwRCYcuiFghkWHCRXzy6F5WscEXimLS6/X46tcIo9ok3ysUdG5rsTAHmANk+woKSxEVK86AB0DcHDLGD39GoUht+GedXg98u+aaBSuyDnLOAYBZQNIRHiXeeiLhUbaRA0fOJuNitHH/nRb/EYmiYumv3SjnLGAOkFSI+Z38YlQGtFrrfwK2oLAUS9Yb9/DOuavpNrdsx98k29F16NAhAECPHj0q3Obvjqt/d3QNGTIECxcuxIgRI9C9e3eMGTMGmzZtQk5ODjZt2mTZoi3syneboNNq0XLGqLLXqrZthAbP9kTUqt1oN/cl2Dk5iFih5cXExKBnz56IiYkRuxRRlHcOaIrUyI9LQ1bUXRErIzLcBREvagHg/DXrv7A9fiEVV28a/8TBkj9sb80BYzEHmANk+y7fyIROJ95nlS10dBUVa7Bqyw2j29nimgPGknsOAMwCsn16vR4XROxssoXrAeD+emLGysxR4899sRaoxrrIPQuYAyQFYg46KCzWiDbTkDHW7bmNnLwSo9stWW+bU5or9BK9oxUQEICUlBRERESgSZMmD72vVqtRrVo15OTkICEhATVq1KhwXxkZGfD19cXixYvxxhtvGF1LmzZtkJJi3DQg9nolPtK1M/pYxlA5O2LIoa8QuWIXon/dh/5b5iIj4jbOzv7Z6H3NUZ5FqUL4nuzhw4cbtX1aWhrWrVuHZ599FlWr/l97dx5mZ1nYjf87kwkz2SDbQEIWgglhkkAAAVkFQXZZRKDYEsCiVqsUCiiV0PaVpbQUX8DX1rYaq2VppCAVZHlRwGooW8JSoJg0YBgmyTAQyAqTdeb3hy/5GUlgZk5mnpw5n891eeVinuc+53ty4rnnnO+572fHDo254447uhKtJD3x/P+24350RZp+9lT++x/v7vJtFPVvgMqztP9Jeaduv80emz3z5IwYvuV91kcM75eaPtVZv6Etry1pfd/7eW3JO9n/99/7/4nBb/8kA9bM6VzoHray7qNZ0f+oLo0d+dbVqU55bWHYmbmgK/NA0vvnAvMA5aS176S8NejTmz32QfNA0vG5YEvzQN3aX2XYqh92LnQPW9tnZN7Y4YtdGjt05Q/Tb115bdtuHtg6zAWUi/bUZPHQv9ji8a31nmBL80Da12fU0vdeImNb0zz4q2mrHvjBJ/6O/qufypB3uv46UASfDW0d5gHKyRuD/jBr+47b7LGe+Gxo+Irvp3b9K52J3OPeGnBqWmv37vS46rblGbns+q0faAva2trS3NycJNl7773zzDPPdOl2arZmqG3J22//5puIq1ev3uzxW265JcuXL8+wYcM2W3Jt2LAhbW1taWpqyvTp07PzzjvnzDPP3MwtfbDXXnstixZ1bv/o7ar6JDt16e46bP8rzs2qpjcy9wf/N0nyyIV/l5Mf/EZeve/JvPZY57YtWdy8OGvbe37J97vPc0e1trZu/LOjYzv73G0NPfH8b21F/RugAo1ek9Rt/tCI4f0zeqcBH3gTNX2qO3Te5ixbvirL3izumgAdstOapIvXVW5ueStZv2Lr5ulmnZkLujIPJOaCjjAP0GN22DkZtPlDHZ0Hkq7PBavXrC/kNaFT+vdPduja0LeWvZMs28Yf3+8wD2w7zAX0iOp+ydAtH+729wRVNdv+PJAkg/t2adg7qzfknXJ4fL/FZ0PbDvMAPWb8hmQLL3M98dnQkjeXJau28dfKsRuS2s4Pa2vvW9g819LS9W0Te23RNWLEiKxcuTJz5szJfvtt+s3/pqamXHbZZUmSqVOnbnb84Ycfnv/8z/9MkkyYMCEPP/xwhg8f3uUsndW3vTrpxi9AjDpi7+x68iG56+OXbPzZysaWPPVXt+aQG7+Uu468JOvf3nxJuDk7j9y5kG9sDBjQuRejd3+B6devX4fHvt9qv+7S3c9/dyjq3wCVZ2m/2ryzhWOvLdnSkd/o7Ld2NmfwDgMzoK7nXxc6Y2XddulqVTVyp6Gp3tInyNuozswFXZkHEnNBR5gH6CmtfXfIW1s49kHzQNK5b/JvTl1tTYYV8JrQGWv7bJ83ujh26OD+6Tdg2358v8s8sO0wF9AT2lOTxe9zfGu9J9ji7bSvL+Q1obOaszZtXfiEs39dnwwpg8f323w2tO0wD9BT3uhbnS1tytcTnw0NHzY4tTts26+VS+uqt/j52fvpk3UZ0YOvee3t7RsvozFy5Mgu306v3brwS1/6Uv7hH/4hY8aMyYMPPpiJEycmSR5//PGcffbZaWxszLp16/Knf/qnueGGG94zft68eVm2bFkWLFiQ6667Lm+88UYeeeSRjB07tkfyr3tndW4dP61H7mtrOOvlW9K3/xaWWHSj2bNnd+r8uXPn5pxzzslNN92UhoaGDo3Zf//9uxKtJOX2/CfF/Rug8lzxD0/n6//QtWXMTT/7dEbvNCALW97OmKO7tu3U3f/n6Jz0sZ6ZC7rqyeffyAFndX67iQOn1uexW07uhkTdqzNzQVfmgcRc0BHmAXrKE8+9ngOn/aTL40udC/749xry7T8/pMv33xPWrtuQMUf/MK+/1fEvriVJ35rqLPzZp7PjsH7dlKx7mAe2HeYCekJ7e3uGffSWLF3R+euOJKXPA7uOGpRf3/97XbrvnnTW1/4j/3rfy50eN/Paj+XTx4/vhkTdx2dD2w7zAD3lD/7s55l5/6+7NHZrfDY0967Tsvuug7s0tqfc/JP5OefyX3Z63Lkn75YfXH1YNyTqXtVFB+guX/va1zJkyJA0NTVlypQp2XPPPTNhwoQcdNBBmTRpUg499NAkyV577bXZ8bvvvnsOOOCAfPrTn85DDz2UFStW5G//9m978iHQDSZMmJAHHnggEyZMKDoK0EUfntS11bVby76ThxV6/x2x/x7Ds+/kzv89fenMSd2QZttiHoDyN3Xi0PTpU1XY/Xfl9bWnbde3Tz5/2u6dHnfGMbuWXcnVWeYBKH9VVVWFvicoh/cDSdd+t99xaF0+ddS4rR9mG2MugPJX5O/kgwb0zW67dHGf8B50xjG7Ztjgzq/sLdfPhnpt0TV27NjMmjUrJ5xwQurq6tLY2Jj6+vrMmDEjd911V+bOnZtky0XXbxs8eHAmTJiQl156qbtj081qamoyZMiQ1NT02l07odcr8o3liOH9svOOXdu/uSdVVVXl0j/cs1Njxu08MGccs2s3Jdp2mAeg/PWrq8mU8UMKu/9yKLqS5ItnTMr2Azt+fZaaPlW56Owp3Zho22AegN6hyPcE5TIPHLz3jjlkn85d4Onic/bIdn37dFOibYe5AMpfka/F+zQMS3V1cV+866i62pr86bTO/X5/+H4jsv8e5THP/a5eW3QlyZQpU3Lvvfdm5cqVWbFiRR577LF89rOfzdKlS9Pc3JyamppMnjz5A2/n9ddfz7x58zJ+fHkt3ea9Fi5cmEsuuSQLFy4sOgrQRTvvOCB77lbMB5zHHjy6kPvtit879kP5iy/s3aFz64fU5b5vH5u62t7/Rs88AL3DcYcUsx/+zjv2zx4TiivZOmP0iAG58/qjUlf7wR9YVldX5Z+v/Gj2m1LfA8mKZR6A3uG4Q4r7vfzYg7fta7K8q6qqKj+6/uPZbZftO3T+2SdOyFc/s/nr2Pc25gIofwfsWZ/Bg7Yr5L6LnIM6a/rn9s7vH/+hDp27+7gdcvs3jkxV1bZf4m1Ory66tuT5559PkjQ0NKS2dtPle9OmTcvXv/71/PjHP85//Md/5Lvf/W4+9rGPpaamJhdddFERcdmKVq1alVmzZmXVqlVFRwFKUNQy6nJbvn3ll/fNty8/OMOHbHmP9I9+eKc8futJmfShwT0XrEDmAegdvnBGQ4p4//WF0xtSU1M+b6E+fuDO+fmMEzJl/OAtnjNu54H59xs+nrNP2q3nghXIPAC9w8f2H5mGXXt+26gDp9Znn4K3Uu+MnYb1y6M3nZSTPzZ2i/PmwP41+csv7JMfXH1YWaxQ2BrMBVD++tXV5A8/2fO/v/atqc5nT53Y4/fbVdXVVbn5msNz+ef3yoB+m/9yc1VVcurHd8l/3nRi6oeW7zbmvf+r25vxbtG1uW0LDzzwwNx000355je/mdWrV2fMmDE54ogjMn369Oyyyy49HRWAzTjrE+Nz6Q2zs/LtdT12n/tOHl6Wy7f/+MxJOe/UibnjZwty+08X5I2lq1Pbt08mfWhwvnBGQ6ZOHFp0RIBO+9Do7XP8oaNz36ye+yZ2TU1VPvep8nlT+64D99oxz9/5qTzydEu+9+//k1cWr8yGDe3Zecf+OeekCTnukNHp06d8yjuA5Derlb505qRc8DeP9+j9ltsX35Jk+JC63PV/js6vF67IP90+N0+9+GbeWb0+gwdtlxMPG5OzT5qQQQOKWRUBUIovnjEpN9z83z16n+V4Tds+fapz9Z/sl0v/cGpuvuel3PvLpixbuTYD+tVkvynD84XTGzJu1KCiY5ZM0fU7zj///Jx//vk9HakkfWr75vB/vCg77DY6G1rXpPWNZXns0u9k1cI3Njlv4Oj6HPrN8zN0j3FZseC13HPcn208NrhhbA7868+l3/Ad0rZufV5/6n/yxOXfS9va9T39cOiCj1x1XsYeu18GjtkxP/7YRVk2rylJsucFn8qEMw7P9h8amYfOvTYLH3xqs+MHjq7Ppx77uyyb++rGn/38c9/IysaWHskPnTVowHb588/vnT+7cXaP3edfX7hf2S7frt2uT876xISc9QkXW+7NBu06Ih/95p+kduigrF3xdh7507/P8v/ZtAQYcdCUHHXr9Kx4efHGn9170uXZsHptkmTAqOE58JrPZfsPjUyS/Or7/zdz//n+nnsQ0AlXfXnfPPDoomzY0N4j93fBH0wpi+s0bk5VVVU+uu+IfHTfEUVHoRvt/LG98uHL/iBVVdWp7tsnL3z7rrx8+y82OWfEIXtk3+lnpe/AurS3tefVB2bnmb+ZmSQZOGbHfGzGJamurk5Vn+osf2lxHv3qP2bt8reLeDjwgT576u755q3/nZebVvbI/e21+9CceVzHtn/aFn1o9Pa59qKPFB2DbrTd9v1z3I+u2PjfffrVZtAuO+WHe342a5f91uq1qqrs9xdnZ9QRe6e6pk9anpybx7/23bSt+81ngN4TUC4mjtshnz11Yr737//TI/dXV9snf9nBS0Rsi7YfuF2+/OnJ+fKnP/hSTuWooouuqVN7z97D827+WRY9/EySpOEPj8tB3/hifvbpqzY5Z+2q1jx97cxsN6h/9v7qmZsc27BmbZ6Y/r0s/VVjqqqrc9i3L8yUL56c5//PnT32GOi6xnsfywvf/nFOuOvqTX7ePOv5vHLXf+aQ67/0gbexdsXbufvor3ZXRNjqLj5nj/zowVfy5AtvfPDJJfqj03fP0QeVx178VK6D//YL+Z9bfpaX/u0/sssnDswh//uPc99Jl7/nvGX/s3CTL7v8tiP++at5/ls/TuM9jyVJ+tUP7s7IUJIPTx6eyz67V67+zrPdfl8Td9khV5+/b7ffD5TisL//0zxw+tez9FeNGTi6PqfO+mYa73si699evfGctcvfzi/++IasevX19Kntm2P+7S+z66mHZsG/P5J3Wt7K/af8xcYvP3zkys9kr4tOz+yv/0tRDwneV/9+Nfn+lYfl8PPuTXs3f+ehpqYqP7jqsGzX94OvewhFWbvinU0+15nyxZMz4qDJm5ZcSXb7g49nyORd8pNjLk3buvU5+H//cSZ97oT89z/cncR7AsrL//7KAXng0UVZ2NL9X8y5+vx9s/uug7v9fuiaityj4tFHH017e3uOPfbYoqNsFRvWrNtYciXJG0/Pz6CxO77nvLXLVuX1J+dm/Ttr3nNs5YLXsvRXjUmS9ra2LHn25c3eRrmrr6/PhRdemPr63nWh7ZbHf5V3mt96z8+XPDPfqix6rZqa6vzg6sO2uMfw5ry25J0sbHk7ry15p8Njxo8ZlOsu9s3H3qK3zgN1w7bPsD0/lJd/9MskSeO9j2fQ2B0zcHTHH+fIj+6ZDavXbnxDmyStbyzb2lFhq/qLL+ydD08a1qkxnZ0L+tZU5wdXfzT96iryO4K9Tm+dB9613Q79kyR9B/XP6qUr37NDx1svLMiqV19P8pv3kW+98EoG7bJTkqRt7fqNJVdVdXVq+v9m1Rdsyz6674hccs6enRrTlfcEX//ih7N3Q+fmG7ZdvX0ueNeETx+R+TMfes/Ph07eJc2/fG7jCq5FDz+d8acfnsR7AsrPDoO2yz9f+dFOXWOwK/PAYfuOyJ9Om9KViPQQ79Z6oUmfPSFNP9v8FnUdUdOvNhPP+njmXH3LVky1bRg2bFjOOuusomNsk/oO7JcT7/+bVPWpzqv3P5nnvnln2tvaio4F72vShwbn7v9zdE748k+zZu2GDzx//9+/u1O3P7K+f376T8dl+4H2rO8teus8MGDU8LzT8lbaN/z/r9urFi3JgNHD37OV8Q7jd85JP/3btG9oy/wf/jzz/uWBJMngiWOy+s0VOfwfLsr243fOqoWvZ/bX/2XjB6KwLdqub5/c9+1j89HP3JP5jSs6NKYzc0F1dVX+9W8+loP22qmrEdnG9NZ5IEl+8YXrc8T3vpr176zJdjsMyM8/e93GDzE3p1/94Iw78cD8bNo1G39W3bcmJ9731xkwuj5LX2zMQ+f+TU9Eh5L8zZ/ul4Utb+eH//fXHTq/s+8JvnBGQ6Z//r2XvqB89ea54F31+05M3ZCBm/18cMmzL2XitKMz918eSNva9dn1lEMzcMxvSj/vCShHRx80Kt/9X4fms/9rVofO7+w8MHXi0Pz4m0e5ru02zrPTy+x5waey/a4j8vTf/GuXxlfV9Mnh/3RxFv3Hf6XpgZ679k1PWbFiRR588MGsWNGxD0IqxTuvL83tH/5C7jn+a3ng967MTgdMypQvnlR0LOiQIw/YOfd/+5gMGtB3q97urqMGZdYPPpEPjd5+q94uxerN88DvbtmzuWvKvfn8r/NvH/5CfnLMpXn4vL/N7ucck3EnHZQkqa7pk50/OjX/dcPt+ckxX82ih5/J4f94UU9Eh5LsNKxffvHPn8ieuw3Zqre7Xd/q3PG/j8zpx+y6VW+XYvXWeaCqT3X2/JNT8/Bnrs0d+/9xHjjjihz6zfOz3eCBmz2/78B++fhNX8vz374rbz2/YOPP29atz91HfzW3Tf1clr+8KLufc0xPPQTosj59qnPzNYfnDz+521a/7QvPmpJvX35w2V6rl83rrXPBb9vt94/MS7f/YpMvwr3r5dt/kcX/8WyO//ercuwd/yvL5jWlbd1vvjjqPQHl6rxTJ+Zfrj4sffps3dfrA/asz8+/d0KGbF+7VW+XrU/R1YtM+eLJ2eWEA/LgWX+VDa1rOz2+qqZPPvadi9P6+tI8+Rf/3A0Ji7d48eJMnz49ixcvLjrKNqVt7fqsfvM3v+CtXbYq83/4cHY6YFLBqaDjjvjIznn29k/mY/uP3Cq395lTdsvTt52S8WOUXL1Nb50H3l60JANGDk3Vb33DbMDOw/L2wiWbnLduVWvWrfzN9gzvNL+VBT9+ZOPr/aqFb+TNFxZk2f8sTJK8/KNfZtjUD6Wq2q+LbPtG1vfP47ecnAvPmpKt8VnkflOG56kfnpJTPz6u9Btjm9Jb54Ghe+ya/iOG5PXZ85Ikb/7Xy2ltWZqhk3d5z7k1A+py9L/+eV796Zy8+E/3bPb22tatz0s//HnGn35Yt+aGraWmpjrfu+Kj+cFVh2WHQaXvxjB8SF3+7RtH5sY/O7BT22FRHnrrXPCumn61GXfywXnphw9v8Zz/uuGO/OSYr+b+U/4iy19alOXzf/MewHsCytk5J++Wx285KVPGDy75tvr0qcr0z+2VX3z/Exm6g5KrHHiV6iUmf+HE7HrqIfnpmVdm7YqO7y/6rqo+1Tn8Hy/KmqWr8uhX/rEbErItqxu2fapqfnNR3ertarLLCQfmzRcWfMAo2LZ8aPT2eei7x+fvpx+UkfX9u3Qbk8cPzr1/f0y+f9VhGezbOpSR1W+uyFsvvJLxp/3mA8ldPnFgVjW98Z5tC/vtODjvtgA1A+oy+qh98+YLryRJFj38TPqPGJr+I4YmSUYdsU+WzW2yjS1lo3+/mtz4ZwfmF//8iew7eXiXbmPY4Nr89YX75bGbT8oeuw3dygmh+7y9aEn6jxyW7cfvnCQZNG5EBu2yU1b8unmT82r6/6bkWvQfz+a5G+7Y5NiAUcNT0+///f5TVZVxJx+ct371ao/kh62hqqoq556yW/77zk/l947dtUsFVU1NVc45aUJe/PdP5QwreilT4046KEt/1ZjlL22+yOtT2zfbbf+b98y1Qwdlz/M/mef//sdJvCeg/O03pT5P3fbJ/OUX9unyFx8O3nvHPH7LSfmrC/ZL7XZ9tnJCuotrdPUC/UcOzUe+/pmseOW1HHfH15MkG9auz72fuCwHf+OLafrpnDT9dE6qt6vJaY/9ffrU1qTvoP4546l/yss/+kWevuZfs+sph2TcJw7MW//9Sk7+2XVJkpbZ8/LE9BkFPjI66oBrPpexx+6ffjsOzrH/9pdZ9/bq3Hnwn2TPPzk1DZ85LnXDts+h3/xyNqxel7uP+WrWvLkie3/1zLS2LM28m36aHQ+YlH2+embaN7SlqqZPXnvk+Tz3zR8V/bCg06qrq/KlT0/O509ryF0/b8w//Nuv8sgzLVm7bsu/lA/s3zfHHzo6X/70pBy27wjbklC2Hr30n3LojV/Onhd8KutWteaRC76VJJv8LrDLJw7M7ucem/b1G1JV0yeNP3ls4zc917euyeOXfTdH3XxZUlWVtSvezi+/dGOBjwi65qP7jsjsmSdn9gtL8u3bfpUf/7wxy1duebeDmpqqfGSP+nzxjIacccyuqav1Fonys3rJ8jx26T/liO9ekva29qSqKo9fNiPvvPbWJvPA5M+fkPp9JqRv/9rscvxHkiSv3PNYnvvmnRm8+5jse/m0JElVdVXefH5Bnvzz7xX5sKBLRu00ILddd2SaXluV7/5oXm7+yUt5ZfGq9x0zYez2OffkCfncp3bPiOFd+9IcbCt2+4OPZ/6/brqa67fngr6D+uf4f78ybRvaUt2nOi9+994s/H/X8vKegN6gdrs+ueLLH86lf7hnZt7/63znR3Pz1Itvpq2tfYtjhu5Qm9OOGpcvnTkpezcM68G0bC1V7e2/e0UHtgXr3lmdW8dPKzpGh5318i3p27+ux+939uzOXUds7ty5Oeecc3LTTTeloaGhQ2P233//rkQrSbk9/0lx/wbgg6xdtyEvzF+ap3/1Zt5Yujpr121I7XZ9MmbEgOw7eXgm7rKD7UjKXGfmgq7MA4m5oCPMA2yr2tvb8+uFK/PUi0vy64Urs2bthvStqc7QHWrz4UnDM3XiEOVWmTMPbDvMBWyr3nirNU+9+Gb+++Wlebt1faqrkwH9+maPCUPy4UnDMmywf7flzGdD2w7zANuqd1rX57/+580886s3s2zl2qxb35a62j6ZMGb77Dt5eHbZeaAvPpc57+ioKLW1tdl9991TW2tLMqgU2/Xtkw9PHp4Pd3EbK3oX8wBUnqqqqowfs73rLpLEPACVqn5ovxx36Ogcd+jooqOwDTAXQOXp368mB+21Uw7aa6eio9BNFF1UlF133TU333xz0TEAKIh5AKCymQcAMBcA9D7VRQcAAAAAAACArlB0UVHmzZuXQw45JPPmzSs6CgAFMA8AVDbzAADmAoDeR9FFRWlvb8+6devS3t5edBQACmAeAKhs5gEAzAUAvY9rdG2javrV5qyXbyk6RofV9HMBz62p3J7/xL8BgK2t3OYC8wDA1lVu80BiLgDYmswDAB2n6NpGVVVVpW//uqJjUBDPPwDmAoDKZh4AqGzmAYCOs3UhAAAAAAAAZcmKLirKuHHjMnPmzIwaNaroKAAUwDwAUNnMAwCYCwB6H0UXFaWuri7jx48vOgYABTEPAFQ28wAA5gKA3sfWhVSU5ubmXH311Wlubi46CgAFMA8AVDbzAADmAoDeR9FFRVm+fHnuvvvuLF++vOgoABTAPABQ2cwDAJgLAHofRRcAAAAAAABlSdEFAAAAAABAWVJ0AQAAAAAAUJYUXVSUoUOH5txzz83QoUOLjgJAAcwDAJXNPACAuQCg96lqb29vLzoEAAAAAAAAdJYVXQAAAAAAAJQlRRcAAAAAAABlSdEFAAAAAABAWVJ0AQAAAAAAUJYUXQAAAAAAAJQlRRcAAAAAAABlSdEFAAAAAABAWVJ0AQAAAAAAUJYUXQAAAAAAAJQlRRcAAAAAAABlSdEFAAAAAABAWVJ0AQAAAAAAUJYUXQAAAAAAAJQlRRcAAAAAAABlSdEFAAAAAABAWVJ0AQAAAAAAUJYUXQAAAAAAAJQlRRcAAAAAAABlSdEFAAAAAABAWVJ0AQAAAAAAUJYUXQAAAAAAAJQlRRcAAAAAAABlSdEFAAAAAABAWVJ0AQAAAAAAUJYUXQAAAAAAAJQlRRcAAAAAAABlSdEFAAAAAABAWVJ0AQAAAAAAUJYUXQAAAAAAAJQlRRcAAAAAAABlSdEFAAAAAABAWVJ0AQAAAAAAUJYUXQAAAAAAAJQlRRcAAAAAAABlSdEFAAAAAABAWVJ0AQAAAAAAUJYUXQAAAAAAAJQlRRcAAAAAAABlSdEFAAAAAABAWVJ0AQAAAAAAUJYUXQAAAAAAAJQlRRcAAAAAAABlSdEFAAAAAABAWVJ0AQAAAAAAUJYUXQAAAAAAAJQlRRcAAAAAAABlSdEFAAAAAABAWVJ0AQAAAAAAUJYUXQAAAAAAAJQlRRcAAAAAAABlSdEFAAAAAABAWVJ0AQAAAAAAUJYUXQAAAAAAAJQlRRcAAAAAAABlSdEFAAAAAABAWVJ0AQAAAAAAUJYUXQAAAAAAAJQlRRcAAAAAAABlSdEFAAAAAABAWVJ0AQAAAAAAUJYUXQAAAAAAAJQlRRcAAAAAAABlSdEFAAAAAABAWaopOgClmT17dqfOX7JkSe6888586lOfyvDhwzs0Zv/99+9KNAAAAAAAgG5lRVeFWbJkSWbMmJElS5YUHQUAAAAAAKAkii4AAAAAAADKkqILAAAAAACAsqToAgAAAAAAoCwpuirMoEGDctxxx2XQoEFFRwEAAAAAAChJVXt7e3vRIei62bNnd/t97L///t1+HwAAAAAAAJ1lRVeFWbNmTZqamrJmzZqiowAAAAAAAJRE0VVhFixYkNNOOy0LFiwoOgoAAAAAAEBJFF0AAAAAAACUJUUXAAAAAAAAZUnRBQAAAAAAQFlSdAEAAAAAAFCWaooOQM9qaGjIk08+WXQMAAAAAACAklnRBQAAAAAAQFlSdFWYxsbGnHfeeWlsbCw6CgAAAAAAQEkUXRWmtbU1L7zwQlpbW4uOAgAAAAAAUBJFFwAAAAAAAGVJ0QUAAAAAAEBZUnQBAAAAAABQlhRdFWbkyJG54oorMnLkyKKjAAAAAAAAlKSm6AD0rB122CHHH3980TEAAAAAAABKZkVXhVm6dGluv/32LF26tOgoAAAAAAAAJVF0VZiWlpZcd911aWlpKToKAAAAAABASSqi6FqyZEm+8pWvZPz48amrq8uYMWNyySWXpLW1NdOmTUtVVVVmzJhRdEwAAAAAAAA6oddfo+u5557LMccck5aWlgwYMCCTJ0/OokWLcv311+eVV17Jq6++miTZa6+9Ck4KAAAAAABAZ/TqFV1LlizJiSeemJaWllx66aV57bXX8vTTT6elpSVXXXVV7rzzzjz77LOprq7OHnvsUXRcAAAAAAAAOqFXF10XXHBBmpqacvHFF+faa6/NwIEDNx67/PLL09DQkPXr12fixInp169fgUl7Tv/+/XPAAQekf//+RUcBAAAAAAAoSa8tul588cXcdttt2XHHHXPVVVe953hVVVX23XffJO/dtnDBggU5+eSTM2jQoAwZMiTnnHNO3nzzzR7J3d3Gjh2bb33rWxk7dmzRUQAAAAAAAErSa4uuW2+9NW1tbZk2bdoWVy/V1tYm2bToWrlyZY444ogsXLgwM2fOzHe+853MmjUrJ554Ytra2noke3fasGFDVq1alQ0bNhQdBQAAAAAAoCQ1RQfoLg899FCS5IgjjtjiOQsXLkyyadH1ne98J4sWLcovf/nLjaueRo0alUMOOSR33313PvnJT3Y4Q3t7e955550kv9kysKqqqrMPY6ubP39+zjnnnNx0001paGgoOg4AAAAAAECX9dqiq7GxMUkybty4zR5fs2ZNnnjiiSSbFl333HNPDj300E229jv44IMzbty4/OQnP+lU0fXOO+9svC7YyJEjU1299RfQnX766Z06//XXX0+S3H///Xnqqac6NObUU0/tdC4AAAAAAICOGDFiRObMmdOlsb226Hr77beTJKtXr97s8VtuuSXLly/PsGHDMmrUqI0/f/HFF3PGGWe85/zJkyfnxRdf7HKe5ubmLo99P+8+zo5qbW3d+GdHxy5atKjTuQAAAAAAALpbry26RowYkZUrV2bOnDnZb7/9NjnW1NSUyy67LEkyderUTY4tXbo0gwcPfs/tDRkyJC+99FKX83TXiq4BAwZ06vx3y61+/fp1eOxvF4EAAAAAAABb04gRI7o8ttcWXUcddVTmz5+fa665JkceeWQmTpyYJHn88cdz9tlnZ9myZUk23bbwXZu7llZ7e3tJeebPn9/pUqojZs+e3anz586dm5kzZ+b444/v8DW6brzxxi4kAwAAAAAA6F5bf4nRNuJrX/tahgwZkqampkyZMiV77rlnJkyYkIMOOiiTJk3KoYcemuS9RdeQIUOydOnS99zesmXLMnTo0B7J3p0mTJiQBx54IBMmTCg6CgAAAAAAQEl6bdE1duzYzJo1KyeccELq6urS2NiY+vr6zJgxI3fddVfmzp2b5L1F16RJkzZ7La4XX3wxkyZN6pHs3ammpiZDhgxJTU2vXcwHAAAAAABUiF5bdCXJlClTcu+992blypVZsWJFHnvssXz2s5/N0qVL09zcnJqamkyePHmTMSeeeGIeeeSRNDU1bfzZ448/nldeeSUnnXRSTz+ErW7hwoW55JJLsnDhwqKjAAAAAAAAlKRXF11b8vzzzydJGhoaUltbu8mxP/qjP8rIkSPzyU9+Mvfcc0/uuOOO/MEf/EE+8pGP5JRTTiki7la1atWqzJo1K6tWrSo6CgAAAAAAQEkquuj63W0Lk2T77bfPww8/nJEjR+bTn/50Pve5z+Xggw/OPffck+rqivzrAgAAAAAA2CZV5IWa3q/oSpLx48fnnnvu6clIAAAAAAAAdFJFLlF6t+iaOnVqwUkAAAAAAADoqopc0fXoo48WHaEw9fX1ufDCC1NfX190FAAAAAAAgJJUZNFVyYYNG5azzjqr6BgAAAAAAAAlq8itCyvZihUr8uCDD2bFihVFRwEAAAAAACiJoqvCLF68ONOnT8/ixYuLjgIAAAAAAFASRRcAAAAAAABlSdEFAAAAAABAWVJ0AQAAAAAAUJYUXRWmtrY2u+++e2pra4uOAgAAAAAAUJKaogPQs3bdddfcfPPNRccAAAAAAAAomRVdAAAAAAAAlCVFV4WZN29eDjnkkMybN6/oKAAAAAAAACVRdFWY9vb2rFu3Lu3t7UVHAQAAAAAAKImiCwAAAAAAgLKk6AIAAAAAAKAsKboAAAAAAAAoSzVFB6BnjRs3LjNnzsyoUaOKjgIAAAAAAFASRVeFqaury/jx44uOAQAAAAAAUDJbF1aY5ubmXH311Wlubi46CgAAAAAAQEkUXRVm+fLlufvuu7N8+fKiowAAAAAAAJRE0QUAAAAAAEBZUnQBAAAAAABQlhRdAAAAAAAAlCVFV4Wprq7OPvvsk+pqTz0AAAAAAFDetB0Vpq2tLc8880za2tqKjgIAAAAAAFASRRcAAAAAAABlSdEFAAAAAABAWVJ0AQAAAAAAUJYUXRVm0KBBOe644zJo0KCiowAAAAAAAJSkpugA9KxRo0blyiuvLDoGAAAAAABAyazoqjBr1qxJU1NT1qxZU3QUAAAAAACAkii6KsyCBQty2mmnZcGCBUVHAQAAAAAAKImiCwAAAAAAgLKk6AIAAAAAAKAsKboAAAAAAAAoS4ouAAAAAAAAylJN0QHoWQ0NDXnyySeLjgEAAAAAAFAyK7oAAAAAAAAoS4quCtPY2JjzzjsvjY2NRUcBAAAAAAAoiaKrwrS2tuaFF15Ia2tr0VEAAAAAAABKougCAAAAAACgLCm6AAAAAAAAKEuKLgAAAAAAAMqSoqvCjBw5MldccUVGjhxZdBQAAAAAAICS1BQdgJ61ww475Pjjjy86BgAAAAAAQMms6KowS5cuze23356lS5cWHQUAAAAAAKAkiq4K09LSkuuuuy4tLS1FRwEAAAAAAChJRRRdS5YsyVe+8pWMHz8+dXV1GTNmTC655JK0trZm2rRpqaqqyowZM4qOCQAAAAAAQCf0+mt0PffccznmmGPS0tKSAQMGZPLkyVm0aFGuv/76vPLKK3n11VeTJHvttVfBSQEAAAAAAOiMXr2ia8mSJTnxxBPT0tKSSy+9NK+99lqefvrptLS05Kqrrsqdd96ZZ599NtXV1dljjz2KjgsAAAAAAEAn9Oqi64ILLkhTU1MuvvjiXHvttRk4cODGY5dffnkaGhqyfv36TJw4Mf369Sswac/p379/DjjggPTv37/oKAAAAAAAACXptUXXiy++mNtuuy077rhjrrrqqvccr6qqyr777ptk020LFy5cmAsuuCAHH3xw+vfvn6qqqrzyyis9FbvbjR07Nt/61rcyduzYoqMAAAAAAACUpNcWXbfeemva2toybdq0La5eqq2tTbJp0fXSSy/ltttuy+DBg3P44Yf3SNaetGHDhqxatSobNmwoOgoAAAAAAEBJem3R9dBDDyVJjjjiiC2es3DhwiSbFl2HHXZYWlpact999+XMM8/s3pAFmD9/fo488sjMnz+/6CgAAAAAAAAl6bVFV2NjY5Jk3Lhxmz2+Zs2aPPHEE0k2Lbqqq3vtXwkAAAAAAECvUlN0gO7y9ttvJ0lWr1692eO33HJLli9fnmHDhmXUqFHdnme33XbrlhLt9NNP79T5r7/+epLk/vvvz1NPPdWhMaeeemqncwEAAAAAAHTEiBEjMmfOnC6N7bVF14gRI7Jy5crMmTMn++233ybHmpqactlllyVJpk6d2iN5mpubu+V23y30Oqq1tXXjnx0du2jRok7nAgAAAAAA6G69tug66qijMn/+/FxzzTU58sgjM3HixCTJ448/nrPPPjvLli1Lsum2hd1p5MiR3bKia8CAAZ06/91yq1+/fh0e2xMr3gAAAAAAgMo0YsSILo/ttUXX1772tfzwhz9MU1NTpkyZkoaGhrS2tubll1/OSSedlDFjxuTnP/95jxVd8+fP73Qp1RGzZ8/u1Pnr16/PZz7zmQwaNCg1NR17+m+88cYuJAMAAAAAAOheW3+J0TZi7NixmTVrVk444YTU1dWlsbEx9fX1mTFjRu66667MnTs3Sc+t6NpW1NTUZMiQIR0uuQAAAAAAALZVvbrtmDJlSu699973/Pytt95Kc3NzampqMnny5AKSFWfhwoW54YYbctFFF2X06NFFxwEAAAAAAOiyXl10bcnzzz+fJGloaEhtbe17jt9xxx1Jkjlz5iRJ7r///tTX12fcuHHZb7/9ei5oN1i1alVmzZqVz3/+80VHAQAAAAAAKElFF11b2rbwjDPO2OS/v/SlLyVJzj333PzgBz/o1mwAAAAAAAB0jKJrM9rb23syDgAAAAAAAF1QXXSAIrxbdE2dOrXgJAAAAAAAAHRVRa7oevTRR4uOUJj6+vpceOGFqa+vLzoKAAAAAABASSqy6Kpkw4YNy1lnnVV0DAAAAAAAgJJV5NaFlWzFihV58MEHs2LFiqKjAAAAAAAAlETRVWEWL16c6dOnZ/HixUVHAQAAAAAAKImiCwAAAAAAgLKk6AIAAAAAAKAsKboAAAAAAAAoS4quClNbW5vdd989tbW1RUcBAAAAAAAoSU3RAehZu+66a26++eaiYwAAAAAAAJTMii4AAAAAAADKkqKrwsybNy+HHHJI5s2bV3QUAAAAAACAkii6Kkx7e3vWrVuX9vb2oqMAAAAAAACURNEFAAAAAABAWVJ0AQAAAAAAUJYUXQAAAAAAAJSlmqID0LPGjRuXmTNnZtSoUUVHAQAAAAAAKImiq8LU1dVl/PjxRccAAAAAAAAoma0LK0xzc3OuvvrqNDc3Fx0FAAAAAACgJIquCrN8+fLcfffdWb58edFRAAAAAAAASqLoAgAAAAAAoCwpugAAAAAAAChLii4AAAAAAADKkqKrwgwdOjTnnntuhg4dWnQUAAAAAACAkii6Kkx1dXX69u2b6mpPPQAAAAAAUN60HRVmyZIlmTFjRpYsWVJ0FAAAAAAAgJIougAAAAAAAChLii4AAAAAAADKkqILAAAAAACAsqToqjCDBg3Kcccdl0GDBhUdBQAAAAAAoCQ1RQegZ40aNSpXXnll0TEAAAAAAABKZkVXhVmzZk2ampqyZs2aoqMAAAAAAACURNFVYRYsWJDTTjstCxYsKDoKAAAAAABASRRdAAAAAAAAlCVFFwAAAAAAAGVJ0QUAAAAAAEBZUnQBAAAAAABQlmqKDkDPamhoyJNPPll0DAAAAAAAgJJZ0QUAAAAAAEBZUnRVmMbGxpx33nlpbGwsOgoAAAAAAEBJFF0VprW1NS+88EJaW1uLjgIAAAAAAFASRRcAAAAAAABlSdEFAAAAAABAWVJ0AQAAAAAAUJYUXRVm5MiRueKKKzJy5MiiowAAAAAAAJSkpugA9Kwddtghxx9/fNExAAAAAAAASmZFV4VZunRpbr/99ixdurToKAAAAAAAACVRdFWYlpaWXHfddWlpaSk6CgAAAAAAQEkqouhasmRJvvKVr2T8+PGpq6vLmDFjcskll6S1tTXTpk1LVVVVZsyYUXRMAAAAAAAAOqHXX6PrueeeyzHHHJOWlpYMGDAgkydPzqJFi3L99dfnlVdeyauvvpok2WuvvQpOCgAAAAAAQGf06hVdS5YsyYknnpiWlpZceumlee211/L000+npaUlV111Ve688848++yzqa6uzh577FF0XAAAAAAAADqhVxddF1xwQZqamnLxxRfn2muvzcCBAzceu/zyy9PQ0JD169dn4sSJ6devX4FJe07//v1zwAEHpH///kVHAQAAAAAAKEmvLbpefPHF3Hbbbdlxxx1z1VVXved4VVVV9t133ySbblt4xx135LTTTssuu+yS/v37p6GhIZdffnlWrVrVY9m709ixY/Otb30rY8eOLToKAAAAAABASXpt0XXrrbemra0t06ZN2+Lqpdra2iSbFl3f+MY30qdPn1xzzTW5//7788UvfjHf/va3c+yxx6atra1HsnenDRs2ZNWqVdmwYUPRUQAAAAAAAEpSU3SA7vLQQw8lSY444ogtnrNw4cIkmxZdP/nJT1JfX7/xvw8//PAMHz48Z599dh555JEcdthh3ZS4Z8yfPz/nnHNObrrppjQ0NBQdBwAAAAAAoMt6bdHV2NiYJBk3btxmj69ZsyZPPPFEkk2Lrt8uud717haHixYt6nKe3XbbLdXVW38B3emnn96p819//fUkyf3335+nnnqqQ2NOPfXUTucCAAAAAADoiBEjRmTOnDldGttri6633347SbJ69erNHr/llluyfPnyDBs2LKNGjXrf2/rlL3+ZJCWtgGpubu7y2Pfz7uPsqNbW1o1/dnRsKQUfAAAAAABAd+m1RdeIESOycuXKzJkzJ/vtt98mx5qamnLZZZclSaZOnfq+t7No0aL85V/+ZY455pjss88+Xc4zcuTIblnRNWDAgE6d/2651a9fvw6P/aAiEAAAAAAAoKtGjBjR5bG9tug66qijMn/+/FxzzTU58sgjM3HixCTJ448/nrPPPjvLli1Lsum2hb9r1apVOeWUU9K3b998//vfLynP/PnzO11KdcTs2bM7df7cuXMzc+bMHH/88R1eoXbjjTd2IRkAAAAAAED32vpLjLYRX/va1zJkyJA0NTVlypQp2XPPPTNhwoQcdNBBmTRpUg499NAkWy66Vq9enVNOOSULFizIT3/60+y88849Gb/bTJgwIQ888EAmTJhQdBQAAAAAAICS9Nqia+zYsZk1a1ZOOOGE1NXVpbGxMfX19ZkxY0buuuuuzJ07N8nmi65169bl937v9/Lkk0/m/vvvz+TJk3s6frepqanJkCFDUlPTaxfzAQAAAAAAFaJXtx1TpkzJvffe+56fv/XWW2lubk5NTc17Sqy2tracffbZ+dnPfpb77rsvH/nIR3oqbo9YuHBhbrjhhlx00UUZPXp00XEAAAAAAAC6rNeu6Ho/zz//fJKkoaEhtbW1mxw7//zzc9ttt+WSSy5Jv3798vjjj2/838KFC4uIu1WtWrUqs2bNyqpVq4qOAgAAAAAAUJKKLro2t23hfffdlyT5q7/6qxx00EGb/G/GjBk9mhMAAAAAAIAt69VbF27J+xVdr7zySg+nAQAAAAAAoCsqekXX1KlTC04CAAAAAABAV1Xkiq5HH3206AiFqa+vz4UXXpj6+vqiowAAAAAAAJSkIouuSjZs2LCcddZZRccAAAAAAAAoWUVuXVjJVqxYkQcffDArVqwoOgoAAAAAAEBJFF0VZvHixZk+fXoWL15cdBQAAAAAAICSKLoAAAAAAAAoS4ouAAAAAAAAypKiCwAAAAAAgLKk6KowtbW12X333VNbW1t0FAAAAAAAgJLUFB2AnrXrrrvm5ptvLjoGAAAAAABAyazoAgAAAAAAoCwpuirMvHnzcsghh2TevHlFRwEAAAAAACiJoqvCtLe3Z926dWlvby86CgAAAAAAQEkUXQAAAAAAAJQlRRcAAAAAAABlSdEFAAAAAABAWaopOgA9a9y4cZk5c2ZGjRpVdBQAAAAAAICSKLoqTF1dXcaPH190DAAAAAAAgJLZurDCNDc35+qrr05zc3PRUQAAAAAAAEqi6Kowy5cvz913353ly5cXHQUAAAAAAKAkii4AAAAAAADKkqILAAAAAACAsqToAgAAAAAAoCwpuipMdXV19tlnn1RXe+oBAAAAAIDypu2oMG1tbXnmmWfS1tZWdBQAAAAAAICSKLoAAAAAAAAoS4ouAAAAAAAAypKiCwAAAAAAgLKk6KowgwYNynHHHZdBgwYVHQUAAAAAAKAkNUUHoGeNGjUqV155ZdExAAAAAAAASmZFV4VZs2ZNmpqasmbNmqKjAAAAAAAAlETRVWEWLFiQ0047LQsWLCg6CgAAAAAAQEkUXQAAAAAAAJQlRRcAAAAAAABlSdEFAAAAAABAWVJ0AQAAAAAAUJZqig5Az2poaMiTTz5ZdAwAAAAAAICSWdEFAAAAAABAWVJ0VZjGxsacd955aWxsLDoKAAAAAABASRRdFaa1tTUvvPBCWltbi44CAAAAAABQEkUXAAAAAAAAZUnRBQAAAAAAQFlSdAEAAAAAAFCWFF0VZuTIkbniiisycuTIoqMAAAAAAACUpKboAPSsHXbYIccff3zRMQAAAAAAAEpmRVeFWbp0aW6//fYsXbq06CgAAAAAAAAlUXRVmJaWllx33XVpaWkpOgoAAAAAAEBJKqLoWrJkSb7yla9k/Pjxqaury5gxY3LJJZektbU106ZNS1VVVWbMmFF0TAAAAAAAADqh11+j67nnnssxxxyTlpaWDBgwIJMnT86iRYty/fXX55VXXsmrr76aJNlrr70KTgoAAAAAAEBn9OoVXUuWLMmJJ56YlpaWXHrppXnttdfy9NNPp6WlJVdddVXuvPPOPPvss6murs4ee+xRdFwAAAAAAAA6oVcXXRdccEGamppy8cUX59prr83AgQM3Hrv88svT0NCQ9evXZ+LEienXr1+BSXtO//79c8ABB6R///5FRwEAAAAAAChJry26Xnzxxdx2223Zcccdc9VVV73neFVVVfbdd98km25bOGvWrBx99NEZOXJkamtrM3r06Jx55pn51a9+1WPZu9PYsWPzrW99K2PHji06CgAAAAAAQEl67TW6br311rS1tWXatGlbXL1UW1ubZNOia+nSpdljjz3yR3/0R9lxxx2zcOHC/PVf/3UOOuigvPDCCxk9enSP5O8uGzZsSGtra/r165c+ffoUHQcAAAAAAKDLem3R9dBDDyVJjjjiiC2es3DhwiSbFl0nn3xyTj755E3O22+//dLQ0JAf/ehHufDCC7shbc+ZP39+zjnnnNx0001paGgoOg4AAAAAAECX9dqiq7GxMUkybty4zR5fs2ZNnnjiiSSbFl2bM3z48CRJTU3X/7p22223VFdv/Z0iTz/99E6d//rrrydJ7r///jz11FMdGnPqqad2OhcAAAAAAEBHjBgxInPmzOnS2F5bdL399ttJktWrV2/2+C233JLly5dn2LBhGTVq1HuOb9iwIW1tbWlqasr06dOz884758wzz+xynubm5i6PfT/vPs6Oam1t3fhnR8cuWrSo07kAAAAAAAC6W68tukaMGJGVK1dmzpw52W+//TY51tTUlMsuuyxJMnXq1M2OP/zww/Of//mfSZIJEybk4Ycf3riyq6P69++fVatW5dBDD83rr7+eqqqqLjyS9zdgwIBOnf9uudWvX78Oj91cEQgAAAAAALA1jBgxostje23RddRRR2X+/Pm55pprcuSRR2bixIlJkscffzxnn312li1blmTL2xZ+73vfy7Jly7JgwYJcd911Ofroo/PII49k7NixHc5QVVWVAQMG5Jlnnin58WzJ7NmzO3X+3LlzM3PmzBx//PEdvkbXjTfe2IVkAAAAAAAA3auqvb29vegQ3eHVV1/N3nvvnaVLl6ampiYNDQ1pbW3Nyy+/nJNOOimrVq3Kz3/+83z/+9/PZz7zmfe9rWXLlmXcuHGZNm1a/u7v/q5nHkAHdbboWr9+fVauXJlBgwZ1+Jpj+++/f1eiAQAAAAAAdKvqogN0l7Fjx2bWrFk54YQTUldXl8bGxtTX12fGjBm56667Mnfu3CRbXtH12wYPHpwJEybkpZde6u7Y3a6mpiZDhgzpcMkFAAAAAACwrerVbceUKVNy7733vufnb731Vpqbm1NTU5PJkyd/4O28/vrrmTdvXg444IDuiNmjFi5cmBtuuCEXXXRRRo8eXXQcAAAAAACALuvVRdeWPP/880mShoaG1NbWbnJs2rRpmTBhQvbee+8MHjw48+fPzw033JCamppcdNFFRcTdqlatWpVZs2bl85//fNFRAAAAAAAASlLRRdfmti088MADc9NNN+Wb3/xmVq9enTFjxuSII47I9OnTs8suu/R0VAAAAAAAALZA0fU7zj///Jx//vk9HQkAAAAAAIBOqi46QBHeLbqmTp1acBIAAAAAAAC6qiJXdD366KNFRyhMfX19LrzwwtTX1xcdBQAAAAAAoCRV7e3t7UWHoOtmz57d7fex//77d/t9AAAAAAAAdFZFbl1YyVasWJEHH3wwK1asKDoKAAAAAABASRRdFWbx4sWZPn16Fi9eXHQUAAAAAACAkii6AAAAAAAAKEuKLgAAAAAAAMqSogsAAAAAAICypOiqMLW1tdl9991TW1tbdBQAAAAAAICSVLW3t7cXHYKumz17drffx/7779/t9wEAAAAAANBZVnQBAAAAAABQlhRdFWbevHk55JBDMm/evKKjAAAAAAAAlETRVWHa29uzbt262LESAAAAAAAod4ouAAAAAAAAypKiCwAAAAAAgLKk6AIAAAAAAKAs1RQdgJ41bty4zJw5M6NGjSo6CgAAAAAAQEkUXRWmrq4u48ePLzoGAAAAAABAyWxdWGGam5tz9dVXp7m5uegoAAAAAAAAJVF0VZjly5fn7rvvzvLly4uOAgAAAAAAUBJFFwAAAAAAAGVJ0QUAAAAAAEBZUnQBAAAAAABQlhRdFWbo0KE599xzM3To0KKjAAAAAAAAlKSqvb29vegQAAAAAAAA0FlWdAEAAAAAAFCWFF0AAAAAAACUJUUXAAAAAAAAZUnRBQAAAAAAQFlSdAEAAAAAAFCWFF0AAAAAAACUJUUXAAAAAAAAZUnRBQAAAAAAQFlSdAEAAAAAAFCWFF0AAAAAAACUJUUXAAAAAAAAZUnRBQAAAAAAQFlSdAEAAAAAAFCWFF0AAAAAAACUJUUXAAAAAAAAZen/A/TN11sma3esAAAAAElFTkSuQmCC"
+      "image/png": "iVBORw0KGgoAAAANSUhEUgAABroAAAJxCAYAAAAdC2LsAAAAOnRFWHRTb2Z0d2FyZQBNYXRwbG90bGliIHZlcnNpb24zLjEwLjAsIGh0dHBzOi8vbWF0cGxvdGxpYi5vcmcvlHJYcgAAAAlwSFlzAAAPYQAAD2EBqD+naQAA9VhJREFUeJzs3Xd4FFXbBvB7N5veGyQEAqGG0Hvv0pvSFbAgiKAggqCCqBQVrK8iCAiKCCK99yaE3iWQBAKEkB7S26Zs+f7gMxJJYHezu7M7c/+u67183Zkz86w72XtnzpkzMq1WqwURERERERERERERERGRlZELXQARERERERERERERERGRIdjRRURERERERERERERERFaJHV1ERERERERERERERERkldjRRURERERERERERERERFaJHV1ERERERERERERERERkldjRRURERERERERERERERFaJHV1ERERERERERERERERkldjRRURERERERERERERERFaJHV1ERERERERERERERERkldjRRURERERERERERERERFaJHV1ERERERERERERERERkldjRRURERERERERERERERFaJHV1ERERERERERERERERkldjRRURERERERERERERERFaJHV1ERERERERERERERERkldjRRURERERERERERERERFaJHV1ERERERERERERERERkldjRRURERERERERERERERFaJHV1ERERERERERERERERkldjRRURERERERERERERERFaJHV1ERERERERERERERERkldjRRURERERERERERERERFaJHV1ERERERERERERERERkldjRRURERERERERERERERFaJHV1ERERERERERERERERkldjRRURERERERERERERERFaJHV1ERERERERERERERERkldjRRURERERERERERERERFZJIXQBRBVx8eJFvdZPTU3Ftm3bMGTIEPj4+OjUplWrVoaURkREZqJPFhiSAwCzgIjIkjEHiIikjdeGiIiId3SRpKSmpmLVqlVITU0VuhQiIhIAc4CISNqYA0RExCwgIhIfdnQRERERERERERERERGRVWJHFxEREREREREREREREVkldnQRERERERERERERERGRVWJHF0mKq6sr+vTpA1dXV6FLISIiATAHiIikjTlARETMAiIi8ZFptVqt0EUQGerixYsm30erVq1Mvg8iIjIcs4CISNqYA0RE0sYcICIi3tFFklJYWIjY2FgUFhYKXQoREQmAOUBEJG3MASIiYhYQEYkPO7pIUqKjozF06FBER0cLXQoREQmAOUBEJG3MASIiYhYQEYmPQugCqGxarRYqpfWMLFE42kMmkwldhmhY2+cP8BggIjI2a8sC5gARkXFZWw4AzAIiImNiDhAR6Y4dXRZKpSzE+lpjhC5DZ6PvroOtk4PQZYiGtX3+AI8BIiJjs7YsYA4QERmXteUAwCwgIjIm5gARke44dSERERERERERERERERFZJXZ0ERERERERERERERERkVXi1IUkKcHBwbhw4YLQZRARkUCYA0RE0sYcICIiZgERkfjwji4iIiIiIiIiIiIiIiKySuzoIkmJiYnBuHHjEBMTI3QpREQkAOYAEZG0MQeIiIhZQEQkPpy6kCRFqVTixo0bUCqVQpdCRGQWMQk5uHQzFZfD0xD1IAvKQjVs5DJ4utmjaT0vtAjxQbP63nBxshW6VLNgDhCR1GRkF/5/DqQiLCoDOXnF0Gq1cHZSoH6QB1o28EHLBr6o7O0odKlmwRwgIqkpLFLj71vpuByeiisRqUjNLESxSgN7WxvUCHBBi/o+aBHijbo13CGTyYQu1yyYBUQkJRqNFrdjsnA5/NE5QUxCHgqL1bBVyOHjYY8WIT5oEeKDxnW9YG9nI3S5BmNHFxERkcgoC1T488A9LNsYgUs3U8td77f//6ejgw1e7FsLb42sj+YhPuYpkoiITEar1eLY+UQs2xiBnX/FQK3WPrPNc22r4K1R9TGgcyAUCk78QURk7W7fz8JPmyKwZmcUMnOKnrl+vRrumDQiGK8MqgMPN3szVEhERKaUkV2INTuj8NOmCETFZJe73qpttwEAnm52eO35upg0oj5qB7qZq0yjYUeXiPi1a4A+2+aVeq04T4nse4m4u+UkIlbvg1atEag6MgceA0TSptVq8fvuO5j+9XmkZRbq3E5ZoMYv22/jl+238VzbKlj5cUcEVXU1YaVkKswBIrocnorxn4biWmS6Xu2OnEvAkXMJqFHFBSs+7oBe7auaqEIyNWYBkbQ9TFdi6qJz+PPAPb3a3bqfhWlfnsfsHy7j00nNMP3lhrCx4cAHa8QcIJI2lUqDr38Lw/wVV6EsUOvcLiO7CN+uvYFv197A6P618P37beHt4WDCSo2LHV0iFL3jNGKPXAJkMjj6eqD28C5oPe9VuNcJwNmZK4Quj8yAxwCR9CSk5OGN+aex92RshbZz5FwCGg3dhi/fbYVJI+tLZvoSsWEOEElPUbEaC1Zcwxer/9bpDq7y3E/IRe83D2LC0Hr45r3WcHW2M2KVZE7MAiLp2XbkPt5ccBoPMwoM3kZ+gQqzvruIbUfv49cFnREc5GG8AsmsmANE0hN+NwOvzj2JizfKn91HF+v33sWRcwlY8XEHDO5W3UjVmRaHZohQ+s1o3NsaintbTuLmT7uwt/9s5MY/RN2XesDe2/puOzQmf39/zJs3D/7+/kKXYlI8Boik5VZ0JtqM3l3hTq5/5ClVeOvzs5i08AzUIhvpxxxgDhCJUb5ShcFTj2DhymsV6uR63M9bb6HLa/uQkiau55dIJQcAZgGR1Hyx6m8MnX60Qp1cjzt3/SHajN6FU1eSjLI9SyKVLGAOEEnLiUuJaDtmd4U7uf6RnKbE8+8cwVe/XjfK9kyNHV0SoFIWIvXKHcjkcrhVryx0OYJyd3dH37594e7uLnQpZsVjgEi87sZmo+vr+xCXnGf0ba/YHIlJC89AqzXORVNLwBxgDhCJTWGRGoPfOYwDp+OMvu2rkWl47o39yMjWfTpcSyfVHACYBURitmj135j9wyWjbzc7txh9Jh3Eub9TjL5tIUk1C5gDROJ16koS+k4+iJy8YqNve9Z3F/H1mjCjb9fY2NElEa41HgVYQUaOwJUIKyMjA5s3b0ZGRobQpZgdjwEi8VEWqND/rUNISjXdaPuft97Ct2tvmGz75sYcYA4Qic3URWdx5FyCybYfFpWBkTOPiWbQg5RzAGAWEInR9qP38eH3xu/k+keeUoUBUw4hKTXfZPswNylnAXOASHwSUvIwaOphvZ7Hpa+Z317AnhMPTLZ9Y2BHlwjZONrD3ssV9t5u8AgORJvPx8O7UU08vBqFnGjx3XKuj+TkZHz11VdITk4WuhST4jFAJA0f/XgZt+5n6dXm4oZBiD08Chc3DNJvP9GZelZnmZgDzAEiMTl0Jg4rt9zSq40hOXD4bAJ+3qrffiyVVHIAYBYQSUFqRgHeXHBarzaG5EBaZqGoZnqQShYwB4jET6vV4o35p5GRXaRXO0Oy4NF+LHemB4XQBZhDamoqFi1ahO3btyM+Ph6+vr4YMWIEFi5ciAkTJmD9+vX4+eefMX78eKFLNYqm04ej6fThpV6L2X8e597/WaCKyNx4DBCJ39m/k/Hd7/rfaeXn44SqlZ31alNQqMZrH4fi9NoBkMlkeu+TzI85QCR+ufnFGP/pKb3bGZIDADDj6wvo27Eqqvm56N2WhMEsIBK/dxafQ0q6fs/kMjQHdhyLwcYD9zCqby2925IwmANE4rd+712DntduSBYkPszHtMXn8NtnXfTenzmIvqPr+vXr6NWrF5KTk+Hs7IyQkBDEx8fj22+/xf379/HgwaNb7po0aSJwpcZze/0RRO88DbnCBh71AtFoyvNw9PGAquDfnl25nQIDD36Je9tPIeyHbSWvd/zfW3Dw9cCR0Z8JUToZiS7HQNeVM6DVanFi4rclr9l5uOD5v77DpflrcW9bqBClG51Wq8XFG6lYtjECe0NjkZVTBCdHBVrU98akEfUxuFt12Nry5layPp///DfMOaDy7N8pOHY+ET3aVjHfTslgzIHS4pLysHJrJNbtuYuEh/mwkctQI8AFr79QF68Orgsvd3uhSyTS29pdUYhNMv7zGcuTm1+MH9aH46sZrc22T6oYZsG/8vKLsWH/PazYEonI6CwUFavh6+mAYT2DMGlEMOoFeQhdIpHe7jzIxh/77pp1nwtXXsPIPjU5+M1KMAf+pdFoceRcPH7aFIkTlxKRm6+Ci5MCXVv5Y/LI+ujRpgqPa7I6Wq0WC1ZcM+s+f99zB/MmN0eNAFez7lcXor66m5qaigEDBiA5ORmzZs1CUlISrly5guTkZCxYsADbtm3DtWvXIJfL0bBhQ6HLNZqc+0lIDA1D/PFruLl8F46O/QI+zWuj3aIJJetoilQInboEjae+AM+Q6gCAwD6tULVnS5yevkyo0slIdDkGzr6/ApXb1EfQ8x1KXmv7+XikXIgUzQ+ZrJwi9J10EG1G78Jvu6KQmlGAYpUGWTlFOHYhEcPfO4Y6AzfjWmSa0KUS6eV+fA72huo/Yqeilm2MMPs+yTDMgUe0Wi0+WXoFNfpuxIIV1xAdn4PCIjXyC1QIv5uJGV9fQMBzG/DrjttCl0qkF61WK8h38i87bkNZoDL7fskwzIJHDpyKQ7Vef2LCvFO4dDMVufnFKCrWID4lH9+vv4ngwVvx+iehKCo23XMtiExhxeZIs+/z5t1MhF7mlHfWgjnwyIPEXLQYtQO93zyIHcdikJFdhGKVBhnZRdh+NAY93ziAlqN2Is6MA4iIjOHY+UTcjtHvcRYVpdVC76nTzUXUHV1Tp05FbGwspk+fjsWLF8PF5d9pNubMmYPg4GCoVCrUrVsXjo6OAlZqWg+vROHe1lDUHNIJvs3rlLyeHhaNG0t3otMPU+Dk74V2X07E+dmroEwW78M4nZyc0KZNGzg5OQldilmVdQwUZuTi9PRlaPPZeDhW9kT1/m3h174Bzr6/QuBqjSM3vxg9JuzHwTPxT10vJiEXXcbtxd+32NlF1mP19ttmvZvrHzv/ikHiQ+t+CDVzQDo5AAAzvj6P+SuuQq0u/w+moFCNcR+HYvkmduSS9ThzLQU372aafb/pWYXYeuS+2fdrTFLNAUCaWbD35AMMmHLomc+t+GX7bYyadRxqtcZMlRFVTHGxBr8INFBnhYVe4NSHVLNAijkQn5yHjq/swbXI9KeudyUiDR1f3WP157skLSu2mH/AAwCs2nYLKpXl/WYSbUdXeHg4Nm7ciEqVKmHBggVPLJfJZGjRogWAJ6ctjI6OxqBBg+Dq6gpPT0+8/PLLSEuz7ovgf3+3BRqVGs1mjSr1+vUftkFTrMKgw18h6cxNRO/U7yGm1iYwMBBLlixBYGCg0KWYXVnHQPyxq4jZcxadf5yKtosm4MyMn1CYkStglcbz3jcXcDk8Vad1s3OL8cK0oxb5JU1UlhOXEgXZr1qtxZlr1v3AZuaAdHJg5/EYfPf7TZ3Xf+vzs7gR9fQTYCJLIVQOCL1vY5ByDgDSyoLUjAKMnHn8qYMdHrf9aAy+X697bhAJ6caddKRnFQqyb2vPAUDaWSClHACAsbNP6DzVc0xCLl756ISJKyIyDq1WK9j38cOMAkRGZwqy76cRbUfX+vXrodFoMGbMmHJHaNjbP3oew+MdXTk5OejWrRvi4uKwYcMGrFy5EqGhoRgwYAA0Guu9CJ5zPwnRO0+jSpcmqNSmfsnrWrUGyRci4eDtjjsbjwtYoXmo1Wrk5uZCrZbetBTlHQMXP/0NrkF+iD92FXFHrwhYofFkZBdi7e4ovdpEx+dgnwBTwRHpS63W4OozRqOZ0uVw6x74wRyQRg4AwPfr9LtYqdFosfRP3tVF1kHI72LmgHWTUhb8uuM28pT6TbX544YIaDQC3DZPpCchv4vjU/KRnKYUbP/GIOUskFIOhN1Ox/GL+nUEHD6bgIh7maYpiMiI4pPzkZJeINj+LfGcQLQdXUePHgUAdOvWrdx14uLiAJTu6Fq5ciXi4+OxY8cODBgwAMOHD8f69etx7tw57Nq1y7RFm9j177dCo1aj2cyRJa9ValUPdV7sjojV+9B6/muwcbATsELTi4qKQvfu3REVpV8niFiUdQyolIXIjUlBRsQDASszrt92RkFZoP8PVj5/iKzBnQfZyM0vFmz/VyJ0u1PSUjEHpJEDkdGZep/UAo8erJuT9/TprYgswVUBny8adifdqp9lJPUcAKSRBVqtFssNeH5RdHwODp6OM0FFRMYl9G/yKzrOnmKppJ4FUsgBAPjJwKnJDW1HZE5Cng8AwGUj5pBWq0VeXh7y8vKgrcBzOmTairS2YP7+/khKSkJYWBgaNmz4xPLCwkJUrlwZWVlZiIuLQ0BAAIB/O8aOHy99d1NQUBC6d++O1atX611Ly5YtkZSk38M6bbVyfKJprfe+9KFwtMego18jfOVeRP52EH23z0da2D1cmPur3tuaJ7+AYpn573gbNmyYXuunpKRgw4YNePHFF1GpUiWd2mzZssWQ0irEHJ//4/psnYfYw5dxc7nhnblCHQNlSXd+AUr7pnq3k2ty4J/5tfELIjKiQkU1pLqNL3PZxQ2D4Ofz9Hnm/XwcobCRQ6XWICm1/JGYSan5aPXik98Jtqp4VMpeqV/RJqZPFhiSA4D4s0BsOZBv1xAZLsMNauub9RPs1HzIOlm2BM8PoZU5lLnsWVlQ0RwAAL+MRbDRWs5ofuaAcYgpCzQyeyR6zjaorVv+EbgWhBq5IiLjSnceDqX9k9e6AOPlAFB+FnjmboVT0XX9ijYhXhsyDjHlAACkuI1HsaKa3u3siu/DN0f/a6NE5pRn1xSZLi+Uucwc14YcC6/DK2+rfkWXQ6PRIDHx0UDVpk2b4urVqwZtR2GUaixQXt6j+VcLCsq+hW/dunXIysqCt7d3SScX8OjZXsOHP3lhJCQkBOHh4QbVkpSUhPj4eL3a2MlsgMoG7U5nrea9gtzYh4hccwAAcOqdHzHoyNd4sO8Cks7qN91PQmICirTmH9n5z+esK6VSWfJPXdvq+9kZgzk+f2MT6hgoU7ViwF7/ZhqtQpDPm0gvzs6AW9mL/HycULWys06bUdjIdV73ccXFaov7O9EnCwzJAYBZoAuLygHPGoCLYU0fpmYC+ZZ1jBM9wUMGyMpepGsWGJoDAJCUlAKosg1qawrMActhMVmgcAc8DWuanatEdjJzgCxcYFG557zmyIGMzGxkZFjO3wmvDVkOi8kBAHDUGnTlu6hYa3HnvERP8KxZ7jmvOa4NKZWFJvk7SU42/Lnwou3o8vPzQ05ODi5duoSWLVuWWhYbG4sPP/wQANC4ceNSyzIyMuDh4fHE9jw9PXHnzh2Da9GXrVYOmHAAREC3pgga1AE7e8woeS0nJhmXP1uPDv+bjJ3dZ0CVp/s8n1X8qwgyYsPZWb8/xH9+wDg6Ourc9vGOUHMx9edvCkIdA2XJcJAh34B2NiiEnwCfN5E+imw88bCcZUmpzz7y9Rm1UxZbWxkqWdjfiT5ZYEgOAMwCXVhSDihtHaD3k+y0WkAmQyVvV9h6WtYxTvRfCdCgvGk5npUFFc0BAPD384Vc66pLqWbBHLAclpIFWiiQYGBbNxdbuCqYA2TZ0h1sUd43uLFy4Gnb8vRwgZOT5fyd8NqQ5bCUHACAhwo1DJmU3N5WAx8LO+cl+q98O1dklLPMHNeGHB1t4WWkv5PH7+iqXNnw3n3RTl04efJk/PTTT6hWrRqOHDmCunXrAgDOnTuHsWPHIiYmBsXFxZg2bRq+++67knZ2dnZ4//33sWDBglLbGz16NC5duoRbt26Zpf7i/AKsrzXGLPsyhtF318HWqezpU0zp4sWLeq0fGRmJl19+GWvXrkVwcLBObVq1amVIaRVibZ8/INwxUJY9Jx5g4JTDerd7+8UQLPmwnQkqIjKe5DQl/Lr9YXD72MOjULWyM+KS81Ct5596tx/VpyY2fFn+8y+FoE8WGJIDALNAF5aUA1k5RajSYwPyC1R6tasd6IZbu4ZBLi/nVhkiC9F8xA6D5+WvaA54uNoh/dQYyGSW83fCHLAclpQFvSbux+Gz+nd33do1DHVruJugIiLjmf39JXyx+m+D2lY0BwDg7O8D0baJ7tO/mhqvDVkOS8qBb9eGYcbXF/Ru98MHbTHlpQYmqIjIeEIvJ6Hza3sNbl/RLPh4YjPMe6u5wft/XF5eHlxcHt2elpubq/fghX/IjVKNBfrggw/g6emJ2NhYNGjQAI0aNULt2rXRrl071K9fHx07dgQANGnSpFQ7T09PZGQ82R+amZkJLy8vs9ROplO7dm0cPHgQtWvXFroUMqG+HauiehX956yaNEL3ix1EQqns7YiASk+fa9mUWoT4CLZvY2AOSIO7qx3GDKild7tJI4LZyUVWoUWIt2D7bl7f26I6ufTFHJCOySPr692mZ7sq7OQiqyBkDtjYyNCknnVfH2MWSMOrg+vCwd5GrzZODgq8PLCOiSoiMp5m9b0h5E9yIXOoPKLt6AoMDERoaCj69esHBwcHxMTEwNfXF6tWrcLOnTsRGRkJ4MmOrvr165f5LK7w8HDUr6//D2WyLAqFAp6enlAoRDtrJwGwsZHj44nN9GozvFcQQmoZOJE/kZm1bCBcZ5Ml/pjRB3NAOqaPbQhXZ1ud1w/0d8Zrz9c1YUVExiPkoANrH/DAHJCOAZ0D0by+7r9bbGxkmDOhqekKIjIiIc8HQmp6wNHBur9DmQXS4OVuj3dG63dn1vSXG8Ld1c5EFREZj4uTLeoJODjHEs8JRNvRBQANGjTA3r17kZOTg+zsbJw9exavv/46MjIykJiYCIVCgZCQkFJtBgwYgFOnTiE2NrbktXPnzuH+/fsYOHCgud8CGVlcXBxmzJiBuLg4oUshExv3Ql3MndhUp3U7t/DDmgWdTVsQkREN6lpdkP36eDqgnQVNUWII5oB01AvywLbvesBJhwsxlb0dsW9pb3i6lfNUdyIL079zNcHuPhzcTZgMMhbmgHQoFHLs+bGXTheB5HIZVn/aCV1a+puhMqKKC/R3QeO6wtxVNahroCD7NSZmgXR8NqUFXuxbU6d1xw6ojXmTjTMVG5E5CPW7vEWID6oIONNQeUTd0VWesLAwAEBwcDDs7Utf0HjjjTfg7++P559/Hnv27MGWLVvw0ksvoXXr1hg8eLAQ5ZIR5ebmIjQ0FLm5uUKXQmYw/60WWLOgM2oHupW53N3VDjNebohDK/rAyZEjuch6jOpTEx4CjDJ7/YW6cLC37r8V5oC0PNc2ACd+7Ydurcq+cCmXy/B89+o4t24gGtTmXb1kPar5uQhyobFJPS+0b2rdAx6YA9Li7+uEM78PxKuD68Deruzpq5rX98a+pb3wymBOVUXWQyaTGTQ9Z0XJ5TJMHG79U/4zC6TDxkaOdV90xaJpLVHZ27HMdfx9nfDlu62wZmFnTmNOVmXisGBBpi+cPLK+RU5lLumOrv9OWwgAbm5uOHbsGPz9/TFq1CiMHz8e7du3x549eyCXS/I/F5FVe2VwHdzaNQwHl/fGlJdC4Ojw6ATXw9UO8YdH4ev32pR70ktkqZwcFXjtefNejJHJHv2IIrI2LRv44tjqfri5fQhmj28Cp//PAVdnW9w/MALb//ccagS4Clwlkf6EuMBpqSe1RE/j5W6PXxd0RvyRUfjmvdYlOeDiqMD59YNw6c/B6N2hqsBVEulvdP9acHPRfZpmYxjUNRDV/PR/HjaRkORyGd4f1wQPDo3Exq+6wfn/Bzo7Oyqw6evuiDkwEjNfa8xOLrI6QVVd0a9TNbPu08PVDqP66HaXpLlZ97BsAz2towsAatWqhT179pizJKNpMn04ms0ciR1d30XmrdhSy/w6NESL2aNh6+IArUaLBwcv4uqiDQAAj+BAtP1iPBx93KEpViHl8m2cn7MamiKVEG+DDCC3U6DVJ68goGtTqItVSL8RjdC3fyi1Tq3hXdDgjQEl/+5UxRvJ5yJw/PWvAAANJg1C7RFdIZPLkHU3AaenLUVRdr5Z34cpyOUy9GpfFb3aV8W2I/cRX5APZ0cFnJ3Me1JAZEyzxzfF+r13kZJeYJb9vTu2IYKqsjPAUrVeMA6BvVvCpVqlUr8BGk0dgtrDu8Ctpj+OvrIYcUcul9nepVoldF01A3K5HDIbObLuJODMzOUoysoz59swqZBanvhsakv8tisK+QX5cHO25YUasmrPta2C/p2rYe/J2GevbARN6nnhVd7xYrFs7G3RZfm7cK9TFWplIZQPM3F21krkxj0sd/2Bh76ESlmEPX3ef2J5r02fwKtBdfzZYJypSzcbbw8HTH+5Eb5dewP5Bflwd7VD60a+QpdFZDAXJ1t8NqUlpnxx1iz7c7C3waJ3WpplX2SY8s4JHLzd0GnJFLhW94O6qBhn31+JlAuRT7RXODmg2+r34N340UVsMWUAANjZ2mBE75qY/tV55ClV8HC1w/BeQUKXRVQhX77bCofPxqOoWGOW/S2a1spiZ8WS5C1K/3R0NW7cWOBKjMurURB8W9Qt92SmKCsPJyZ9hx1d3sWePu/Dr10Igl7oCABQFxbh/OzV2N7pHex6bibsXJ3Q4M1B5iyfKqjFnDHQarTY1mEKdnZ9F5fmr31inbubT2BXz5kl/1MmZ+DetpMAAP/OjVF7eBfsHTAbO7q8i/Qb99Hsg5fM/TaISEc+ng746aMOZtlX3eruWPh2C7PsiwwTs/cs9g3+CLmxKaVeTwwNw5ExnyP5XMRT2+cnp2P/4LnY1XMmdnafgfykNDR5d5gpSyaiCpLJZFj5cQezTGWrUMiwZkFn2NnyLnhLduv3w9jecSp29ZyJ2MOX0e7rN8tdt9kHLyLl0u0ylwWP6/tEnhCRZZo8sj66ljNFs7EtfLsF6gV5mGVfZJjyzglazBmDh5ejsK3DFJyathSdl74Dmc2Tl4Q1KhVuLN2JQyPmm6tkIqqgkFqeZnu23HNtq+CNYfXMsi9DSLKj68yZM9Bqtejdu7fQpRiN3E6Btp+Px7kPfga02jLXSb8RjdwHj8JOXViM9Bv34Vq9MgAgJzoJGRExAACtRoPUa3fhGmjd8++XxdfXF++88w58fcU1ck/haI86I7vhyhd/lLymTMl8ahufprXh4OuBBwcvAQC8Qmog+XwEVHmP7g6JO3IZtYZ1NlnNRFRxQ56rofcUhkmp+YhLzkNSqm53a9rb2eC3hZ3h6GCZI3b0JdYcSD4XgfzE9CdeT70ahZyY5Ge21xSpoC4oAgDI5HIonB7d/U1Elq1KJWcsn6vfoAd9cwB4dHGzabC3vuVZJLHmgLqwGPHHrpb8+8MrUeWez1VqUx9uQf64t+XkE8tcg/wQNLgDwn7cbrJaich45HIZfpnXCd4e9s9e+f8ZkgPdWvlj2pgGhpRokcSaBeWdE9QY1A4Raw4AANL+vgvlw0xUbv3kFMiaIhUST4WJalYHIil475VG6NzCT682+maBr6cDVn3a0aKnMRfHVStCs5mjcG9rqM4j7xx9PVBjQFscHvP5E8sUjvaoO7oHLi1cZ+wyBeft7Y3Ro0cLXYbRudbwQ0FGDppMGwr/zo2hLijCta83IfFUWLlt6rzUHXe3nIBWpQbw6MdOvbE94eDjjoLULNQc2hl2rk6w83BBUSYf0EpkqVbM7Yi0zELs+uuBTuu3enGXztu2Vcix+evuaNtEPAMfxJoDxiC3VWDAvi/gXNUXGeExOPrKIqFLIiIdjOxTE8lpSryz+JxO6+uTAwDw7tgGmPWaeGbCkEoO1H+9H2IPPzldrcLRHq3nv4pjryyGW9B/7gKRydDh60k4P3sVtMVqM1VKRBUVVNUVB37qjR4T9iM7t/iZ6+ubA60a+mDH98/Bpow7gKyVVLIAAOw9XSCTy1GYll3yWm7sQzhX9RGwKiIyJoVCjl0/9ET38ftwJSJNpzb6ZIGHqx0OLu+N6lUs+3EW4kkpCfNtURc+TWsh8v9HZzyLrYsjeqz9AGHLdiI9LLrUMpnCBl1WTEf8X38j9uBFU5QrqOzsbBw5cgTZ2dnPXtmKyG1t4FbDD5lRcdjT532cm70KXZZPg723W5nr2zjaIWhQB0RtOFbyWtLZm7i5YjeeWzcb/fd8DmVyBgBAU8zntBFZMltbObZ80wNjBtQy6nZdnGyxe0lPDOwaaNTtCk2sOWAMmmIVdvWciY2NxyPrbjzqvdxL6JKISEdTRzfAyo87QKEw7gjLj95oim/ea2PRIzf1JYUcaDR1CNyC/HBl0R9PLGv58VhE/noQ+UlPjvhvOGkQks6FI/3mfTNUSUTG1LKBL/5a3Q9+Po5G3W731v44srIv3FxMP02uOUkhCx6n/c/MT2LKdSJ6xN3VDsdW9UOXlvrd2fUsVSo54cSv/dGsvuV3jrOjSwT82oXAvXYAhl1YhmEXlsHJ3xs9N3yEgO7NnlhX4eyAnn98hAeHLiF8xZ5Sy2QKG3RdOR3KlAxcmPuLuco3q4SEBMyePRsJCQlCl2JUuXEPoVGrcW9rKAAgIzwGOQ9S4Fm3Wpnr1xjQDplRcci6HVfq9VtrD2FPn/exd8BsJJ8LR158aslUhkRkuWxt5Vj7WRes+6ILPN0qfhL6XNsquLHtBfTuUNUI1VkWseaAMWmKVbjz53FOX0tkZSYMC8bFPwajST2vCm+rehUXHFnZFwvebiG6i2Fiz4EGbw5C9X5tcGT0Z1Ari55YXql1fTSdPgzDLixDl+XT4BkciMF/fQcAqNw2BLVHdsWwC8vQd+dC2Lk7Y9iFZbBzdzb32yAiAzSr74Mb24bgpX4VHwDnYG+Db2e2waEVfUTXyQWIPwseV5jxaIaexwdCO1f1QV5cqlAlEZGJuLva4ejPffHlu61gb1fxZ+uOHVAbYVuHoHHdip9fmAOnLhSBsB93IOzHHSX/PuzCMhwZ+wUyb8WWWk/h9KiTK/6va7j+3ZZSy2Q2cnRZ/i4KM3Jx5r3l5iibjKgwPQeJp26gStcmiD92Fc5VfeAaWAlZd+PLXL/OqO6l7ub6h2MlDyhTMmHjaIems0YhbNlOU5dOREYik8kwun9tdG9dBR/9eBl/7LuLgkL9ph2qW90d749rjNeeryO6C5v0dM4BPihMz4FKWQjIZKgxqD3SI3SbDpOILEfTYG9c+GMQvl17Az/8EY7Eh7o/fwV4NC3JhKH1MHdiU7g6i+/CptiFTByAoBc64NCI+SjKLvuz39VjRsn/92vXAC0/eRl7+rwPADj68hcly1yq+mLAwcXY0nqyaYsmIqPy9nDA+kVdMaJ3EBasuIbL4fp1ZtjYyPB8t+r4fGpL1K3hbqIqydxidp9F/Vf74No3m+DdpBYcK3kg+UKE0GURkQnY2Mgx87XGGNg1ELN/uIRdfz2AWq3f87dbNfTBxxObYUAX65rhhx1dItf+6zcRe+gSYg9dQsiEfvBtVhu2Tvao3rc1AOD+nrO4/v02BA3ugBr92yL95n0MOvwVACD54i2cn71KyPJJD2dnrUDH795Cy4/GQKvW4MzMFVCmZJY6BgDAtXpleDeuiSOPncj+o9efcwG5HDa2CtzdcgKRv+w399sgogry93XC6nmd8NX01liz8zY2H4rGtVvp5XZ6BVRyQodmlTFhaD30aFOFHVxWqs3n4xHYuxUcK3mg96aPUZxXgG3tp6DRlBcQ/GofOHi7oeP3b0FdUIxdvWaiMC0bTWeOhDI5A7fWHoJHvWpoMWcMAEAmlyEtLBoXPlot8LsiIkPY2drgg9ebYMbLjbDzeAzW7IrC+bCHSM0o+y59NxdbtAzxwej+tTGqT004OfIU0Ro5+Xuh9aevIvt+Evps+RQAoC5SYW//D584HyAi8RvcrToGdQ3ExRupWL45AicuJeFeXE6Z69oq5GhY2xMDuwTijWH1EFCZd3Faq/LOCS4tXIdOP07FkNNLoC5WIXTKEmjVGgAodU4AAAMPfQnHSp6wc3fG8MsrkHTmBkKnLBHybRGRAYKDPLDtu+cQl5SHFVsisfdkLMLupEOlKrvTq1Y1V3Rt6Y+Jw4PRqqGvmas1Dp7FiNDjo+4evzvr+vfbcP37bWW2ubctFPe2hZq8NjKd3AcpODD0kyde/+8dejkxyVhfZ2yZ29jZfUaZrxOR9fFyt8f0lxth+suNoFJpEHEvE1EPsjH+01BkZBfB290eN3cMRWVv487jT8I4P3tVmYNTwpZsR9iS7WW2ufbVxpL/H3/sKuKPXTVZfURkfra2cgzrFYRhvYKg1WoRm5SHsKh0jJ19AhnZRfByt8O5dYNQq5ob5HIOcrB2+YnpWOM/rMxl5c3YkXT2ZsndXP+VG/cQfzYYZ7T6iMj8ZDIZWjfyRetGjy5YZmQX4mpEGobPOIr07CJ4udnh4Io+aFTHyyhTXJHwyjsnKEjNwuFRC8ps8/g5AQDs7jXLJLURkTCq+jljwdstsODtFigsUiMsKh0xCbmYMO8UMrKL4ONhj6g9w+HhZi90qRXGZ3SRpNjb26NevXqwt7f+P14iIl0pFHI0quuFIc/VgJPDozEuDvY2kuzkYg4QkRTJZDIE+rugf+fAkhxwtFegTnV3yXVyMQeISKo83ezRvU0VOP6TAw4KtGzgK8lOLmYBEUmRvZ0NWjbwxdCeQSXnBPZ2NqLo5AJ4RxdJTFBQEH7//XehyyAiIoEwB4iIpI05QEREzAIiIvHhHV1ERERERERERERERERkldjRRZJy69YtdOjQAbdu3RK6FCIiEgBzgIhI2pgDRETELCAiEh92dJGkaLVaFBcXQ6vVCl0KEREJgDlARCRtzAEiImIWEBGJD5/RZaEUjvYYfXed0GXoTOEojofWWQpr+/wBHgNERMZmbVnAHCAiMi5rywGAWUBEZEzMASIi3bGjy0LJZDLYOjkIXQYJhJ8/ERExC4iIpI05QEQkbcwBIiLdcepCIiIiIiIiIiIiIiIiskq8o4skpUaNGtiwYQMCAgKELoWIiATAHCAikjbmABERMQuIiMSHHV0kKQ4ODqhVq5bQZRARkUCYA0RE0sYcICIiZgERkfhw6kKSlMTERCxcuBCJiYlCl0JERAJgDhARSRtzgIiImAVEROLDji6SlKysLOzatQtZWVlCl0JERAJgDhARSRtzgIiImAVEROLDji4iIiIiIiIiIiIiIiKySuzoIiIiIiIiIiIiIiIiIqvEji4iIiIiIiIiIiIiIiKySuzoIknx8vLCK6+8Ai8vL6FLISIiATAHiIikjTlARETMAiIi8WFHF0mKXC6Hra0t5HIe+kREUsQcICKSNuYAERExC4iIxIff6CQpqampWLVqFVJTU4UuhYiIBMAcICKSNuYAERExC4iIxIcdXURERERERERERERERGSV2NFFREREREREREREREREVokdXURERERERERERERERGSV2NFFkuLq6oo+ffrA1dVV6FKIiEgAzAEiImljDhAREbOAiEh8FEIXQGROAQEBmD9/vtBlEBGRQJgDRETSxhwgIiJmARGR+PCOLpKUwsJCxMbGorCwUOhSiIhIAMwBIiJpYw4QERGzgIhIfNjRRZISHR2NoUOHIjo6WuhSiIhIAMwBIiJpYw4QERGzgIhIfDh1oYXSarVQKa1nZInC0R4ymUzoMkTD2j5/gMcAEZGxWVsWMAeIiIzL2nIAYBYQERkTc4CISHfs6LJQKmUh1tcaI3QZOht9dx1snRyELkM0rO3zB3gMEBEZm7VlAXOAiMi4rC0HAGYBEZExMQeIiHTHqQuJiIiIiIiIiIiIiIjIKrGji4iIiIiIiIiIiIiIiKwSpy4kSQkODsaFCxeELoOIiATCHCAikjbmABERMQuIiMSHd3QRERERERERERERERGRVWJHF0lKTEwMxo0bh5iYGKFLITNSqTS4EZWOvScfQFmgAgAUFquRr1QJXBkRmRtzQJq0Wi0eJObi0Jk4KAsfffcXFKmRkqYUuDIiMjfmgHRlZhfi5KXEf3OgUI07D7Kh1WoFroyIzI1ZIE2FRWpcDk+FslANAFAWqnElPBVFxWqBKyMiY+DUhSQpSqUSN27cgFLJC1til5KmxOrtt7H7xANcu5UGZUHpHy6pGYVwbbcW9YPc0aWlP94cHoxGdb0EqpaIzIU5IB2FRWpsORyN9Xvv4tLNVDzMKCi1PC2zEJW7/YGqlZ3RtrEvXh1cB306VIWNDceBEYkZc0A6tFotTl1Jxqptt3D6WjLuxuaUWp6WVYg6AzbD3dUOzYO9MbxXEMYMqAVXZzuBKiYic2EWSEd0XA5WbInEoTPxuHEnA8UqTcmy9KxCtBi1E7YKORrV8UTv9lXxxrB6qBHgKmDFRGQodnQRkahEx+Vg7tLL2HwoGkXFmqeuq9FocfNuJm7ezcSyjRHo1Lwy5k5shp7tAsxULRERGVu+UoUvVv+NFZsjn+jcKktcch62HM7DlsP3ERTgimljGuCtUfXZ4UVEZKW0Wi3+2HcXi3+5jrCojGeun5VThOMXE3H8YiLe/99FvDKoDj6d1AzeHg5mqJaIiEzhcngqPl56GftPxeFZN+4WqzS4EpGGKxFpWPTL3+jfqRrmv9Uczer7mKdYIjIKdnSJiF+7BuizbV6p14rzlMi+l4i7W04iYvU+aNVPv/BP1k3Kx4BGo8VPmyLw/ncXkWfglIShV5LRa+IBjB9SF9+81wZuLhzNSUTWRco5AAChl5Pw2scnnxi1r6vo+By8s/gc/th3F2sWdkZwkIdxCyQiMgMpZ0F8ch4mLjiNvSdjDWqfk1eMHzeEY9PBe1g+twNe6FHDuAUSEZmBlHOgsEiN+cuvYvGv16FW6z81rVYL7DkZiwNn4jD79aaY80YT2NnamKBSIjI2dnSJUPSO04g9cgmQyeDo64Haw7ug9bxX4V4nAGdnrhC6PDIDqR0DufnFGD7jGA6cjjPK9lZtu41DZ+Oxf1lvhNTyNMo2iYjMSWo5oNVq8dnP1/Dx0ivPHLGpi/NhD9F0+A6sntcRo/vXrvgGiYgEILUsOHY+AUOmH0VWTlGFt5WSXoAh7x7FG8PqYdmc9rzLl4isktRyIPFhPvq9dRDXItMrvC2VSov5K65iz8kH2LesNyp7OxqhQiIyJf5aE6H0m9G4tzUU97acxM2fdmFv/9nIjX+Iui/1gL23m9DlCcrf3x/z5s2Dv7+/0KWYlJSOgdz8YvR+84DROrn+8SAxD51f24uw2xX/gUREloM5IL4c0Gq1+PD7S5j7o3E6uf5RWKTGmA9PYNXWW8bbKBEJTio5AEgrCw6cikO/tw4ZpZPrcSu33MKYD09ALdI7H4ikSipZIKUciE/OQ8dX9hilk+txVyLS0OnVPUh8mG/U7RKR8bGjSwJUykKkXrkDmVwOt+qVhS5HUO7u7ujbty/c3d2FLsWsxHoMqNUaDJ1+FGeupZhk+2mZhej15gHEJeWZZPtEZH7MAXHlAAB8u/YGFv9y3WTbf2P+Kew8HmOy7ROReUk1BwDxZsHFGw8xZPoRFBapTbL9Pw/cwzuLz5lk20QkDKlmgVhzICevCL3ePIB7cYZNX/4sUTHZ6DXxAPLyi02yfSIyDnZ0SYRrjUcBVpBhmi99a5GRkYHNmzcjI+PZDyUWGzEeA0v+CMehM/F6tbm4YRBiD4/CxQ2DdFo/KVWJ8Z+GQmvM2wSISDDMAXHlwPXb6fjw+0t6tdE3B7RaYPynp5CSpjSkRCKyMFLOAUB8WaAsUGHs7BNQFujeyaVvDgDA0j8jsC/UsOd+EZHlkXIWiC0HAOD97y4i/G6mzusbkgM37mRg9g/6nXcQkXlJoqMrNTUV7733HmrVqgUHBwdUq1YNM2bMgFKpxJgxYyCTybBq1SqhyzQaG0d72Hu5wt7bDR7BgWjz+Xh4N6qJh1ejkBOdJHR5gkpOTsZXX32F5ORkoUsxKSkcA1ExWZi9RP8fGX4+Tqha2Rl+Pk46tzl4Jh6/bL+t976IyPIwB8STA8XFGrz60UkUq/SbTsqQHEjNKMBbn5/Rt0QiskBSyQFAGlnwybIruHU/S682huQAAEyYdwqZ2YV6tSEiyySVLJBCDhw7n4CfNkXq1cbQHPjhj3CcvJSoVxsiMh+F0AWY2vXr19GrVy8kJyfD2dkZISEhiI+Px7fffov79+/jwYMHAIAmTZoIXKnxNJ0+HE2nDy/1Wsz+8zj3/s8CVUTmJoVj4P3/XdRr5GZFzfruIl7qVwuODtb5tRmfnId1e+/gQWIetFotqlZ2xpgBtRHo7yJ0aURkAlLIgTU7b+NqZJrZ9rfl8H2cvJSIzi2t81kO+UoVNh68h79vpUNZqIKnmz0GdwtE28aVIJPJhC6PiExA7FlwPz4H36y9Ybb9JaTkY/Gv1/HFO63Mtk9j0mi0OHYhAQdPxyMrtwjOjgq0bVwJL/SoDjtbG6HLIyITEHsOaLVas08t+86X53Bl4/NW+/v5VnQmNuy/h+Q0JRQKOWpVdcXYgbXh7eEgdGlEFWadV2x1lJqaigEDBiA5ORmzZs3C3Llz4eLy6KLuwoULMXfuXCgUCsjlcjRs2FDgao3n9vojiN55GnKFDTzqBaLRlOfh6OMBVcG/D+aV2ykw8OCXuLf9FMJ+2Fbyesf/vQUHXw8cGf2ZEKWTkehyDHRdOQNarRYnJn5b8pqdhwue/+s7XJq/Fve2hQpRuk5ik3Kx8/gDs+4zPasQGw/ew6uD65p1vxV1Ly4bs769iB3HY6BWl55+ce7SKxjYpRoWT2uFekEewhRIRCYh9hzQarVYujHC7PtdujHC6jq6CgpV+HjpFfy89RYyc4pKLVv8y3U0C/bGp5OaYVC36gJVSESmIvYsWLElEhqNeacXX7XtNj6d1Bz2dtbVMfTbzih8vupv3I75791vN+Hn44i3R4Xgg9cbw8ZGEpP+EEmG2HPg5OUk3Lhj3uknr0Wm4+zfKWjf1LqecXbxxkN8+P0lHD2f8MSyD76/hFF9amLxu61Q2dtRgOqIjEPUv2KmTp2K2NhYTJ8+HYsXLy7p5AKAOXPmIDg4GCqVCnXr1oWjo3j+kHPuJyExNAzxx6/h5vJdODr2C/g0r412iyaUrKMpUiF06hI0nvoCPEMeXdgI7NMKVXu2xOnpy4QqnYxEl2Pg7PsrULlNfQQ936Hktbafj0fKhUiL/iEDACu33DL7SS0ALBPgompFXL+djrajd2PrkftPdHIBj0Z17jz+AO3G7sbFGw8FqJCITEXsOXD27xT8fSvd7PvddvQ+Eh/mm32/hsrNL0bPNw7gqzVhT3Ry/eNqZBoGv3MES/64aebqiMjUxJwFhUVqrNpm/qnFUzMKsOVwtNn3WxGzv7+EV+eeLKOT65GkVCU++vEyhs04huJi/aYDJiLLJuYcAIS7RmNt14YOnIpDl9f2ltnJBTzK1N92RaHdmN2ISRDPs9tIekTb0RUeHo6NGzeiUqVKWLBgwRPLZTIZWrRoAaD0tIVxcXGYOnUq2rdvDycnJ8hkMty/f99cZZvEwytRuLc1FDWHdIJv8zolr6eHRePG0p3o9MMUOPl7od2XE3F+9iook8X7ME4nJye0adMGTk76zcNr7co6BgozcnF6+jK0+Ww8HCt7onr/tvBr3wBn318hcLXPtv3ofUH2e/FGKuKT8wTZt75S0pToO/kgHmYUPHPdjOwi9H/rkNW8N6KKYA6IJQdiBNmvSqXF3pOxguzbEKM/+Aunrur27Impi85hx7H7pi2IyAJINQcAcWXB6avJSNXhd64pCJVBhli+KQJfrP5bp3V3HIvBO4vPmrgiIssg1SwQUw6oVBrsPmHemX7+sfP4A0EGXxviRlQ6hk4/CmXhsx/9ER2fg76TDyFfqTJDZUTGJ9qOrvXr10Oj0WDMmDHlBpe9vT2A0h1dd+7cwcaNG+Hh4YEuXbqYpVZz+Pu7LdCo1Gg2a1Sp16//sA2aYhUGHf4KSWduInrnaYEqNI/AwEAsWbIEgYGBQpdidmUdA/HHriJmz1l0/nEq2i6agDMzfkJhRq6AVT5bXn4xIqL1e+C0MV0OTxVs3/pYsSUSCSm633XwMKMAP/4ZbsKKiCwDc8D6cwAQ9rvYWnLg4o2H2PWXfif/nyy7Cq3WOk7aiQwl5RwAxJMFzIFnKy7WYN7yq3q1WbHlFuKSOPiNxE/KWSCWHIi4l2nW57Y/Lje/uNy7ZC3N4l+vI79A946riHuZ2HjwngkrIjId0XZ0HT16FADQrVu3cteJi4sDULqjq3PnzkhOTsa+ffswcuRI0xZpRjn3kxC98zSqdGmCSm3ql7yuVWuQfCESDt7uuLPxuIAVmodarUZubi7UamHCUEjlHQMXP/0NrkF+iD92FXFHrwhYoW7+vp0u6MiZy+Fpgu1bVyqVBis2R+rdbtW22ygskt7fBkkLc8D6c0Cr1eJKpHDfxZcjrOMC50+b9J9S5frtR88cIBIzKecAIJ4sEPK7+H5CLtIyhbmbTB87jscgKVWpVxuNRouVW/U/jyCyNlLOAuaAkfZvBYMeHqYrsemg/tPtWtvUjET/EG1HV0zMo+kEatSoUebywsJCnD9/HkDpji65XLT/SXD9+63QqNVoNvPfDrxKreqhzovdEbF6H1rPfw02DnYCVmh6UVFR6N69O6KiooQuRRBlHQMqZSFyY1KQESHMLd/6io4Xdr7ge/HZgu5fF1ci0hCvx91c/0jNKMBpHae4IrJWzAHrz4H0rEJklfO8KXOIjrfs0a3/2HHMsKm1DG1HZC2kngOAOLLgXpyw5wRCn5PowtDvc2uampHIUFLPAuaA9e9fFwdOx6HIgGcvXrppPY/tIHqcQugCTCUv79EfZEFB2SOt1q1bh6ysLHh7eyMgIMCktbRs2RJJSUl6tbHVyvEJWuvVJunsTazxH1bu8qyoeKyt+m+IKRzt0fH7t3Hl8z8Q+dtB9N0+Hy3mjMaFub/qtV8AqFunLopl5n9w7bBh5b/fsqSkPBqlvH//fly+fFmnNi+88ILedVWUIZ8/oP8xYEzmOgby7JoDLoPLXHZxwyD4+Tx9jm0/H8eSf8YeHlXuekmp+Wj14q4nXt+ybReOrx2jR8XmV2BbG3Ada1Db4aNehWMxpzAUs0SP6YDcHYlJiahatarQ5RiFPllgSA4A1pMFUsgBtcwV8Hyv3OXPyoKK5kBaepbF/+1oIUOG16cGtV26fA3++HaIUeshy8IcEHcOANLIgmS3SYDCr8xlxsoBoPws6Nt/MOxVln0xONVlDGBX59kr/kf47RjRfDdQ2aSeAwCvDYkhB7IcewKOHctcZo4c+PLr/2HFgqN6VGx+ufZtAee+BrVt3roTbNWc6UHMLCkLNJp/vzM6duyIq1f1m3r5H6Lt6PLz80NOTg4uXbqEli1blloWGxuLDz/8EADQuHFjk9eSlJSE+Ph4vdrYyWyAyiYq6P+1mvcKcmMfInLNAQDAqXd+xKAjX+PBvgtIOntTr20lJCagSGv+W77/6dDUlVKpLPmnrm31/eyMwRyfv7GZ7RjwqA64lL3Iz8cJVSs767QZhY1c53UfV5CfK8gxoRdnF8DVsKbpaUlAjoW/P6oYVzUgBzRqteUfyzrSJwsMyQGAWaALs+WAjSvgWf5iXbPA0BzQalTW8bfjUQzIbfVulp+bifxEK3h/ZDjmQMk/mQPGZ7YscCws92qGqXMAAFJTkgGlhf/9BOYABkzYoikuEM13A5VD4jkA8NqQKZktBypnAY5lLzJHDuRmZyI3xcL/frweAoa9PaQkxQNF7OgSNQvNguRkw2eaEm1H13PPPYeoqCh8/vnn6N69O+rWrQsAOHfuHMaOHYvMzEwApactNBU/v7JHmj2NrVYOmHAAREC3pgga1AE7e8woeS0nJhmXP1uPDv+bjJ3dZ0CVp/u841X8qwhyR5ezs37f2P/8gHF0dNS5ranv+CuLqT9/UzDXMaC0tUd6OcuSUp89XZ+fjyMUNnKo1Jqnzllf3racHWTwEOCY0IdapkWSVg3IbPRrqNWisocaCjfLfn9UMYk2NtAAkNvYwN/Cj2Vd6ZMFhuQAwCzQhblyQAsFErQaQFb2dNPPyoKK5oANCuFnBX87KZpkFMv1H5nn4ZgHZyt4f2Q45gBzwJTMlQWpChUKy1lmrBx42rYq+bjAVmPZfz/ZdlkwZGItB1kqvEXy3UBlk3oOALw2ZErmyoEcBwXKe7CEOXLA3cUWLraW/fdTZFOIhwCg1QIymc7tZJp8+Ps6QgbLfn9UMZaUBRqNBomJiQCAypUN792XabVarbGKsiQPHjxA06ZNkZGRAYVCgeDgYCiVSty9excDBw5Ebm4ujh8/jl9//RWvvvpqmdtYs2YNXnvtNURHR5f7rC9TKc4vwPpalj092uNG310HWycHs+/34sWLeq0fGRmJl19+GWvXrkVwcLBObVq1amVIaRVibZ8/YL5jIC4pD9V6/Wlw+9jDo1C1sjPikvNQraf+21n1aUe8PqSewfs3l5Ezj+n90NEBnath94+9TFQRWYqqz21AfEo+Aio5Ie7Ii0KXYxT6ZIEhOQAwC3Rhzt8CjYZsw407GQa1rWgODH2uBrZ828OgfZvTmp238drcUL3auLnYIv7Ii3Bx0v9OMLIezAHmgCmZKwtmfXsBX60JM6htRXPAxckWWWfGQi7X/aKhEOKS8lCj70ao1fpd8jm8sg+ea8uLm2Im9RwAeG3IlMyVA8fOJ6DHhP0Gta1oDgDAyV/7o1ML/W9sMCetVovmI3fgWmR5w8XLNv3lhvjmvTYmqooshSVlQV5eHlxcHk3flZubq/fghX+UPRRWBAIDAxEaGop+/frBwcEBMTEx8PX1xapVq7Bz505ERkYCMM8dXWQ5ateujYMHD6J27dpCl0IGCqjshEpe5u9U/UeLEB/B9q2Pt0eF6N/mRf3bEFkb5oA4CPldbC05MLJ3Tfh66peXrw2uy04uEj3mgDgI+V3cvL63xXdyAUBVP2cM6VFDrzbBQe7o3rqKaQoisiDMAuvXvL63YPuWyYBmAu5fVzKZDFNebKBXGxsbGd4crvtAICJLItqOLgBo0KAB9u7di5ycHGRnZ+Ps2bN4/fXXkZGRgcTERCgUCoSE8MKulCgUCnh6ekKhEO2snaInk8nQWaBRM55udmhQ6ykPhrEgnVr4Yd7k5jqvP+u1RujdQRwPIiZ6GuaAOAiVAwDQqbl1PCjB0UGBrd/2gL2dbj/32zTyxedTWz57RSIrxxwQhw5NK+szC5NRWUsOAMCyOe1Rt7q7Tut6uNphyzc9rKITj6iimAXWz8PNHo3regmy72bB3lYzOOy15+tgzIBaOq+//KMOqKNjbhBZGlF3dJUnLOzRFAfBwcGwt7d/YvmWLVuwZcsWXLp0CQCwf//+Uv9O1isuLg4zZsxAXFyc0KVQBbwxTJjRJeOerwtbW+v52pw7sSm+fLcVbGzKP1mVy2X4dFIzLJpm/mkYiITAHBCHEb2D4OZi/pPLkFoe6NDMei5wdmrhh8Mr+sLnGXd29elQFYdW9IGTIy/2kPgxB8Shqp8z+neqZvb9ymTAeCuYxvwfPp4OOPFrP7Ru6PvU9apXccHJX/ujQW3rGNRHVFHMAnF4Y5gw38cTreiOJ5lMhl/nd8bkkfWfup69nQ3WLOiM8UOtJ+OI/st6rtga0T8dXeVNWzh8+HAMHz4cS5cuBQBMnjwZw4cPx48//mi2Gsk0cnNzERoaitzcXKFLoQro0aYK6lR3M/t+3xzx9B8GlkYmk2Hma40Rc2AkPnmzGapXcSlZZiOXYc6EJojePwKfTGoOmVBDYonMjDkgDi5OtnhlUB2z73fyyPpW933ZqYUfYg6MxOp5ndCq4b9TfclkwNgBtXH294HYt6wX3FzsBKySyHyYA+LxrIt2ptC/UzXUCHA1+34rws/HCWfXDcThlX0wpEcNODn8O6jBzlaOjV91w+3dw9BIoDsjiITALBCHsQNqw9nMA7XcXGzxUj/d75CyBAqFHEvntEfkzqGYNqZBqceBKGxk+Gp6a8QfGYVXBpv//IrImCQ5bPNZHV1arX4PaxVa6wXjENi7JVyqVcKOru8i81YsAKDDt5NRqVU9qAuKUJSrxIWPfkH6zftP3VavTZ/Aq0F1/NlgnBkqp4pwDfJDp++nwN7LFUXZeTg1bSmybpc9GsnG3hYDD30JlbIIe/q8/8Tysj73mkM7oeHk5wGNBlotcGXRH4g/dtVUb0cvcrkMs8c3wWtzQ822z5F9glA70Pyda8YQUNkZn05ujk8nN0dAjz+Q8FAJPx9HLJzCKaqIxMrG3hZdlr8L9zpVoVYWQvkwE2dnrURu3MNS67lU9UXH79+GV8MayI5OKjMjLNW0MQ2watstKAvUZtlfQCUnvDzQOp/j4OSowLgX6mLcC3VLcqCKrxPWft5F6NKIyIiqdG2C5h++BJlMDrmtDW4s24m7m0+UXkkmQ8u5YxHQrSnkChskX4jEuQ9+hqZYBZdqldB11QzI5XLIbOTIupOAMzOXoygrT5g39Ay9O1RF8/reuBKRZpb9yWTAB69b5zO+5XIZnmsbgOfaBgBASRb4ejpgRO+aAldHRMZS3vXBx9Ua3gUN3hhQ8u9OVbyRfC4Cx1//CgonB3Rb/R68Gz/6XrD0a4NuLnaY+lIDfLH6b7Ptc9rohlYzbeF/1QvywHez2uK7WW1LcqCytyPee7WR0KURGYWk7+hq3LixwJUYR8zes9g3+CPkxqaUev3BgQvY0fVd7Oo5EzeW7kSXFdOfup3gcX2f2AZZrvZfTsTtdYexveNU3Fi6Ex2+mVTuus0+eBEpl26Xuaysz93OwwVtv5iAwy8uwK6eM3H+o9Xo+P3bRq2/ol4ZVAd9zPRMKV9PByz5oJ1Z9mVq1nYnAhEZ7tbvjzJiV8+ZiD18Ge2+fvOJdYpylbiyeANOTv5egAorpmZVN3xhxmdK/fxJR7g6W/9dT8wBIvHqvHQaTk9bht29ZuLo2C/Q/suJUDiXnrq0zks94BlSHbt7zcL2Tu8AAOqP7wcAyE9Ox/7Bc7Gr50zs7D4D+UlpaPLuMLO/D13J5TKsWdAZtgrzXNaY+lIDq5q+9mmYBUTiVN71wcfd3XwCu3rOLPmfMjkD97adBABoVCrcWLoTh0bMN1fJFfbxm01Rv6aHWfbVsLYnZk+wzgEP/8UcIDGSZEfXmTNnoNVq0bt3b6FLMYrkcxHIT0x/4vXYQ5egVWsAAA+v3IZLVV+U98Re1yA/BA3ugLAft5u0VjIOB283eDeqibtbH/0Yidl7Dq6BlR59xv9RqU19uAX5496Wk08sK+9zl8llkMlkJSfGdm7OyE80z0hJXclkMvz8SUe4u+p30TEpNR9xyXlISs3Xuc2yOe3h6+Wob4lERIJRFxaXugv34ZUouAZWemK9osxcpFyIhCq/0JzlGc2UlxqgU3P9LjoakgPjXqiLvgI8C4aISF927k4AAFtXJxRk5EBTpCq13CukOhJPXoem+NHr8ceuoNawR3d3aopUUBcUAQBkcjkUTg7Qaix7tpNGdb3wyZvN9GpjSA7UDnTD52YcXEFEZIjyrg+Wx6dpbTj4euDBwUsAHuVA4qkwi72TtywO9gqsWdD5qc8m/y9DckChkGHNws6wt7MxpEwiMgNJTl0oRSHj+yPu2BWgrGkZZTJ0+HoSzs9eBW2xeab/EYqvry/eeecd+Po+/WG8ls45wAf5yeklHZkAkBufCueqPqWmpVI42qP1/Fdx7JXFcAvyL72Rp3zuhek5OPv+Sgw69BUKM3Nh42BnkSN6qvo5Y+f3z6HPpIMoKNTt2G314i699jFvcnMM6xVkSHlEZIHEkgP6qv96P8Qevix0GUYnl8uw5Zse6PTqXtyOydKpjb450LWVP378UBx39RKRuHPgxMRv0W31TKjyC2Hn7ozjr39V0qH1j9Rrd1B3TE9E/nYQmiIVggZ3hEu1f/9byG0VGLDvCzhX9UVGeAyOvrLI3G9Dbx+Ob4LI+5lYt+euTuvrmwO+ng7Y+2MvOJn5OTBEZDpizgJ91HmpO+5uOQGtyrqvBbZu5IvVn3bCq3OfHOBdFn1zQCYDflvQBS1CfJ69MhEJRpJ3dElNzaGdUGNgO5yduaLM5Q0nDULSufBnPr9LDLy9vTF69Gh4e3sLXUqF/bfPsqzbjlt+PBaRvx5EftKTI3qe9rnbujii3iu9sbvP+9jSahJOT1+Grj/PgMzG8r4yurT0x54lvUzyANJPJzXD3IlNjb5dIhKOmHJAV42mDoFbkB+uLPpD6FJMopK3I46v7ouQWh5G33aPNlWwe0lPODrw4iaRWIg1B2Q2cjSa8gKOvboYW1pNwsHh89Dx+7dh5+FSar27m08g4a9r6Lt9AXpv+QSZt2KheWzQm6ZYhV09Z2Jj4/HIuhuPei/3Mvdb0ZtcLsOv8ztj7ADjP0fR39cJx1f3Q90a7kbfNhEJR6xZoA8bRzsEDeqAqA3HhC7FKF4ZXAe/zO8Eudy4U/LZ2Mjw28LOeKl/LaNul4iMz/KuWpNR1RjUHk2nj8ChUQtQkJZd5jqV24ag9siuGHZhGfruXAg7d2cMu7AMdu7OZq7W9LKzs3HkyBFkZ5f938Ja5MWnwtnfq1THk3MVb+TFpZZar1Lr+mg6fRiGXViGLsunwTM4EIP/+g7A0z/3Kl2aoDgnH9l3EwAAcYcvw97DGc5VLHP0So+2VXDm94FoGuxllO15udvjj0Vd8cmk5py3mEhkxJIDumrw5iBU79cGR0Z/BrWySOhyTKZKJWec+m0ARhvpBFQul2HWa42wd2kvq33YNBGVTaw54NUwCE5+nki5eAsAkPb3XSiTM+AVUv2Jdf/+bgt295qJ/YPnIutOPLKi4p5YR1Oswp0/j6PWsM4mr90YFAo51izsjK9ntIaDvXGmlerZrgrOrx+IBrU9jbI9IrIcYs0CfdQY0A6ZUXHIuv1kBlir156vi0PL+yDQ3zjXM2tUccGRlX0xdmAdo2yPiEyLHV0iVmNgOzR//0UcHDkPefGp5a539OUvsKXlJGxpPRn7B3+Eoqw8bGk92arm5NVVQkICZs+ejYSEBKFLqZCCtGyk37iPWkMfnXhW798WubEPS01bCAC7eszAltaTsaX1ZJx483/IiHyAnV3fBfD0zz3nQTK8GwXBwdsNAODboi4gk5d5Z5ilaFzXCxfWD8a8yc1hZ2v4V9sLParj5vYheLEfR+sQiZFYckAXIRMHIOiFDjg0cj6KsnWff95aebrZY90XXbHj++fg7+tk8Hbq1/TA6d8GYPG7rTkHP5EIiTUH8uJT4eTvDbdaVQAArjX84Fq9MrLvJZZaz8beFnZuj74j7b1c0ejt5xG2dAeAR9OjKxztH60ok6HGoPZIj3hgtvdQUXK5DDNeaYRrm55H+6ZPPpdSV24utlj5cQccXN4H1fxcnt2AiKyOWLNAH3VGdRfN3VyP69G2CsK2DsEbw+rB0HHLMhkwaUQwwrYNQddW/s9uQEQWgfOwiECbz8cjsHcrOFbyQO9NH6M4rwDb2k9B56XvQJmSiR6/vl+y7sER81CYkYumM0dCmZyBW2sPCVg5VcSZWSvQ8X9vodHUISjOVeLU1CUAgPZfv4nYQ5cQe+iSwdtOD4vG9SXb0WfbPGiK1dCo1Dgx8dsn5vi3NLa2cnz8ZjO8OTwYv+y4jeWbIxGTkPvMdi5Othg7oBYmjaiPRnWNc1cYEZGQnPy90PrTV5F9Pwl9tnwKAFAXqbC3/4elckJup8DQs0thY6+ArasThl9egbtbT+DK59Y7zeHgbtXRp0NVbD18H8s2ReD01eRntpHJgH6dquGtkfXRu0NVo095QkRkagWpWTg7awW6/TwDWo0WkMlw7sNVyE9KL/W9b+vqhL7b50Oj1kBuI0f4z3sR9//PcPSoVw0t5owBAMjkMqSFRePCR6uFfFsGqRfkgVO/DcDpq8lYtjECWw7fR7FK88x2jep4YvLI+hjdvxZcne3MUCkRkXGVd33wv9eJXKtXhnfjmjjy8hdPbGPgoS/hWMkTdu7OGH55BZLO3EDolCXmfisV4uZihxUfd8QH45pgxZZIrN5+G6kZBc9s5+vpgPFD6uGNYfVQI8DVDJUSkTGxo0sEzs9ehfOzVz3x+trAUeW2ufbVxjJfz417iD8bjDNabWQ62XcTsG/gnCdeP/Pe8jLXTzp7E3v6vF/msrI+94hV+xCxal/FCxVAJW9HfPB6E8x8tRFu3MnA5fA0XLr5EDGJuSgs0sBWIYePpz2a1/dBi/reaBHiA2dOTUVEIpKfmI41/sPKXPZ4TmiKVNjcYqK5yjIbezsbvNS/Fl7qXwv343Nw6WYqLkekIvxuJvILVJDLZXBxtEWjOp5o2cAHrRr6orK3o9BlExFVSPSO04jecfqJ1x//3i9IzcL2Tu+U2T7+2FXEH7tqsvrMSSaToWNzP3Rs7oelcwpx6WYqLt1MxbVbacjKKYJKrYWjvQ3qVHdDixAftAjxQb0a7py2nIisWnnXB/97nSgnJhnr64wtcxu7e80ySW1CCKrqikXTWmHe5Oa4FpmGy+GpuByRhsSH+Sgq1sDeTg4/Hye0/P8caBrsBTtbzuhAZK3Y0UVEomVjI0eTet5oUs8b416oK3Q5REQkgBoBrqgR4IphvYKELoWIiATg6WaPnu0C0LNdgNClEBGRAOztbNCmcSW0aWz4tLZEZPn4jC6SFHt7e9SrVw/29vZCl0JERAJgDhARSRtzgIiImAVEROLDO7pIUoKCgvD7778LXQYREQmEOUBEJG3MASIiYhYQEYkP7+giIiIiIiIiIiIiIiIiq8SOLpKUW7duoUOHDrh165bQpRARkQCYA0RE0sYcICIiZgERkfiwo4skRavVori4GFqtVuhSiIhIAMwBIiJpYw4QERGzgIhIfPiMLgulcLTH6LvrhC5DZwpHPsDTmKzt8wd4DBARGZu1ZQFzgIjIuKwtBwBmARGRMTEHiIh0x44uCyWTyWDr5CB0GSQQfv5ERMQsICKSNuYAEZG0MQeIiHTHqQuJiIiIiIiIiIiIiIjIKvGOLpKUGjVqYMOGDQgICBC6FCIiEgBzgIhI2pgDRETELCAiEh92dJGkODg4oFatWkKXQUREAmEOEBFJG3OAiIiYBURE4sOpC0lSEhMTsXDhQiQmJgpdChERCYA5QEQkbcwBIiJiFhARiQ87ukhSsrKysGvXLmRlZQldChERCYA5QEQkbcwBIiJiFhARiQ87uoiIiIiIiIiIiIiIiMgqsaOLiIiIiIiIiIiIiIiIrBI7uoiIiIiIiIiIiIiIiMgqsaOLJEUul6NZs2aQy3noExFJEXOAiEjamANERMQsICISH36jk6RoNBpcvXoVGo1G6FKIiEgAzAEiImljDhAREbOAiEh82NFFREREREREREREREREVokdXURERERERERERERERGSV2NFFREREREREREREREREVokdXSQprq6u6NOnD1xdXYUuhYiIBMAcICKSNuYAERExC4iIxEchdAFE5hQQEID58+cLXQYREQmEOUBEJG3MASIiYhYQEYkP7+giSSksLERsbCwKCwuFLoWIiATAHCAikjbmABERMQuIiMSHHV0kKdHR0Rg6dCiio6OFLoWIiATAHCAikjbmABERMQuIiMSHUxdaKK1WC5XSekaWKBztIZPJhC5DNKzt8wd4DBARGZu1ZQFzgIjIuKwtBwBmARGRMTEHiIh0x44uC6VSFmJ9rTFCl6Gz0XfXwdbJQegyRMPaPn+AxwARkbFZWxYwB4iIjMvacgBgFhARGRNzgIhId5y6kIiIiIiIiIiIiIiIiKwSO7qIiIiIiIiIiIiIiIjIKnHqQpKU4OBgXLhwQegyiIhIIMwBIiJpYw4QERGzgIhIfHhHFxEREREREREREREREVkldnSRpMTExGDcuHGIiYkRuhQiIhIAc4CISNqYA0RExCwgIhIfTl1IkqJUKnHjxg0olUqhSyEyuaJiNQ6ejsf5sBRcDk/DjTsZSHiYDwBITFWi95sH0KK+D1o38kGfDlXhYM9IIPFjDpCUaLVaXAh7iJOXk3ApPBVXI9P+zYGH+Wg7ehdahPigZQMf9O1YFX4+TgJXTGR6zAGSmpiEHBw4HY/L4am4HJ6K2KQ8pGYWAACS05QYO/svtKjvg+5tqqBxXS+BqyUyD2YBSUlOXhH2hcbh4o2HuByRhlv3s5CU+uicIClViUFTDqNFiDfaNamEHm2qwMaG98WQdeJVTSIikXmQmIsVmyOxatstpKQXlLmORqPFoTPxOHQmHgDg7WGPcc/XxaQR9RFU1dWc5RIRkZHl5Rdj3d67WLYxAtdvp5e5jkYLnA97iPNhDwEACoUMQ5+rgckj6qNzS39zlktEREam1Wpx4FQclm2MwN7QWGi1Za+nUmuxbs9drNtzFwDQvmklTBpRH8N7BcHezsaMFRMRkbHdvJOBZRsjsHb3HeTmF5e5jlqjxe4TD7D7xAMAQKC/MyYOC8b4IfVQydvRnOUSVRg7ukTEr10D9Nk2r9RrxXlKZN9LxN0tJxGxeh+0ao1A1ZE58BiQNrVag/+tu4mPfryMgkK1Xm3TMgvx1ZowfL/+Jj55sxlmvdYYCgVH8RBZG+YAHTkXj9c/CcWDxDy92qlUWmw8EI2NB6IxrGcNLJ3dnie3RFaKWSBt9+Nz8PonoTh2IVHvtmeupeDMtRQsWv031izsjJYNfE1QIRGZGnNA2pQFKnz042V89/uNcgc6lOdBYh7mLLmML9eE4buZbfDq4DqQyWSmKZTIyNjRJULRO04j9sglQCaDo68Hag/vgtbzXoV7nQCcnblC6PLIDHgMSM/9+By89MFfOPt3SoW2U1SswZwll7HtaAw2LO6KOtXdjVQhEZkTc0B6CovUmLb4HJZvjqzwtrYcvo+/LiVh9acdMahbdSNUR0RCYBZIzy/bb+OdxefKHbmvq5t3M9F2zG58+HoTzJvcHHI5L3ISWSPmgPRcDk/FS+//hdsxWRXaTlZOEcZ9HIoth6OxZkFn+HpxABxZPg7XF6H0m9G4tzUU97acxM2fdmFv/9nIjX+Iui/1gL23m9DlCcrf3x/z5s2Dv7+4p+ThMSAtkdGZ6PDKngp3cj3ucngqOr6yB3/fSjPaNoksAXOAOSBG+UoVBk45ZJROrn+kZhTg+WlHsHrbLaNtk8gSSCUHAGaB1CxceRWvfxJa4U6uf6jVWixceQ1jZ5+ASsU7P0hcpJIFzAFp+etiIrqO21fhTq7H7QuNQ+fX9iI+Wb/ZIoiEwI4uCVApC5F65Q5kcjncqlcWuhxBubu7o2/fvnB3l9ZdKjwGxCs6Lgc9JuxHQkq+0bedkl6Anm8cwK3oTKNvm0gozAHmgNgUFasx5N0jOHw2wejb1mqBCfNOYf3eO0bfNpFQpJoDALNAzL5eE4a5P14xybb/2HcXr38SCo1Gz/mviCyYVLOAOSBe5/5OwYC3DxltsMPjIqOz0HPiAaRmlP0MeCJLwY4uiXCt8SjACjJyBK5EWBkZGdi8eTMyMjKELsXseAyIj0qlwYiZx0zSyfWPhxkFGDbjGAqL9HvmF5GlYg4wB8Tm02VXcfBMvMm2r9UCr39yChH3Mk22DyJzknIOAMwCMTpxKREzv71g0n2s3X0HyzdFmHQfROYk5SxgDohPRnYhhs44ijylymT7iLiXiXEfh0Kr70O/iMyIHV0iZONoD3svV9h7u8EjOBBtPh8P70Y18fBqFHKik4QuT1DJycn46quvkJycLHQpJsVjQBq+/i0Ml26m6tXm4oZBiD08Chc3DNK5zY07GVi48pqe1RFZJuYAc0BMLt54iMW/XtevjQE5UFikxqsfneTUVSQKUskBgFkgBXn5xRj3caje7QzJglnfXUR0HC+MkzhIJQuYA9Iw/avzeg+ANiQHdp94gHV7ONMDWS6F0AWYQ2pqKhYtWoTt27cjPj4evr6+GDFiBBYuXIgJEyZg/fr1+PnnnzF+/HihSzWKptOHo+n04aVei9l/Hufe/1mgisjceAyI393YbHyyTP/pSfx8nFC1srPe7b5Y/TdG9g5Cwzpeere1FMoCFe48yEaeUgV3VzvUre4GGxuO9yBxYg6In0ajNWgqKUNz4MKNh/hxQzimjW2od1tLodVqERWTjfSsQjg5KlCzqitcnGyFLovIZJgF4jdv+VXcM6DzyZAsyFOqMGnhaRxY3kfv/VmShJQ8JDzMh0wmQ9XKzqjs7Sh0SUQmwxwQv6PnErBmZ5Te7Qw9J5i66Bz6daoGbw8HvdtaiuzcIkTH56CgUA1vDwfUquYKmUwmdFlkBKLv6Lp+/Tp69eqF5ORkODs7IyQkBPHx8fj2229x//59PHjwAADQpEkTgSs1ntvrjyB652nIFTbwqBeIRlOeh6OPB1QFRSXryO0UGHjwS9zbfgphP2wreb3j/96Cg68Hjoz+TIjSyUh0OQa6rpwBrVaLExO/LXnNzsMFz//1HS7NX4t72/QfGUjms/TPcBQVm29kvVqtxZIN4VjxcUez7dNY7sZm46eNEfhlx21kZP/7N1CjigsmDg/G6y/Uha8XT3BJXJgD4nfkXDzCosw73c7/1t/ElJdCrG6QQFZOEX7bFYWfNkUgMvrfh3O7OCnw8sA6mDyyPhrU9hSwQiLTYBaIW25+MZZvjjTrPg+eicfNOxlW952pVmuw+8QDLNsYUeqZljIZ0L9TNUwaWR99OlSFXM4LnSQuzAHx+/b3MLPuLzOnCGt2RmHGK43Mul9juBKeimUbI/DH/rtQFvz7eI7Gdb0waUQwxgyozUFwVs66zlL1lJqaigEDBiA5ORmzZs1CUlISrly5guTkZCxYsADbtm3DtWvXIJfL0bCh9Y5O/a+c+0lIDA1D/PFruLl8F46O/QI+zWuj3aIJJetoilQInboEjae+AM+Q6gCAwD6tULVnS5yevkyo0slIdDkGzr6/ApXb1EfQ8x1KXmv7+XikXIjkDxkLl69U4dcd+o/Yqah1e+4iM7vQ7PutiI0H7iHk+a34Zu2NUp1cAHA/IRcffn8JDYdsw6WbDwWqkMg0mAPit/RP8z8rJSYhF/tPxZl9vxURGZ2JpsO3453F50p1cgFAbr4KyzZGoPGw7fhpI589Q+LDLBC3dXvuICev2Oz7XWZl35d5+cV4/p0jeGHa0VKdXMCj51DuORmL/m8dwugP/uJziUl0mAPidi8uW5Df5j9titB7VgkhabVaLP7lb7QYtROrt98u1ckFANdvp2PSwjNo/dIu3I/nFL3WTNQdXVOnTkVsbCymT5+OxYsXw8XFpWTZnDlzEBwcDJVKhbp168LRUbyj+R9eicK9raGoOaQTfJvXKXk9PSwaN5buRKcfpsDJ3wvtvpyI87NXQZks3odxOjk5oU2bNnBychK6FLMq6xgozMjF6enL0Oaz8XCs7Inq/dvCr30DnH1/hcDV0rPsOBaDzJyiZ69oZPkFKmw+FG32/Rpq1/EYvPTBX8+88y0lvQA93ziA8Lvi/e6jfzEHmANi8DBdiT0nYwXZtxADLQz1IDEX3cfvx/2E3Keup9FoMfmzM/hl+20zVUZCkmoOAMwCsRHq+/j3PXes5pmNKpUGw2Yc0ykz/zxwDy/POWFVF2/JcFLNAuaAuKzddQdaAb6y7sbm4NQV63nG2//W3cQH/7v0zPUi7mWix4T9SElTmqEqMgXRdnSFh4dj48aNqFSpEhYsWPDEcplMhhYtWgAoPW3hli1bMHToUFSvXh1OTk4IDg7GnDlzkJv79BNkS/f3d1ugUanRbNaoUq9f/2EbNMUqDDr8FZLO3ET0ztMCVWgegYGBWLJkCQIDA4UuxezKOgbij11FzJ6z6PzjVLRdNAFnZvyEwgzrPtal4FxYimD7Ph9mHXc+FRSqME6PZ9dk5hRh0sIzJq6KLAFzgDkgBpdupgp2IU7IDNLXjK/PI/Gh7g/mfuvzM0jLLDBhRWQJpJwDALNALAqL1LgamSbIvnPyihFxL1OQfetr7e47OHBa97sdNh2Mxo5jMSasiCyFlLOAOSAe564L97v83HXruDYUm5SL9765oPP69+Jy8NGPl01YEZmSaDu61q9fD41GgzFjxpQ7QsPe3h5A6Y6ur7/+GjY2Nvj888+xf/9+vPnmm1i2bBl69+4NjcY6Ri2VJed+EqJ3nkaVLk1QqU39kte1ag2SL0TCwdsddzYeF7BC81Cr1cjNzYVaLb0pCco7Bi5++htcg/wQf+wq4o5eEbBC0tXl8FTh9h0h3L71sflQNNIy9Ztm8eTlJNy8w7u6xI45wBwQAyG/ixNS8pGUqnvnkVASUvL0vlhZUKg26GHeZF2knAMAs0AswqLSUSzgXVXWcE6g1Wqx9M9wvdv9tMm6pmYkw0g5C5gD4qDVanE5QpgBD4B15AAArNxyS+8Bguv3Wt9jO+gR0XZ0HT16FADQrVu3cteJi3s0sufxjq7du3dj06ZNGD16NLp06YJp06ZhyZIlOHPmDE6dOmXaok3s+vdboVGr0WzmyJLXKrWqhzovdkfE6n1oPf812DjYCVih6UVFRaF79+6IipLmRYyyjgGVshC5MSnIiHggYGWkj7Ao4TpjbtzJsIrpPFYbOP3U6u23jFwJWRrmAHNADITMAeDRPPaW7vc9d6BS659Xq7YxB8RO6jkAMAvEgDnwbNdvp+OKAReBj5xL4DNaJEDqWcAcsH7JaUqkZgg3E4E15ABg2LWh/AIV/jxwzwTVkKnJtFohZvM0PX9/fyQlJSEsLAwNGzZ8YnlhYSEqV66MrKwsxMXFISAgoNxtRUREICQkBH/88QdefPFFvWtp2bIlkpL0m7vUVivHJ5rWeu9LHwpHeww6+jXCV+5F5G8H0Xf7fKSF3cOFub/qva158gsolpl/RNmwYcP0Wj8lJQUbNmzAiy++iEqVKunUZsuWLYaUViHm+Pwf12frPMQevoyby3cZvA2hjgGp0QJI8JpX7vKLGwbBz6f8ecb9fByhsJFDpdYgKbX8eYeTUvPR6sWyjwf/9IWQw/wPvtZHkvu7UNt46N3OoSgC3rl/Gr8gC5LoMR0auTvkmiz4Z34rdDlGoU8WGJIDgPizgDlgXVJdxqDQrk6Zy4yVA0D5WeCV8ycciy17xHuG00DkO7TUv6G2GAEZC41fkAVhDjAHysMssB659m2Q5dyvzGXPygGg4ucETgWX4Zlv+HFiDkrbYKS76n/9BgB8sn+Fveq+cQuyIFLPAYDXhsrDHLAexXJvpHhMLXe5qa8NWcP3hxYKJHjNNaiti/IU3JWHjVyRZbGkLNBoNEhMTAQANG3aFFevXjVoOwpjFmVJ8vLyAAAFBWX3bq9btw5ZWVnw9vZ+aicXAJw8eRIAEBwcbFAtSUlJiI+P16uNncwGqGzQ7nTWat4ryI19iMg1BwAAp975EYOOfI0H+y4g6exNvbaVkJiAIq35b/n+53PWlVKpLPmnrm31/eyMwRyfv7EJdQxIklf5i/x8nFC1svMzN6Gwkeu0XlkSExMBjYXfxu2qBWz0b1ZQUCjI37xZuaoBOaBRq0XzXvXJAkNyAGAW6II5YEY1CoFybsI3Rw6kp2cA2Rb+/RGgBBwMaSgTzXdjuZgDJf9kDhgfs8BMvLOAcr7Cdc0BwPAsyM/PR76lf3+4+QGuhjVNTU0D8iz8/VWExHMA4LUhU2IOmImdCvAof7Gpzwk0aq3lf3/I7J56/expcnPzkJtk4e+voiw0C5KTkw1uK9qOLj8/P+Tk5ODSpUto2bL0aM7Y2Fh8+OGHAIDGjRs/dTvx8fH4+OOP0atXLzRr1szgWvRlq5UDJhwAEdCtKYIGdcDOHjNKXsuJScblz9ajw/8mY2f3GVDl6X4LbBX/KoKM2HB21u/L+J8fMI6Ojjq3fVZHqCmY+vM3BaGOASlK0BZBKyv7Cueznpuiz6idMmk1qOLvCxks+2bgFFk+iuGpdzsnOxU8BfibN6dEGxtoAMhtbOAvkveqTxYYkgMAs0AXzAHzSbOXo7xfacbKgadty9vLFQ6ulv39ke2ogSETT9locuEnku/G8jAHmAOmxCwwjzx7Z2SWs0yX5yhW9JzA2ckWHhb+/VGosIXeT5DRagGZDJW8HGDrYdnvryKkngMArw2ZEnPAPFRyNzytO8DU14Zs5GqL/82sBZCoLYBWpv/oNzdnwNXC319FWVIWPH5HV+XKhvfui3bqwsmTJ+Onn35CtWrVcOTIEdStWxcAcO7cOYwdOxYxMTEoLi7GtGnT8N1335W5jdzcXHTt2hVJSUm4cOECqlSpYrb6i/MLsL7WGLPtr6JG310HWyeDhs1WyMWLF/VaPzIyEi+//DLWrl2r8x16rVq1MqS0CrG2zx8Q7hiQouYjduBqpGEPHY09PApVKzsjLjkP1XrqP0VfvRruiNyl37QQQvjf7zfw7lfn9W538tf+6NRC/8EJ1qTqcxsQn5KPgEpOiDti2HQulkafLDAkBwBmgS6YA+Yz54dL+HzV3wa1rWgOAMC9fSMQVNXAYfJmEhmdifqDt+rd7sPXm+DzdwyY8tCKMAeYA6bELDCP0MtJ6PzaXoPbVzQLls1pj0kj6xu8f3PQaLSo3X8zovV83lazYG9c3jgYMpnMRJUJT+o5APDakCkxB8xDo9HCo8PvyMkz7LESFc2BgV0CsWtJT4P2bU6TF57GT5si9WpjYyPD/f0jUdXPsNkvrIUlZUFeXh5cXFwAPOqP0Xfwwj/kxizKknzwwQfw9PREbGwsGjRogEaNGqF27dpo164d6tevj44dOwIAmjRpUmb7goICDB48GNHR0Th06JBZO7nIdGrXro2DBw+idu3aQpdCZLAWId4C7ttHsH3r45XBdeDooN/chQ1re6JjcyubF4L0xhwgMRDyu9jTzQ41AlwE27+ugoM80L21v15t5HIZJg6vZ6KKyFIwB0gMmtX3hpD9MNZwTiCXyzBphP6Pn5g8sr6oO7noEWYBWTu5XIZmwUJeGxJu3/qYNEL/QRmDu1YXfSeXWIm2oyswMBChoaHo168fHBwcEBMTA19fX6xatQo7d+5EZOSj3tyyOrqKi4sxYsQIXLhwAfv370dISIi5yycTUSgU8PT0hEIh2lk7SQJaNfQVbN8tG1j+SS0AeLrZ47Mpuo/IV9jI8N2sNjyplQDmAImBkN/FrRr6Ws135ZfTW8PJQfe/9VmvNUL1KpZ9pxpVHHOAxMDFyRb1a3oIsm9bhRyN6+o/RbgQJg4PRqM6utfappEvRvevZcKKyFIwC0gMWgl8TmANGtX1wpvDdR/04O5ih4VTWpiwIjIl0XZ0AUCDBg2wd+9e5OTkIDs7G2fPnsXrr7+OjIwMJCYmQqFQPNGJpdFoMHbsWBw+fBi7du1C69atBaqeTCEuLg4zZsxAXFyc0KUQGWxIjxqwt9PvbiVjsLGRYUSvILPv11DTxjTAJ28++9mKdrZy/LG4G55rK+75l+kR5gCJQaC/Czo2E+YO1NH9rOcCYIsQH+z84Tm4OD37ItbkkfX1GiBB1os5QGIh1Pfx0OdqwMHeOjoH3FzssH9ZbzSo5fHMdVs28MHuJT3hqMcACbJezAISg5cEyoFKXg7o0cZ6Zj5b8mE7jB3w7Ls3vdztsW9ZL8EGklDFibqjqzxhYWEAgODgYNjb25da9vbbb2Pjxo2YMWMGHB0dce7cuZL/MQCtX25uLkJDQ5Gbmyt0KUQG8/F0wMje5u9wer5bdQRUtp7bt2UyGT6d3Bz7l/VGr/ZPdmLZ2coxun8tnFs3EMOtqAOPKoY5QGIxWYBno3i522OEAPlTEc+1DcDFDYMxYWi9Mu/u6tS8MjZ93R0/zm4Hudw67lSjimEOkFi8PqQebBXmv6QjRP5UREBlZ5xeOxAL3m6OqmWcywQFuGLxtFb4a3U/+Ho5ClAhCYFZQGLQPMQHbRqZ/86q11+oJ8jga0MpFHL89llnrP2sc5n/vVydbfHWqPq4uGEQ2jfl4yysmaQ7usqatnDfvn0AgM8++wzt2rUr9b9Vq1aZtU4iovJMecn8U6oKsU9j6NOxKg4u74Pbu4fB080OwKNnzMQeGoV1X3RFs/rWMR0jEdHjhvasgSqVnMy6zwlD61nNKP7HBQd5YOUnHRF/ZBS2fNMdHq6PsqCSlwNOrhmA4b2CrGY6RiKif1T2dsSoPjXNus+mwV5W+Uxbd1c7fPRGM0TvH4GDy3uX5ICPhz2i9gzDrHGN4exkK3CVRET6m/pSA7Puz1Yht8pn2spkMowdWAfn1g/ClY2DS64NebnbIf7IKPw4uz1qVnUTuEqqKOs7UzWCp3V03b9/38zVGJ+NvS26LH8X7nWqQq0shPJhJs7OWoncuIdPrNty7lgE9GgOrVqDwowcnHlvOXLuJwlQNRmLrp+/b4u6aLdoAgBAZqtAyoUInP/oF2iKVEKUTXpq2cAXk0YE46dNkWbZ38sDa6NLS3+z7MtU6lR3h5ODAhnZRXByUKCSN0dskrjI7RRo9ckrCOjaFOpiFdJvRCP07R9KraPrd3+vTZ/Aq0F1/NlgnNnqJ/3Y2drgpzntMfidI2bZX1CAKz56o6lZ9mUqHm72GNozCO8sPofMnCJB7oQgMqWef86Fo68HoNGgOK8A5+esRvrN+6XWcanqi47fvw2vhjWQHZ2EPX3e/3dZtUroumoG5HI5ZDZyZN1JwJmZy1GUlWfeN0I6+3J6K+w5+QAZ2UUm35dcLsOKuR2temCAQiFHr/ZV4eyoQGZOEeztbGBjwywg8Rh2YRnUhcVQFzz6Tri+ZDvu7zpTah2/dg3w3PrZyL6bUPLa3oFzoC4oYg5YoRf71cTq7bdw7EKiWfb3yZvNrP6Zts3q+5RcG3K0V8DV2U7okshIJN3R1bhxY4ErMZ1bvx9G/LGrAIDg1/qg3ddv4vCoBaXWqda7FSq3DcGu596DVqVG42lD0fzDl3Bi4rdClExGpMvnnx5+H7v7fgCtSg3IZOi26j3UG9sLEav3CVEyGWDxu62w71QcYhJ0n24hKTW/1D914e/rhP+931bv+ojIvFrMGQOtRottHaYAABwreTyxji7f/cHj+iI3NgVeDaqbq3Qy0KBu1TG6fy2s33tX5zaG5AAA/DK/E1w42p3Iop144xsUZT/62w7s0wodvpuM3b1mlVqnKFeJK4s3wM7VCU1njiy1LD85HfsHzy25QNp6/qto8u4wXPz0N/O8AdKbn48TlnzYDmM+PKFXO0OyYOarjdBagCmyiEg/x8d/jcxbsU9dJ/N2XKmBDv9gDlgfmUyG1fM6odHQ7cjNL9a5nSE50CLEB++PE++1dLJ+kuzoOnPmzLNXsmLqwuKSTg4AeHglCiET+pe5ro2dAjb2tlCp1LB1cUR+Ypq5yhSEr68v3nnnHfj6ivcHuq6fv1r576g/GzsFbBzsoNVozFIjGYersx3+XNwN3Sfsg7JArVObVi/u0msfdrZyrP+iCzzd7J+9MpEVEGsOKBztUWdkN2xqPrHkNWVK5hPrPeu73zXID0GDO+DUtB8R2KeVSWsm41jyYTtcjUxD+N1MndbXNwcA4OOJzdC1lXXf1Uv0D7HmAICSTi4AsHV1glajfXKdzFykXIiEX7snpzp6/O5emVwOhZNDqW2SZXqpXy0cu5CIX7bf1rmNvlnQsVllfDqpmb6lEVksMWdBRTAHrFONAFes+rQjXnz/OLRPRn+Z9M0Bbw97rP+iCxScEYEsmCQ7uqSm/uv9EHv48hOvxx66BL/2DTDy+iqocpXIS0rHgRc+EaBC8/H29sbo0aOFLsOsyvv8gUdTl3T/dRZcg/wQd+QKbq8zz/RHZDxtm1TCzu97YvA7h3Xu7NKVna0cm7/ujm6tqxh1u0RCEmsOuNbwQ0FGDppMGwr/zo2hLijCta83IfFU2BPrlvvdL5Ohw9eTcH72KmiLjft9Qqbj6WaPwyv6oNvr+3E7Jsvo2393bAN8OpkXN0k8xJoD/+j4wxT4t3/UiXX4pYV6t5fbKjBg3xdwruqLjPAYHH1lkbFLJCOTyWRYMbcDlAUqbNh/z+jbb9PIF3t+7GWVz2gkKo+Ys6Dz0ncgk8vw8EoULn/xBwrTsp9Yx71WFQw89CW0ag2i/jyOW78dLFnGHLBOI/vURG5+MSbMO6VzZ5euvNztcWh5H9QL8jDuhomMjN2wItdo6hC4BfnhyqI/nljm3bgm3GsHYHOzN7Cx6RtIDA1D289fF6BK88nOzsaRI0eQnf1k0IvR0z5/AMiNe4hdPWdiY+MJsLG3RfV+bcxcIRlDz3YBOLS8D6pUcjLaNit5OWDf0t4Y1I1Tl5G4iDUH5LY2cKvhh8yoR9OQnJu9Cl2WT4O995MP1C3vu7/hpEFIOhf+xPNcyPJVqeSM0DX90al5ZaNt08ZGhs+ntsQ377Wx6uexEP2XWHPgH6emLsHmlm/iyuINaPnxy3q31xSr/j8jxiPrbjzqvdzLBFWSsSkUcvz+eRdMG/PknXoVMbhbII783Bfurnx+CYmLWLNg/wsfY9dz72FXr1kozMhFp+/ffmKdtLB72NR8Inb3moVj475EvZd7ocbAdiXLmQPW6/Uh9bDxq+5wdTbedON1qrvhxC/90DzEx2jbJDIVdnSJWIM3B6F6vzY4MvqzUlMV/aP2iK5IOn3j0W3IWi3ubvoLfh0aClCp+SQkJGD27NlISEh49spW7lmf/+NU+QWI3nEaNYd0MlN1ZGwdm/vhxrYheGVQnQpva1Sfmri5fSh6tOWdXCQ+Ys2B3LiH0KjVuLc1FACQER6DnAcp8Kxbrdw2//3ur9w2BLVHdsWwC8vQd+dC2Lk7Y9iFZbBzdzbLe6CKqeTtiL9+6Y//zWoDRwebCm2rUR1PXFg/CB+Ob8JOLhIdsebAf93dfAJ+7RvA3tPFoPaaYhXu/HkctYZ1NnJlZCo2NnJ8N6stjq3qixpVDPvc/+Hhaoe1n3XG9v89x+czkiiJNQvy4lMBAFqVGuE/70HlNvWfWKc4V4ninEfTEeYnpiN6x6ky12MOWKfhvYJwY9sQ9GofUKHtyGTA9Jcb4tqmF9CwjpeRqiMyLXZ0iVTIxAEIeqEDDo2cX+58ujkPkuHfqRFkikcXQ6r1aomMyAfmLJNMRJfP37V65ZLPXm6rQGC/NkiPiDFnmWRknm72WLOwM47+3BeDugZCLtf94qRMBvTtWBUHfuqNDV92g4+ngwkrJSJjK0zPQeKpG6jStQkAwLmqD1wDKyHrbnyp9Z723X/05S+wpeUkbGk9GfsHf4SirDxsaT0ZRVl55n0zZDC5XIZ3xjTEjW1D8PaLIXqP5mxQywNLZ7fDpT8Hc9QmkZWxdXGEY2XPkn8P7NsahRm5KMzI1XkbzgE+UDj+/3NZZTLUGNQe6RE8P7Q23VpXQdi2IVg8rRWCAlz1auvj6YD3xzVG+I6hGDuwDgc7EFkRhaM97Nz+neUl6IWOSLsR/cR6jpU8Hl0AAKBwdkDV51og7cZ9AMwBsQj0d8GBn3pj09fd0bmFn15tbRVyvNi3Js6vH4Rv3msDJ0dOW0vWg0erCDn5e6H1p68i+34S+mz5FACgLlJhb/8P0f7rNxF76BJiD11C5K8H4FGnKp4//i3URSooUzJwduYKYYunCtP18/dr3wAhbwyAVq2BTGGDpFNhuP7dFmGLJ6Po3qYKurepgpiEHKzbcxcXbjzE5fBUxKeU7vT093VCi/reaNXQF6P710Ktak9OcUZE1uPsrBXo+N1baPnRGGjVGpyZuQLKlEx+90tQzapuWPJhO3w+tQU2HojGyctJuByRisjoLGg0/07a7+Jki+b1vdEixBvPd6uOTi38eFGTyErZujmh26r3oHCwg1ajRUFaNo6+/AUAlMoBuZ0CQ88uhY29ArauThh+eQXubj2BK5//AY961dBizhgAgEwuQ1pYNC58tFrIt0UGcnGyxaxxjfHeq41w4HQc9oXG4nJ4Kq7dSkdB4b/P4VQoZGhQyxMtQnzQvbU/hj5Xg8/iIrJSDr7u6LZqJmQ2cshkQE5MCk5NWQKgdA5U798W9V7pDa1KDZnCBjG7z+LOn8cAgDkgIjKZDMN7BT26wysqHX8euIdLN1NxOSINqRkFpdYNCnBFixBvtG1cCWMG1EZlb0eBqiaqGP6CEaH8xHSs8R9W5rIz7y0v+f+aIlWpfydx0PXzj9pwDFEbjpmrLBJA9SqumPNG05J/z8wuRJ5SBa0WcHZSwNPNXrjiiMjoch+k4MDQT5543ZDv/ty4h/izwTij1kfm5+psh/FD62H80HoAAGWBCpk5RShWaeBgZwMfTwe97v4lIsuVn5CGvf0+LHPZf88BN7eYWOZ68ceuIv7YVZPUR8KQy2Xo16ka+nV6NJWxSqVBelYhCorUsLOVw9PNHvZ2FZvulogsQ+6DFOzuNbPMZY/nQOSvBxD564Ey12MOiFPDOl5Y+P/TD2q1WqRnFSK/QAUbuQyuzrZwdeZzGEkc2NFFkmJvb4969erB3p4X+El6PNzs4cHOLZI45gBJmaODAo4O/PlP0sYcIClTKOSoxJH6RMwCkiyZTAZvDwd4C10IkQnwTJckJSgoCL///rvQZRARkUCYA0RE0sYcICIiZgERkfjIhS6AiIiIiIiIiIiIiIiIyBDs6CJJuXXrFjp06IBbt24JXQoREQmAOUBEJG3MASIiYhYQEYkPO7pIUrRaLYqLi6HVaoUuhYiIBMAcICKSNuYAERExC4iIxIfP6LJQCkd7jL67TugydKZw5AM8jcnaPn+AxwARkbFZWxYwB4iIjMvacgBgFhARGRNzgIhId+zoslAymQy2Tg5Cl0EC4edPRETMAiIiaWMOEBFJG3OAiEh3nLqQiIiIiIiIiIiIiIiIrBLv6CJJqVGjBjZs2ICAgAChSyEiIgEwB4iIpI05QEREzAIiIvFhRxdJioODA2rVqiV0GUREJBDmABGRtDEHiIiIWUBEJD6cupAkJTExEQsXLkRiYqLQpRARkQCYA0RE0sYcICIiZgERkfiwo4skJSsrC7t27UJWVpbQpRARkQCYA0RE0sYcICIiZgERkfiwo4uIiIiIiIiIiIiIiIisEju6iIiIiIiIiIiIiIiIyCqxo4uIiIiIiIiIiIiIiIisEju6SFLkcjmaNWsGuZyHPhGRFDEHiIikjTlARETMAiIi8eE3OkmKRqPB1atXodFohC6FiIgEwBwgIpI25gARETELiIjEhx1dREREREREREREREREZJXY0UVERERERERERERERERWiR1dREREREREREREREREZJXY0UWS4urqij59+sDV1VXoUoiISADMASIiaWMOEBERs4CISHwUQhdAZE4BAQGYP3++0GUQEZFAmANERNLGHCAiImYBEZH48I4ukpTCwkLExsaisLBQ6FKIiEgAzAEiImljDhAREbOAiEh82NFFkhIdHY2hQ4ciOjpa6FKIiEgAzAEiImljDhAREbOAiEh82NFFREREREREREREREREVonP6LJQWq0WKqX13EKtcLSHTCYTugzRsLbPH+AxQERkbNaWBcwBIiLjsrYcAJgFRETGxBwgItIdO7oslEpZiPW1xghdhs5G310HWycHocsQDWv7/AEeA0RExmZtWcAcICIyLmvLAYBZQERkTMwBIiLdcepCIiIiIiIiIiIiIiIiskq8o4skJTg4GBcuXBC6DCIiEghzgIhI2pgDRETELCAiEh/e0UVERERERERERERERERWiR1dJCkxMTEYN24cYmJihC6FiIgEwBwgIpI25gARETELiIjEh1MXkqQolUrcuHEDSqVS6FKIyEwysgtxJTwNVyJSkZVbBADIyi3CjxvC0SLEG03qesPJkXEoFcwBIukpLFIjLCodl8PTEB2fg8ycR1mQk1eMg6fj0CLEBz6efGi6VDAHiKRHo9HizoNsXA5PRfi9zJIcyM4twqaD99AixAc1q7pCJpMJXCmZC7OASHqS05S4HJ6KqxFppa4Nrdp6Cy1CvNGgtifsbG0ErpIqglf2iIhIdPLyi/HHvrtYvjkSVyLSnliem6/ClC/OAgBsbGTo3tofk0fWx4DOgVAoeLMzEZG102i0OHw2Hss2RmD/qTgUqzRPrJOdV4w+kw4CAEJqeWDisGC8PLA2PNzszV0uERGZQPjdDPy0KRLr995BRnbRE8tz8lUYOfM4AMDH0wEvD6yNSSPqo3agm7lLJSIiE0jNKMCvO27j5223EBWT/cTy3HwVJsw7BQCwt7PBwC7V8NaoEHRp6cfBD1aIHV0i4teuAfpsm1fqteI8JbLvJeLulpOIWL0PWvWTJ/kkHjwGSOqKizX4as11fLkmDFk5T57MlkWt1uLw2QQcPpuAqpWdsXhaK7zYryZ/1JBVYg4QAXtPPsC0L8/jzoMnT2bLE343E+8sPocPv7+EKS+F4NNJzeBgz1Mlsk7MApK6qJgsvP3FWRw6E69zm9SMAny79ga+XXsDg7oGYsmH7RDo72LCKolMhzlAUpebX4zZP1zCyi23UFik1qlNYZEaWw7fx5bD99GglgeWfNgO3VpXMXGlZEw8exOh6B2nEXvkEiCTwdHXA7WHd0Hrea/CvU4Azs5cIXR5ZAY8BkiKwm6n49W5J8u8g0tXccl5GP3hX9h8OBo/fdQefj5ORqyQyHyYAyRFGdmFmLb4HNbuvmPwNvILVFj8y3XsPB6DX+d3RtsmlYxYIZF5MQtIajQaLX5YfxOzl1yCskC3C5tl2fXXAxy/mIhvZrTG+KH1OACOrBZzgKTo+IUEjPs4FPcTcg3exs27meg+fj8mj6yPxe+2gouTrRErJFPh/EwilH4zGve2huLelpO4+dMu7O0/G7nxD1H3pR6w95b2Lfj+/v6YN28e/P39hS7FpHgMkNTs/usBWr20q0KdXI/bcSwGzUbswI2odKNsjywHc4A5QOJ0Pz4HrV7cWaFOrsdFRmehwyt78NvOKKNsjyyHVHIAYBaQtBQWqTF8xjG8+9X5CnVy/SMnrxhvzD+N1+aGQlXG9Ldk3aSSBcwBkpolf9xE9/H7K9TJ9bhlGyPQbsxuJD7MN8r2yLTY0SUBKmUhUq/cgUwuh1v1ykKXIyh3d3f07dsX7u7uQpdiVjwGSMx2HY/BC+8e0fl2dF0lpSrRZdw+3LyTYdTtkrCYA8wBEp+YhBx0fm0v7sbmGHW7Go0Wr849iV933DbqdklYUs0BgFlA4lVcrMHQ6Uex7eh9o2/7t11ReHnOCag5zZuoSDULmAMkZv/7/QamLjpn9O3euJOBLuP2IjlNafRtk3Gxo0siXGs8CrCCDONeALA2GRkZ2Lx5MzIypHfhmscAidHViFSMmHkcarXWJNtPzypE7zcPID2r0CTbJ/NjDjAHSFzylSr0nXwIsUl5JtvH+E9P4dj5BJNtn8xLyjkAMAtInKYuOou9J2NNtv0N++/hox8vm2z7ZH5SzgLmAInRjmP38e5X5022/aiYbAx4+xCKiznowZKxo0uEbBztYe/lCntvN3gEB6LN5+Ph3agmHl6NQk50ktDlCSo5ORlfffUVkpOThS7FpHgMkBQUFavx6txQve7kurhhEGIPj8LFDYN0bhOfko+pi84aUiJZIOYAc4DEZe7Sy4i4l6lXG32zQKPRYtwnocjJKzKgQrI0UskBgFlA0nDoTByWb47Uq40h5wRf/hqGc3+n6FseWSipZAFzgKQgNaMAE+ef1quNITlw6WYqFv3yt77lkRkphC7AHFJTU7Fo0SJs374d8fHx8PX1xYgRI7Bw4UJMmDAB69evx88//4zx48cLXapRNJ0+HE2nDy/1Wsz+8zj3/s8CVUTmxmOApOCzlX/j+m39nqHl5+OEqpWd9d7X+r13MbxXEAZ3q653WyIhMAdICs5cS8Z3v9/Qu50hWRCTkItZ317ET3M76L0/IqEwC0jssnOLMP7TU3q3MyQH/pnO9uqm5+HoIIlLaSQCzAGSgilfnEVKeoFebQy9NrRgxTUM7lYdjet66d2WTE/0d3Rdv34dDRs2xDfffIPk5GSEhISgqKgI3377LcaMGYNbt24BAJo0aSJwpcZze/0RHBwxD4dfWoiL89aiID0bjj4eUBX8OwpVbqfA4OPfotHUIaXadvzfW3hu/Rxzl0xGpssx0HXlDHRZMb1UOzsPF4y49jNqDulk7pKJ9JKZXYivfrtu1n1+tOQytFrTTJFoalk5RfhxQzi6vPbvvNLpWYU4eSnRat8TPR1zgKTg46VXYM6vsJVbbyEmwTqn+VGrNdh1PAaDpx5GcvqjHEjNLMC6PXdQUKgSuDoyFWYBid3KLbdMOnXtf926n4X1e++abX/Gdi0yDW/MO4WU/8+Bh+kF+GzlNT53RsSYAyR2N6LS8eeBe2bbX7FKg/nLr5ptf8aWlJqPhSuvot2YXSXf/RnZRfj7VprAlRmHqDu6UlNTMWDAACQnJ2PWrFlISkrClStXkJycjAULFmDbtm24du0a5HI5GjZsKHS5RpNzPwmJoWGIP34NN5fvwtGxX8CneW20WzShZB1NkQqhU5eg8dQX4Bny6A6FwD6tULVnS5yevkyo0slIdDkGzr6/ApXb1EfQ8/+OTG77+XikXIjEvW2hQpRNpLPfdkVBWaD7lIXGcONOBk5dsa6pLbRaLb5Y9TcCntuAKV+cxcnLSVD9//PMlIVqdBm3D02GbceNKP3ujCPLxxwgsYuMzsRRMz83S6PRYuWWW2bdpzEcORePmv02YfA7R7DrrwdQqR7lQGGRBmNnn0C1Xhuxbs8dgaskU2AWkJhpNFr8tCnC7PtdtjHC6gaKxSXloctre9FsxA78vPUWiv8/B4pUGnz042VU6/knpi46y2fPiBBzgMTup036TV1rDDuOxyA+2XyDLIyhqFiNtz47g2o9/8TcH6/g3PWHJdeG8gtUaDp8B7q9vg8JKdb1vv5L1B1dU6dORWxsLKZPn47FixfDxcWlZNmcOXMQHBwMlUqFunXrwtHRUcBKTevhlSjc2xqKmkM6wbd5nZLX08OicWPpTnT6YQqc/L3Q7suJOD97FZTJ4n0Yp5OTE9q0aQMnJyehSzGrso6BwoxcnJ6+DG0+Gw/Hyp6o3r8t/No3wNn3VwhcLdHTabVaQX7MABDkZNpQWq0W0xafw+wfLiFPWf5o/bCoDHR4ZQ+uhKeasTrhMAeYAyQOK/R8Houx/Lz1FoqKzTvQoiJ2HY9B30kH8SCx/JPW1IwCjJ19Akv/DDdjZcKRag4AzAISl8Nn43Evzvx32V6NTMOFsIdm36+hHiTmot3Y3Th5ufznMRWrNFjyRziGv3cUKpU0OrukmgXMARKT3PxirN1t/sFaarUWP2+1nsFvKpUGQ989imUbI0o6t8ry18VEtBu7G3FmvFPa2ETb0RUeHv5/7d15eFTl/f7xeyaTnZCEJJCwBDABEvZ9cWETARWpCxQU0LpVa60LKCrWVoRat0qtLW64FKGIAgKCKJtWZJFVZQsghJCEJBAI2ffM7w9/5islkUwyMydz5v26Li90zvOc5x4S88nMZ85ztHjxYjVv3lyzZs264LjFYlGfPn0knb9t4aZNm3TVVVcpJiZG/v7+at26tSZMmKCDBz3nzc2afDdniaoqKtVr+sTzHv/+H8tUVV6hseteVOaW/Upe4djN+zxNbGysXn31VcXGxhodxe1q+h5I37hHKau2avA/H9DA5+7WlmmvqTSnwMCUwMWlZRXq0PFcQ9Zev+2kx3yCc/7KH/SP/9TtTcu8gnKN+cM6FRaVuziV8agD1AGYw7qt6YasezqnRHuPeMaHwpLT8jVx+he/+IL25/7w1636erf5b0zvzXVAohbAPIyqA9KPrwk8gd1u1/UPrldaHa88WPHFCc1641vXhmokvLkWUAdgFt98f1oFBr2Hsd7NO0s0xJ/n7taqr1LrNPZERqFunLreY973+l+mbXQtXLhQVVVVmjx5cq2f0PD395d0fqMrJydHXbt21T/+8Q+tXbtWzz//vPbv369BgwYpLS3NLdldIf94ppJXbFbLIT3UfEBi9eP2yiplbU9SQESoflj8hYEJ3aOyslIFBQWqrPScT+I6S23fAzue/rdC2kcrfeMepW3YbWBCoG52GXjl0emckjq/UDSS3W7XS//e69CcjNNFbt3b2ijUAeoAPF9hUbkOJhvzgQfJ2DrkiNc+PKji0rr/rLPbpb8v2O/CRI2DN9cBiVoA89h10Lj7iew66Bl14IvtGdqT5Njf0z8/OKDiEvPfu9GbawF1AGZh5O/kew6eUWVl478CtrCoXP9a7NjFOzv2ZXvcbTt+YtpG14YNGyRJw4YNq3XMT42rnze6xo4dqzlz5mj8+PEaMmSIJk2apKVLlyo3N1dLly51bWgX+/6VpaqqrFSvRydUP9a8Xyd1uHm4Dr79qfo/c7t8AvwMTOh6R44c0fDhw3XkyBGjoxiipu+BiuJSFaScUs7BEwYmA+put4EvaiVp5/7G/8L2691Z2veD41cc/OsDz7vngKOoA9QBeL7vDp9VVZVxP6s8odFVXFKhtz8+7PA8T7zngKO8vQ5I1AJ4Prvdrt0GNps84fWA9OP9xBx1NrdUH36e7II0jYu31wLqAMzAyA8dFJVUGLbTkCMWrTmm3Pwyh+f9a7FnbmlusZv0Ha2YmBhlZmZq79696tq16wXHS0tL1aJFC+Xm5iotLU2tWrWq9VxnzpxRZGSk/vnPf+r3v/+9w1n69u2rzEzHtgHxtVv156r+Dq/lCFugv8ZueEkH3lytpH9/rqs/fkZn9h7T9qfedfhcM63bVW5xfyd73LhxDo0/deqUFi1apJtvvlnNmzev05wlS5bUJ1qDuOPr/3Ojl85U6rpd2v/6ynqfw6jvAXifnKDrVBTQt8ZjOxaNVXRk7fusR0cGyuZjVUVllTKzi39xnczsIvW7+cL/J8IKP1Fw6U7HQrtZfsAVygsaUa+5MWdnyyrP2sLQkVpQnzogmb8WUAfgSYp9E3U2ZGKNxy5WB6S614La6kBA2UFFFHzgWGg3K/OJ0enQe+s1t1n+Bwos96xt26kDzkEtgKewy6aTzZ6q9bizXhPUVgdkr1CrnAtvkdHYZIQ9qiprk4sP/B9BJbsUXlT/nwNG4L0h56AOwJOcDrldZb7tajzmjveGIvPelX/FcUciu93Z4BtU7N/T4XnWqlzFnHvZ+YFqUVVVpYyMDElSz549tWfPnnqdx+bMUI1JYeGPn0QsKSmp8fiCBQuUm5uriIiIGptclZWVqqqqUmpqqmbMmKGWLVtqwoQJNZzp4jIzM5We7tj+0X4WH6lFvZars34zb1NB6mklvfeZJOnrB/+psetf0olPtytzq2PblpzMOKkyu/sv+f7p61xXxcXF1X/Wda6jXztncMfX39mM+h6AF2pdKgXUfCg6MkitWwRf9BQ2H2udxtXkXG6Bzp0x7p4AddKiVKrnfZUzss5KFXnOzeNijtSC+tQBiVpQF9QBuE1oSymk5kN1rQNS/WtBSWmFIT8THBIUJIXWb+rZc0XSuUb+/P4HdaDxoBbALayBUrPaD7v8NYHF1vjrgCSF+dZrWlFJpYo84fn9DO8NNR7UAbhNXKVUy485d7w3lH3mnFTQyH9WxlZK/o5Pq7L7GlbnsrLqv22iaRtd0dHRys/P186dO9W37/mf/E9NTdUTTzwhSerevXuN84cMGaLNmzdLkuLj47Vx40ZFRkbWO4ujfO1WyYUfgGg1rKfaj71MK66cVv1YfkqWdv1loS77+31aMXyaKgprbhLWpGVMS0M+sREc7NgPo59+gQkMDKzz3F+62s9VXP31dwWjvgfgfXIC/VVUy7HM7NqO/MjRT+3UJCy0iYID3P9zwRH5AX6qb6sqpkUzWWt7B7mRcqQW1KcOSNSCuqAOwF2KfUN1tpZjF6sDkmOf5K9JgL9NEQb8THBEmU9Tna7n3GZhQQoMbtzP739RBxoPagHcwS6bTv7CcWe9Jqj1PPYKQ34mOCpDZaqqxzucQQE+CveA5/dzvDfUeFAH4C6nfa2qbVM+d7w3FBkRJv/Qxv2zMifAWuv7Z7/ER+WKduPPPLvdXn0bjZiYmHqfx7RbF95333167bXX1KZNG61fv14dO3aUJG3btk1TpkxRSkqKysvL9dBDD2nOnDkXzD906JDOnTun5ORkvfjiizp9+rS+/vprxcbGuiV/eVGJFsZNdstazjDp6AL5BtVyiYUL7dixw6HxSUlJuvXWWzV//nwlJCTUaU6/fv3qE61BPO3rLxn3PQDvM/O13Xr6tfpdxpy6bqJatwhWWlah2lxVv22nVv7jKl031D21oL627z2tAZMc325iYPcobV0w1gWJXMuRWlCfOiBRC+qCOgB3+eb7Uxo4+ZN6z29oLfjdrxM094+X1Xt9dygrr1Sbqz7QqbN1/+CaJPnarEpbN1HNIwJdlMw1qAONB7UA7mC32xVxxQLl5Dl+3xGp4XWgfasQHVvz63qt7U6THv9S//n0qMPzFj0/VBOvjnNBItfhvaHGgzoAd7nlsS+0aM2xes11xntDSStuUqf2YfWa6y7vf3JEtz75lcPzbhvbQe/NHuyCRK5lNTqAqzz++OMKDw9XamqqunTpom7duik+Pl6DBg1SYmKiLr/8cklSjx49apzfqVMnDRgwQBMnTtSGDRuUl5enF154wZ1PAS4QHx+vzz//XPHx8UZHAVBPvRPrd3Wts/TpHGHo+nXRr2uk+nR2/O/pvgmJLkjTuFAHAM/XvWMz+fhYDFu/Pj9f3c3P10d339TJ4XnjR7b3uCaXo6gDgOezWCyGvibwhNcDUv1+t2/eLEA3jmjn/DCNDLUA8HxG/k4eEuyrDm3ruU+4G40f2V4RYY5f2eup7w2ZttEVGxurTZs26ZprrlFAQIBSUlIUFRWlefPmacWKFUpKSpJUe6Pr58LCwhQfH68ffvjB1bHhYjabTeHh4bLZTLtrJ2B6Rr6wjI4MVMvm9du/2Z0sFoum397NoTntWjbR+JHtXZSo8aAOAJ4vMMCmLnHhhq3vCY0uSbp3fKKaNqn7/VlsPhY9PKWLCxM1DtQBwByMfE3gKXXg0p7NdVkvx27wNPXWrvLz9XFRosaDWgB4PiN/FvdKiJDVatwH7+oqwN+mhyY79vv9kL7R6tfVM+rc/zJto0uSunTpotWrVys/P195eXnaunWr7rzzTuXk5CgjI0M2m02dO3e+6HlOnTqlQ4cOKS7Osy7dxoXS0tI0bdo0paWlGR0FQD21bB6sbh2MeYNz1KWtDVm3Pn496hI9dU/POo2NCg/Qp3NHKcDf/C/0qAOAOYy+zJj98Fs2D1LXeOOabI5oHR2sZS+PUID/xd+wtFoteueZK9S3S5QbkhmLOgCYw+jLjPu9fNSljfueLD+xWCxa+vKV6tC2aZ3GTxkTr0d/U/N97M2GWgB4vgHdohQW4mfI2kbWIEfNuKunbr76kjqN7dQuVB+9NFwWS+Nv4tXE1I2u2uzdu1eSlJCQIH//8y/fmzx5sp5++mktX75cX375pd566y0NHTpUNptNDz/8sBFx4UQFBQXatGmTCgoKjI4CoAGMuoza0y7ffub3fTT3yUsVGV77HulX9G6hbQuvU+IlYe4LZiDqAGAO94xPkBGvv+4ZlyCbzXNeQl05sKW+mHeNusSF1TqmXcsm+njOlZpyXQf3BTMQdQAwh6H9YpTQ3v3bRg3sHqVeBm+l7ogWEYHaMv86jR0aW2vdbBJk05/u6aX3Zg/2iCsUnIFaAHi+wACbbr/e/b+/+tqsuvOGjm5ft76sVovef3aInry7h4IDa/5ws8Ui3XBlW22eP0ZRzTx3G3Pzf3S7Bj81umratnDgwIGaP3++XnnlFZWUlKhNmzYaNmyYZsyYobZt27o7KgCgBpOujdP0OTuUX1jutjX7dI70yMu3fzchUXfc0FFL1iXro7XJOp1TIn9fHyVeEqZ7xieoe8dmRkcEAIdd0rqprr68tT7d5L5PYttsFt11o+e8qP3JwB7NtXfZjfp6d5be/viwjp/MV2WlXS2bB+nW6+I1+rLW8vHxnOYdAEg/Xq1034REPfDcNreu62kffJOkyPAArfjHVTqWlqc3PkrSrgNnVFRSobAQP40Z3EZTrotXSLAxV0UAQEPcOz5Rc97f79Y1PfGetj4+Vs3+Q19Nv7273l/1g1Z/lapz+WUKDrSpb5dI3TMuQe1ahRgds8FodP2P+++/X/fff7+7IzVI/1l3KHZUXzVp01zLhz6sc4dSLxgTPaiLRiycobyjJ6sfW33dk6osKVPMFd3U70+3Vj8eEBmq4tPn9MnI6W7JD+fpMXW8ej06odbvgw43D1e3+2+QrBZlfL1X2x5/S/bKKr4H4HFCgv30x7t76rG/73Dbmn99sK/HXr7t7+ejSdfGa9K13GzZrBr6u0BYQqwG/vUuBUaGqqq8Qqd2HdY3T76tqrIKdz4NwCGzft9Hn29JV2Wl3S3rPXBLF4+4T2NNLBaLrugTrSv6RBsdBS5Ul1og1f6aQJK6/G6s4n89VBarRblHT2rzQ/9SWV6RO58GUGd33tBJryzcr6Op+W5Zr0enZpowum7bPzVGl7Ruqucf7m90DLhQq2E91euxm2X1tamyuFRbpr+hnAMp541p0jpKl79yv5p1bae85EytGv3YeceDW0Vq4LN3qeklMZKkg+9+pqR31rjtOQCO6NguVHfe0FFvf3zYLesF+PvoT3W8RURj1LSJn34/sbN+P/Hit3LyRF7d6Ore3Rx7D6es3qp9c5frmhWzf3HcucNpFxQwScrYtFcrr3q0+r+vnP+EMjfvc3pOuFazbu0V1aejCtJO13i8SZvm6jV9olZe9ahKsnM1/L3H1OGWK3X4/XV8D8AjTb21q5auP67t+2r+nnem347rpKsGecZe/PBODf1doLK0TN/MeFs5B1NksVo1eO6D6nLvWO39xzJXRQYarHfnSD1xZw/NfvNbl6/VsW2oZt/fx+XrAA1Rl1rwS68JYgZ3V/z4IVo9ZoYqCkt+/BDd47fomxnz3PgsgLoLCrTp3WcGa8gdq2V38WcebDaL3ps1WH6+F7/vIWAEv9BgXfHPB7Tm+qeUeyRdLQZ11uB/PagVw6aeN66soFi7n18kv5Ag9Xx0wgXnGfbOo9r76nKlrNoqSQqMCnNHfKDe/vbIAH2+JV1pWYUuX2v2/X3UqX2Yy9dB/XjlHhVbtmyR3W7XqFGjjI7iFFnbDqoo46xTzhXYIlwxl3fV0SX/dcr5GpuoqCg9+OCDiooy1422rX42DXz2Lm17/C3V9ht+2zEDlbJmu0qycyVJh+av1SXXX37BOLN/D8A8bDar3ps9uNY9hmuSmV2ktKxCZWbX/ZPJcW1C9OJUPvloFmatAw39XSA/OVM5B3/8tKe9qkrZ3x5VSGxzZ8UDXOape3qqd2KEQ3McrQW+Nqvem32FAgO88jOCpmPWOiDVrRb80muCZp3bKeubg6ooLJEkpa3fpbhxg10bGmigK/pEa9qt3RyaU5/XBE/f21s9ExyrN2i8zFgLQtpFqyQ7V7lH0iVJWVsPqEnrKDXr1v68cWXnCnRqe5IqikovOEfMFd1UWVJW3eSSpOLT51yaG2io0BA/vfPMFQ7dY7A+dWBwn2g9NLlLfSLCTXi15kVC41rqurUvyF5ZpSMffKFD//78gjHxvx6qtI17VHImz4CErhcREaFJkyYZHcPpej06UceWblJB6qlaxzRpFanCn13tVZB2WsGtLrzfkNm/B2AuiZeEaeU/rtI1v1+r0rLKi47vd/NKh84fExWktW+MVtMm7FlvFmatA3VVl98FbIH+6jjpSu2cvcCAhIBj/Hx99OncUbriN6t0JKVuv7s4UgusVov+89xQDerRor4R0ch4ex34pdcEZ747qk5TrlJAZKhKsnN1yU2D5RcSJL+wJio7V2BUZOCinnuor9KyCvXBZ8fqNN7R1wT3jE/QjLsvvPUFPJcZa0HesQwFRDRVVO8OOr37iGKv7i/fJoFq0qa5zu5NrtM5wjq2UcmZPA157WE1jWupgrRT2vH0v1Vwovb3moDG4KpBrfTWny/XnX/eVKfxjtaB7h2bafkrI7ivbSPHV8dLnNl7TB/2vkefjJyujXe8oE63jlS76wZdMC5+wjAd+c8GAxK6R15entavX6+8PPM0caL6dFRkzzglvffZRcfaf3a1V22fczD79wDMZ/iAllozd6RCgn2det72rUK06b1rdUnrpk49L4xlxjpQV3X5XcBi89GQN6Yq/cvvlPq5++6BBzREi4hA/feda9WtQ7hTz+vna9WSvw3XuJHtLz4YHsOb68BPantNkLl1v/a/8YlGLJiha1c9q+KsHElSVTn3a0Tj5uNj1fvPDtHt13dw+rkfnNRFc5+81GPv1YuambEWlOcX6Yu7XlKfJydrzOfPK/rSLso5lOrQz3CrzUctr+iu7+Z8pE9GPqr0jXs05PWHXZgacJ47buiof88eLB8f5/68HtAtSl+8fY3Cm/o79bxwPhpdXqK8oFjl+T9ejlmUcVbJy79WiwGJ541pMTBRtkB/nfzyOyMiusXJkyc1Y8YMnTx50ugoThM9qLNC41tp3Pa5Grd9roJiInTVoj+q1fBe540rSM9Wkzb/tw1VcOsoFaZnnzfGG74HYE7D+rfUtx9dr6H9Ypxyvt/8qoN2L/6V4trQ5DIbM9aBurrY7wIWm4+GvjlVxadytP2pd4yKCdRLTFSQti0YqwcndZEz3ovs2yVSuz74lW64sl3DT4ZGxZvrgHTx1wSH5q/VqtGPafWYGcradkCF6dnVWxkCjZnNZtXbM6/Qe7MGKzSk4bsxRIYH6MOXhuvvjw10aDsseAaz1oKsbQf12U1/1qpRj2nnrPcV1CK8eivDuihIO60z+5J17nCaJOno0q8U0f0SWay8fQzPcOvYDtq24Dp1iQtr8Ll8fCyacVcP/ffda9UslCaXJ+AnlZcIbB6mn17124ID1HpEH53Zd/y8MR0mDtcPH34he1WV+wOi3vb+c7k+7PVbLel/n5b0v09FGWe07ubZSt+457xxKau3qe3V/RUQGSpJ6nTrSCWv2HzeGL4H4Mkuad1UG966Wv+aMUgxUUH1OkfnuDCt/tdIvTtrsML4tA5M5pd+F7D4WDXk9YdVmlOgLY+8blxIoAGCAm36+2MD9d93rlWfzhduz1wXEWH++uuDfbX1/evUtUMzJycEjHex1wSBzcMkST6Bfuo5faL2zl1hREygXiwWi277VQftX3ajfj2qfb0aVDabRbdeF68DH9+o8VzRCw/z089wSerx8DhlbN6n/OOZdZ6fvnGPgqKbKSj6x9+BWg3rpXNJqbxHBI/St0uUdi2+Xn+6p1e9P/hwac/m2rbgOv3lgb7y9/NxckK4CvfoMoEBz96l2FH9FNg8TKM+/JPKC0u07NI/6NKX7lXq2p1KXbtTba8dqE63jZK9olIWm49SPtmqHz7YWH0OW3CA2l47UCuunGbgM4Gz/fx7oODEKe156UNds3K2LFarMr7eq8M/26KQ7wGYgdVq0X0TO+vumxK04osUvfbhQX29J0tl5bX/Yt4kyFdXX95av5+YqMF9otmWBB6pob8LtP/VZWp37UCd3X9cY9e9KEnK2nFI38yYZ+TTAurlij7R2rForHbsy9bcxQe1/IsU5eaX1TreZrOof9co3Ts+QeNHtleAPy+R4JnqUgsu9ppg5AdPSVarfHxtOrrkv0p6Z42Bzwion1YtgrX4xeFKzSzQW0sP6f1PftDxk798n7n42Ka6bWy87rqxk6Ij6/ehOcBovaZPVIsBibL4WHV612FtmTpX0vnvDVn9bLpp67/k42+Tb0iQxu96Q0eX/le7n/2PKopLte2JtzTi/Scki0VleYX66r6/G/ukgHrw9/PRzN/31vTbu2nRmmN6c2mSdh04o6oqe61zmoX666YR7XTfhET1TIhwY1o4i8X+8w260WiUF5VoYdxko2PU2aSjC+QbFOD2dXfscOz+IUlJSbr11ls1f/58JSQk1GlOv3796hOtQTzt6y8Z9z0AXExZeaX2HcnR7oNndDqnRGXllfL381Gb6GD16Rypjm1D2Y7EwzlSC+pTByRqQV1QB9BY2e12HUvL164D2TqWlq/Sskr52qxqFuqv3omR6t4xnOaWh6MONB7UAjRWp88Wa9eBM9p/NEeFxRWyWqXgQF91jQ9X78QIRYTxfevJeG+o8aAOoLEqKq7Qd4fPaM/BMzqXX6byiioF+Psovk1T9ekcqbYtm/DBZw/HKzp4FX9/f3Xq1En+/mxJBngLP18f9e4cqd713MYK5kIdALyPxWJRXJum3HcRkqgDgLeKahao0Ze31ujLWxsdBY0AtQDwPkGBNg3q0UKDerQwOgpchEYXvEr79u31/vvvGx0DAGAQ6gAAeDfqAACAWgAA5mM1OgAAAAAAAAAAAABQHzS64FUOHTqkyy67TIcOHTI6CgDAANQBAPBu1AEAALUAAMyHRhe8it1uV3l5uex2u9FRAAAGoA4AgHejDgAAqAUAYD7co6uRsgX6a9LRBUbHqDNbIDfwdCZP+/pLfA8AgLN5Wi2gDgCAc3laHZCoBQDgTNQBAKg7Gl2NlMVikW9QgNExYBC+/gAAagEAeDfqAAB4N+oAANQdWxcCAAAAAAAAAADAI3FFF7xKu3bttGjRIrVq1croKAAAA1AHAMC7UQcAANQCADAfGl3wKgEBAYqLizM6BgDAINQBAPBu1AEAALUAAMyHrQvhVTIyMjR79mxlZGQYHQUAYADqAAB4N+oAAIBaAADmQ6MLXiU3N1crV65Ubm6u0VEAAAagDgCAd6MOAACoBQBgPjS6AAAAAAAAAAAA4JFodAEAAAAAAAAAAMAj0egCAAAAAAAAAACAR6LRBa/SrFkz3XbbbWrWrJnRUQAABqAOAIB3ow4AAKgFAGA+Frvdbjc6BAAAAAAAAAAAAOAorugCAAAAAAAAAACAR6LRBQAAAAAAAAAAAI9EowsAAAAAAAAAAAAeiUYXAAAAAAAAAAAAPBKNLgAAAAAAAAAAAHgkGl0AAAAAAAAAAADwSDS6AAAAAAAAAAAA4JFodAEAAAAAAAAAAMAj0egCAAAAAAAAAACAR6LRBQAAAAAAAAAAAI9EowsAAAAAAAAAAAAeiUYXAAAAAAAAAAAAPBKNLgAAAAAAAAAAAHgkGl0AAAAAAAAAAADwSDS6AAAAAAAAAAAA4JFodAEAAAAAAAAAAMAj0egCAAAAAAAAAACAR6LRBQAAAAAAAAAAAI9EowsAAAAAAAAAAAAeiUYXAAAAAAAAAAAAPBKNLgAAAAAAAAAAAHgkGl0AAAAAAAAAAADwSDS6AAAAAAAAAAAA4JFodAEAAAAAAAAAAMAj0egCAAAAAAAAAACAR6LRBQAAAAAAAAAAAI9EowsAAAAAAAAAAAAeiUYXAAAAAAAAAAAAPBKNLgAAAAAAAAAAAHgkGl0AAAAAAAAAAADwSDS6AAAAAAAAAAAA4JFodAEAAAAAAAAAAMAj0egCAAAAAAAAAACAR6LRBQAAAAAAAAAAAI9EowsAAAAAAAAAAAAeiUYXAAAAAAAAAAAAPBKNLgAAAAAAAAAAAHgkGl0AAAAAAAAAAADwSDS6AAAAAAAAAAAA4JFodAEAAAAAAAAAAMAj0egCAAAAAAAAAACAR6LRBQAAAAAAAAAAAI9EowsAAAAAAAAAAAAeiUYXAAAAAAAAAAAAPBKNLgAAAAAAAAAAAHgkGl0AAAAAAAAAAADwSDS6AAAAAAAAAAAA4JFodAEAAAAAAAAAAMAj0egCAAAAAAAAAACAR6LRBQAAAAAAAAAAAI9EowsAAAAAAAAAAAAeiUYXAAAAAAAAAAAAPBKNLgAAAAAAAAAAAHgkGl0AAAAAAAAAAADwSDS6AAAAAAAAAAAA4JFodAEAAAAAAAAAAMAj0egCAAAAAAAAAACAR6LRBQAAAAAAAAAAAI9EowsAAAAAAAAAAAAeiUYXAAAAAAAAAAAAPBKNLgAAAAAAAAAAAHgkGl0AAAAAAAAAAADwSDS6AAAAAAAAAAAA4JFsRgdAw+zYscOh8dnZ2Vq2bJluvPFGRUZG1mlOv3796hMNAOAG1AEAAAAAAAB4M67o8jLZ2dmaN2+esrOzjY4CADAAdQAAAAAAAABmQqMLAAAAAAAAAAAAHolGFwAAAAAAAAAAADwSjS4AAAAAAAAAAAB4JBpdXiYkJESjR49WSEiI0VEAAAagDgAAAAAAAMBMLHa73W50CNTfjh07XL5Gv379XL4GAKB+qAMAAAAAAADwZlzR5WVKS0uVmpqq0tJSo6MAAAxAHQAAAAAAAICZ0OjyMsnJybrpppuUnJxsdBQAgAGoAwAAAAAAADATm9EBgJrY7XZVFHvW1Qa2QH9ZLBajYwAAAAAAAAAA4DVodKFRqigu1cK4yUbHcMikowvkGxRgdAwAAAAAAAAAALwGWxcCAAAAAAAAAADAI9HoAgAAAAAAAAAAgEdi60Ivk5CQoO3btxsdAwBgEOoAAAAAAAAAzIQrugAAAAAAAAAAAOCRaHR5mZSUFN1xxx1KSUkxOgoAwADUAQAAAAAAAJgJWxd6meLiYu3bt0/FxcVGRwEAlzubW6ot32Zp5/5s7T54RqdzilVWXiV/Px+1iQ5Wn8RI9e0SqUE9miswwDtKInUAAAAAAAAAZuId7+oBALyG3W7XN9+f1tzFB/Xh2mSVllXWOG7rd9KHnydLksKb+un26zvq3vEJ6tA21J1xAQAAAAAAADQAjS6YSvSgLhq9bOZ5j5UXFivvWIaOLvlKB9/+VPbKKoPSAXC11MwC/XbmZn22Oc2heTl5ZXp5/j69PH+f7h2foBem9lNIsJ+LUgIAAAAAAABwFhpdMKXk5ZuVun6nZLEoMCpM8eOHqP/M3yi0QyttffQNo+MBcIF3lx/WQy9sU15BeYPO8/pHSVqzOU3vPnOFhvVv6aR0AAAAAAAAAFzBanQAuFdMTIxmzpypmJgYo6O41Nn9yTq2dJOOLflK+19bqdXXzlBB+ml1vOVK+Uc0NToeACey2+168h87dcefNjW4yfWTlJMFGnnvZ1r06VGnnK8x8ZY6AAAAAAAAAO9Ao8vLhIaG6uqrr1ZoqHfdg6aiuFTZu3+QxWpV07YtjI4DwIn+9K/denbed04/b0WFXZNn/FdL1iY7/dxG8tY6AAAAAAAAAHOi0eVlcnJy9NFHHyknJ8foKG4X0u7HBldJTr7BSQA4y0drkzX7zW9ddv6qKrsmPfGlDhw1z89Mb64DAAAAAAAAMB8aXV4mKytLL774orKysoyO4lI+gf7ybxYi/4imCkuI1YBn71JEt0t0es8R5SdnGh0PgBOcPlus+/6yxaE5OxaNVeq6idqxaGyd55SVV+k3T32liooqRyM2St5SBwAAAAAAAOAdvKLRlZ2drUceeURxcXEKCAhQmzZtNG3aNBUXF2vy5MmyWCyaN2+e0THhRD2njtfN+9/Vzfve0fVfvKzE20crZc032njb80ZHM4zdbldhUbnOnCsxzRv28G4PPr9N2TklDs2JjgxS6xbBio4Mcmjejn3ZmvP+PofmAAAAAAAAAHA9m9EBXO3777/XyJEjlZWVpeDgYHXu3Fnp6el6+eWXdfz4cZ04cUKS1KNHD4OTwpkOL1yv5BWbZbX5KKxTrLr94XoFRoapoqSseszQN6fJbrfrv/e8XP2YX1gTXf/lHO18Zr6OLdtkRHSny84p0TsfH9brHyUpOf3HbRt9fCwaOzRW901I1JUDWspisRicEnBMclq+PvjsmFvXfPn9fXpwchf5+fq4dV0AAAAAAAAAtTP1FV3Z2dkaM2aMsrKyNH36dGVmZmr37t3KysrSrFmztGzZMn377beyWq3q2rWr0XHhRPnHM5Wxaa/Sv/hW+19fqQ1T/qrI3vEa9Nzd1WO2PvaGWgxIVPvrL6t+bOCzd+nU9iTTNLmWrT+u2FEf6LG/76hucklSZaVdH29I0VW//Uwj7l6jc3mlBqYEHPfGkiTZ7e5dMzO7WMs3prh3UQAAAAAAAAC/yNSNrgceeECpqamaOnWqnn/+eTVp0qT62JNPPqmEhARVVFSoY8eOCgwMNDCp+wQFBWnAgAEKCnJs2y5Pd3r3ER1bukmX3HiFonp3kCSV5hRo89S5GvCXuxTYIlxtrx2o6Eu7aOtjbxic1jmWrT+ucdM2qLik8hfHbdyeoZH3fqbConI3JQMaprKySm9/fNiQtd9aesiQdZ3JW+sAAAAAAAAAzMm0ja4DBw5o8eLFat68uWbNmnXBcYvFoj59+ki6cNvC5ORkjR07ViEhIQoPD9ett96qM2fOuCW3q8XGxurVV19VbGys0VHc7rs5S1RVUale0ydWP5a+cY9SVm3V4H8+oIHP3a0t015TaU6BgSmd42xuqabM+G+dr3jZsS9bf56727WhACc5nJLn8L25nGXrd6dUWenZ97jz5joAAAAAAAAA8zFto2vhwoWqqqrS5MmTa/3Uur+/v6TzG135+fkaNmyY0tLStGjRIr355pvatGmTxowZo6oqz35zU5IqKytVUFCgyspfvsrHjPKPZyp5xWa1HNJDzQckVj++4+l/K6R9tNI37lHaBnM0e95bcVhFJRUOzXln+WEVFTs2BzDCrgPZhq1dWFyhwyl5hq3vDN5cBwAAAAAAAGA+pm10bdiwQZI0bNiwWsekpaVJOr/R9eabbyo9PV3Lly/XmDFjNH78eC1cuFDbtm3TypUrHcpgt9tVWFiowsJC2d19M5laHDlyRMOHD9eRI0eMjmKI719ZqqrKSvV6dEL1YxXFpSpIOaWcgycMTOZcb3zk+PZqOXll+mhtsgvSAM71bZKxV9juOejZV/h6ex0AAAAAAACAuVjsjaUD42QxMTHKzMzU3r171bVr1wuOl5aWqkWLFsrNzVVaWppatWol6f8aY1988cV549u3b6/hw4fr7bffrnOGwsLC6vuCxcTEyGp1fl9x3LhxDo0/deqUFi1apJtvvlnNmzev05wlS5bUJ1qD+Nqt+nNVf7etN3rpTKWu26X9rzvWzPy5mdbtKrcYf9WfXVadbPbnes1tUvy1QovXOTkR4Fw5QWNVFNCnxmM7Fo1VdGTt956KjgyUzceqisoqZWYX/+I6mdlF6nfzhT8TQgtXq0npdsdCu5BZ6wAAAAAAAAC8R3R0tHbu3FmvuTYnZ2k0CgsLJUklJTXfx2XBggXKzc1VREREdZNL+vHeXuPHj79gfOfOnXXgwIF658nIyKj33F/y0/Osq+Li4uo/6zo3PT3d4VwN5WfxkVq4fdkGOZlxUmX2RrAVmMVPala/qQWFxSrIcP/XG3BI6xIpoOZD0ZFBat0i+KKnsPlY6zSuJrm5eco903j+PzFrHQAAAAAAAADqwrSNrujoaOXn52vnzp3q27fvecdSU1P1xBNPSJK6d+9+3rGcnByFhYVdcL7w8HD98MMP9c7jqiu6goMde6P2pzc1AwMD6zz3541Ad/G1WyXjL45ySMuYlo3kii7ppL1csvg6PDckyKqmBny9AUecC/RTbe2ZzOyiX5zr6BVdNQkLDVZwQOP5/8SsdQAAAAAAAADeIzo6ut5zTdvoGjFihI4cOaJnn31Ww4cPV8eOHSVJ27Zt05QpU3Tu3DlJ59+f6ycWi+WCxxq6w+ORI0ccfjOyLnbs2OHQ+KSkJC1atEhXX321EhIS6jTn73//ez2SNUx5UYkWxk1223qf3VS/rf5+7vCRw/INquUyEzebMuNLLVh11OF536ydq8RLwpwfCHCil97bq0dfrnnrwJq2Gvy51HUT1bpFsDKzi9Xmqg/qtf6yRa9pWP+W9ZrrCmatAwAAAAAAAEBdOP8So0bi8ccfV3h4uFJTU9WlSxd169ZN8fHxGjRokBITE3X55ZdLurDRFR4erpycnAvOd+7cOTVrVs/94BqR+Ph4ff7554qPjzc6Clzo9xM6OzxnWL8YmlzwCH06Rxi6fu/ESEPXbyjqAAAAAAAAAMzEtI2u2NhYbdq0Sddcc40CAgKUkpKiqKgozZs3TytWrFBSUpKkCxtdiYmJNd6L68CBA0pMTHRLdley2WwKDw+XzWbai/kgaUD3KP1qWGydx/varHrm971dmAhwHiMbTfGxTRUa4mfY+s5AHQAAAAAAAICZmLbRJUldunTR6tWrlZ+fr7y8PG3dulV33nmncnJylJGRIZvNps6dz7/yZcyYMfr666+Vmppa/di2bdt0/PhxXXfdde5+Ck6XlpamadOmKS0tzegocCGLxaKFfx2qof1iLjrW12bVwueG6vLe9d8DFXCn0BA/jbzUmHtGjb+qvSHrOhN1AAAAAAAAAGZi6kZXbfbu3StJSkhIkL+//3nHfvvb3yomJkbXX3+9Vq1apSVLluiWW25R//799atf/cqIuE5VUFCgTZs2qaCgwOgocLHgIF999tooPf27XoqODKxxzMhLW+mLt6/R+JGe/+Y9vMt9E9x/ha3FIt0zvpPb13U26gAAAAAAAADMxKsbXf+7baEkNW3aVBs3blRMTIwmTpyou+66S5deeqlWrVolq9Ur/7rgwfz9fPTn3/VWyucT9NFLwxUS9ONWZU2DfXX4k3H6/PXRuqxXC4NTAo679oo2atuyiVvXHDM4Vm1bhrh1TQAAAAAAAAC/zCtv0PFLjS5JiouL06pVq9wZCQ00bvtcVZaWq7KkTJL0/asf6/jKLTWO9fH31XVrX1BFcZlWjX6s+vEuvxur+F8PlcVqUe7Rk9r80L9Ullfklvyu5ufro3Ej2+uhF7Ypv6hCIcG+6tA21OhYQL3ZbFa9+vggjX1gnVvWCwzw0d8e6e+WtQAAAAAAAADUnVc3urp3725wEjjTF3e9pHOHUi86rtfjN+vUzsNq1qVd9WMxg7srfvwQrR4zQxWFJeoxdbx6PX6Lvpkxz4WJATTEdUNjNXlMnBasOurytZ79Q1+awwAAAAAAAEAj5JV78W3ZskV2u12jRo0yOorbRUVF6cEHH1RUVJTRUQzRfECimraP0bElX533eLPO7ZT1zUFVFJZIktLW71LcuMFGRATggFceG6QObZvWeXxmdpHSsgqVmV33qzVHX9ZaD0zqUp94jZK31wEAAAAAAACYi1de0eXNIiIiNGnSJKNjuMTgfz0oi9Wi07uPaNdf/6PSM3nnHbcF+qv/M7/RxtueV9P2MecdO/PdUXWacpUCIkNVkp2rS24aLL+QIPmFNVHZuQJ3Pg0ADmgW6q91b4zWkDs+VcrJi/+/2u/mlQ6d/4reLbTkb8NltVrqG7HRMXMdAAAAAAAAgPfxyiu6vFleXp7Wr1+vvLy8iw/2IGtu+JNWjnhEK0dOV2lOga545f4LxvT90xQlvfu5ijLPXnAsc+t+7X/jE41YMEPXrnpWxVk5kqSq8gqXZwfQMG1bhujr98aoR6dmTj3vr4bF6rPXRis4yNep5zWaWesAAAAAAAAAvBONLi9z8uRJzZgxQydPnjQ6ilMVpmdLkuwVlTrw1iq1GJB4wZjm/RPVc+o4jds+V0Nef0jhCbH61Zdzqo8fmr9Wq0Y/ptVjZihr2wEVpmdXb2UIoHFrHR2s7f8Zqz/+tqd8fBp29VXTJr5655kr9PHfRygo0HwXPpu1DgAAAAAAAMA7me8dPHgdW6C/rL4+Ksv78Z477W+4XGf2JV8wbuWV06r/PXpQF/X9861aNfqx6scCm4ep+NQ5+QT6qef0ido7d4XrwwNwGj9fH826v49uGN5Wf3nrO634MkWVlfY6zw8OtGnStXF66re91Do62IVJAQAAAAAAADgLjS54vICoUA2b96gsPlZZLFJ+yil9/YdXJUmXvnSvUtfuVOranRc9z8gPnpKsVvn42nR0yX+V9M4aV0cH4AK9O0dq6ZwrlZZZqHnLDumLHRnaffCMCorKLxgbFR6gPp0jdc0VrXXrdR0UGuJnQGIAAAAAAAAA9UWjCx6v4MQpfTLy0RqPbXnk9Rofz9y6/7yruSRpxfBpNY4F4JlaRwfr6ft662lJVVV2/XAiT5fd+omyz5UqMtxfuxdfr9YtgmWxNGyrQwAAAAAAAADGodHlZfz9/dWpUyf5+/sbHQUA3MZqtahju1D5+/lIkvx9fdQmuonBqYxBHQAAAAAAAICZ0OjyMu3bt9f7779vdAwAgEGoAwAAAAAAADATq9EBAAAAAAAAAAAAgPqg0eVlDh06pMsuu0yHDh0yOgoAwADUAQAAAAAAAJgJjS4vY7fbVV5eLrvdbnQUAIABqAMAAAAAAAAwE+7RhUbJFuivSUcXGB3DIbZAf6MjAAAAAAAAAADgVWh0oVGyWCzyDQowOgYAAAAAAAAAAGjE2LoQAAAAAAAAAAAAHokrurxMu3bttGjRIrVq1croKAAAA1AHAAAAAAAAYCY0urxMQECA4uLijI4BADAIdQAAAAAAAABmwtaFXiYjI0OzZ89WRkaG0VEAAAagDgAAAAAAAMBMaHR5mdzcXK1cuVK5ublGRwEAGIA6AAAAAAAAADOh0QUAAAAAAAAAAACPRKMLAAAAAAAAAAAAHolGFwAAAAAAAAAAADwSjS4vY7Va1atXL1mtfOkBwBtRBwAAAAAAAGAmvMvlZaqqqrRnzx5VVVUZHQUAYADqAAAAAAAAAMyERhcAAAAAAAAAAAA8Eo0uAAAAAAAAAAAAeCQaXQAAAAAAAAAAAPBINLq8TEhIiEaPHq2QkBCjowAADEAdAAAAAAAAgJnYjA4A92rVqpWeeeYZo2MAAAxCHQAAAAAAAICZcEWXlyktLVVqaqpKS0uNjgIAMAB1AAAAAAAAAGZCo8vLJCcn66abblJycrLRUQAABqAOAAAAAAAAwExodAEAAAAAAAAAAMAjcY8uNEp2u10VxZ61rZYt0F8Wi8XoGAAAAAAAAAAAeA0aXWiUKopLtTBustExHDLp6AL5BgUYHQMAAAAAAAAAAK/B1oUAAAAAAAAAAADwSFzR5WUSEhK0fft2o2MAAAxCHQAAAAAAAICZcEUXAAAAAAAAAAAAPBKNLi+TkpKiO+64QykpKUZHAQAYgDoAAAAAAAAAM6HR5WWKi4u1b98+FRcXGx0FblJVZdeRlFx9+PkxFRaXS5IKiyv05Y4M5RWUGZwOgLtRBwAAAAAAAGAm3KMLMKGKiiqt+uqE3lxySJu/zVJeQfl5x8/ll2nYnZ9Kkjq0baqbRrTTPeMS1K5ViBFxAQAAAAAAAACoFxpdgIlUVdn1rw8O6IV39yotq7BOc46k5Om5t7/X8+98rzGDY/Xi1H7q1D7MtUEBAAAAAAAAAHACGl0wlehBXTR62czzHisvLFbesQwdXfKVDr79qeyVVQalc60fTuTp9qe+0td7suo1326XPvnvCa3blq7Z9/fRQ5O7yMeH3U0BAAAAAAAAAI0XjS4vExMTo5kzZyomJsboKC6VvHyzUtfvlCwWBUaFKX78EPWf+RuFdmilrY++YXQ8p1v91QmNf2SjiksqG3yuktJKPfK37fp8S5qWzRmhJkG+TkgIoLHwljoAAAAAAAAA78DlGl4mNDRUV199tUJDQ42O4lJn9yfr2NJNOrbkK+1/baVWXztDBemn1fGWK+Uf0dToeE61fONxXf/Qeqc0uX5u3daTGnXvZyosKr/4YAAew1vqAAAAAAAAALwDjS4vk5OTo48++kg5OTlGR3GriuJSZe/+QRarVU3btjA6jtNs++6UJjz6hSoq7C45/5ZvT+nXj26U3e6a8wNwP2+tAwAAAAAAADAnGl1eJisrSy+++KKysup3HydPFtLuxwZXSU6+wUmco7ikQrf98SuVldf9nmM7Fo1V6rqJ2rFobJ3nfLopTW8uOVSfiAAaIW+uAwAAAAAAADAfr2h0ZWdn65FHHlFcXJwCAgLUpk0bTZs2TcXFxZo8ebIsFovmzZtndEw4kU+gv/ybhcg/oqnCEmI14Nm7FNHtEp3ec0T5yZlGx3OKP/5zlw6n5Do0JzoySK1bBCs6MsiheY/8bbuOp5ujQQgAAAAAAAAAMA+b0QFc7fvvv9fIkSOVlZWl4OBgde7cWenp6Xr55Zd1/PhxnThxQpLUo0cPg5PCmXpOHa+eU8ef91jKmm+07bG3DErkXCcyCvT3Bfvdtl5BUblmvr5H784a7LY1namsvFLL1h/X6k2pys0vV1Cgj/okRur26zsqMjzA6HgAAAAAAAAAgHoydaMrOztbY8aMUVZWlqZPn66nnnpKTZo0kSTNnj1bTz31lGw2m6xWq7p27WpwWjjT4YXrlbxis6w2H4V1ilW3P1yvwMgwVZSUVY8Z+uY02e12/feel6sf8wtrouu/nKOdz8zXsWWbjIheJ28uSVJVlXvvm/XBZ8f00rT+igjznMaQ3W7XPxbu11/f/l5ZZ4rPO7b4s2Q99a/dmnxtnP7+2EA1CfI1KCUAAAAAAAAAoL5MvXXhAw88oNTUVE2dOlXPP/98dZNLkp588kklJCSooqJCHTt2VGBgoIFJ3ScoKEgDBgxQUJBjW9d5mvzjmcrYtFfpX3yr/a+v1IYpf1Vk73gNeu7u6jFbH3tDLQYkqv31l1U/NvDZu3Rqe1KjbnKVlVfqraXuv2dWSWml3l1+xO3r1pfdbteDz2/TQy98c0GT6yelZZV6++PDGnrHp8rNL6txDGA23lIHAAAAAAAA4B1M2+g6cOCAFi9erObNm2vWrFkXHLdYLOrTp4+k87ctTEtL0wMPPKBLL71UQUFBslgsOn78uLtiu1xsbKxeffVVxcbGGh3FrU7vPqJjSzfpkhuvUFTvDpKk0pwCbZ46VwP+cpcCW4Sr7bUDFX1pF2197A2D0/6ynfuzdepsiSFrr96Uasi69fGvDw7q1f8cqNPYXQeydcvjX7g4EdA4eGsdAAAAAAAAgDmZttG1cOFCVVVVafLkybV+at3f31/S+Y2uH374QYsXL1ZYWJiGDBnilqzuVFlZqYKCAlVWVhodxe2+m7NEVRWV6jV9YvVj6Rv3KGXVVg3+5wMa+Nzd2jLtNZXmFBiY8uJ2Hcg2bO3dB7PdvmVifZSXV+mvb3/n0JxPN6Vpz0Hj/m4Bd/HmOgAAAAAAAADzMW2ja8OGDZKkYcOG1TomLS1N0vmNrsGDBysrK0uffvqpJkyY4NqQBjhy5IiGDx+uI0c8Zws6Z8k/nqnkFZvVckgPNR+QWP34jqf/rZD20UrfuEdpG3YbmLBudh04Y9jaeQXlOpqaZ9j6dbXyyxSdPFXk8LzXPkxyQRqgcfHmOgAAAAAAAADzMW2jKyUlRZLUrl27Go+Xlpbqm2++kXR+o8tqNe1fCSR9/8pSVVVWqtej/9fErCguVUHKKeUcPGFgsro7kWHsFWcnMgoNXb8uPtucVq95a76u3zwAAAAAAAAAgDFsRgdwlcLCH9+MLymp+V5GCxYsUG5uriIiItSqVSuX5+nQoYNLmmjjxo1zaPypU6ckSWvWrNGuXbvqNOeGG25wOFdD+dqt+rP6Ozwvc+t+vRdT+99J7pF0zW/tmiv1OnboqHJLlUvO/XOnQ+6UfGu+t86ORWMVHVnzVp2SFB0ZWP1n6rqJtY6TpMzsIvW7eeUFj0+4eZICyhv3lSBng8dL/l0dnpeeeVatW7d2QSI0FhlhUyVrqDIyM0zztTZrHQAAAAAAAID3iI6O1s6dO+s117SNrujoaOXn52vnzp3q27fvecdSU1P1xBNPSJK6d+/uljwZGRkuOe9PDb26Ki4urv6zrnPT09MdztVQfhYfqYXbl22QkxknVWZ3wz1v2hdJvjUfio4MUusWwRc9hc3HWqdxNTlzOksqdP/3hENa50r+jk+zVxQb8v0ONwqplKxSVWWlab7WZq0DAAAAAAAAQF2YttE1YsQIHTlyRM8++6yGDx+ujh07SpK2bdumKVOm6Ny5c5LO37bQlWJiYlxyRVdwsGPNip/e1AwMDKzzXHdc8fa/fO1WyfUXRzlVy5iWbrmi64xfpWq+TvHHq7B+SXRkoGw+VlVUVikzu/gXx9Z2rqiIIPmFuf97whEFfjnKrce8AHumIgz4fof7ZPj4qEqS1cdHMSb5Wpu1DgAAAAAAAMB7REdH13uuxW63252YpdE4ceKEevbsqZycHNlsNiUkJKi4uFhHjx7Vddddp4KCAn3xxRd699139Zvf/KbGc7z33nu6/fbblZycXOu9vn5JYWGhmjRpIkkqKChw+M3IutixY4dD4ysqKpSfn6+QkBDZbHXrc/br168+0RqkvKhEC+Mmu33dhph0dIF8gwJcvs5zb3+nJ16p3yWcqesmqnWLYKVlFarNVR84PN/fz0d5W6fIz9enXuu7S15BmVqNWKSCogqH5q2ZO0qjLzfHdnaoWesRi5R+qkitmgcpbf3NRsdxCrPWAQAAAAAAAKAunH+JUSMRGxurTZs26ZprrlFAQIBSUlIUFRWlefPmacWKFUpKSpLkviu6Ggubzabw8PA6v7mJxqdP50jD1u7eMbzRN7kkqWkTP915QyeH5iReEqaRl3LVCsyPOgAAAAAAAAAzMW2jS5K6dOmi1atXKz8/X3l5edq6davuvPNO5eTkKCMjQzabTZ07dzY6plulpaVp2rRpSktLMzoK6mlAtygFBhjTbBrWL8aQdevjuYf6anCful3uGhkeoOV/HyGr1eLiVIDxqAMAAAAAAAAwE1M3umqzd+9eSVJCQoL8/f0vOL5kyRItWbJEO3f+uD3cmjVrzvtvT1ZQUKBNmzapoKDA6Ciop6ZN/HTL1XFuX9dikX47LsHt69ZXgL9Na+aO0oTR7X9xXJe4MG3+9xh1bBfqpmSAsagDAAAAAAAAMBOv3Lfop0ZXbdsWjh8//rz/vu+++yRJt912m9577z2XZgPq4r4JiXr748NuXXP0Za0V16apW9dsqKBAmz54Ybhm/u6cXv8oSas3peroiTxV2aUAfx+tenWkhg+IkcXClVwAAAAAAAAA4IlodNXAbre7Mw4ayMffV0Nef1ihHVqrsrhUxafPaev0N1WQdvq8cdGDumjEwhnKO3qy+rHV1z2pypIy2YICNOztRxTR/RJJ0gdd7nDrc3BU786RGjs0Viu/POGW9SwW6Y+/7emWtVyhU/swzZk+UHOmD1TrEYuUfqpIEaH+unJgS6OjAQAAAAAAAAAawKsbXd27dzc4CZzl0PvrlL5xjyQp4fbRGvTSvVo3cdYF484dTtOq0Y9d8HhVRYX2/WuFSnPyNfLDP7k8rzO89sdL9dWuTJ3LL3P5Wg9P6apLe7Zw+ToAAAAAAAAAADjCK+/RtWXLFtntdo0aNcroKG4XFRWlBx98UFFRUUZHcZrK0vLqJpcknd59RCGxzR06R1VZhTK+3quy3EJnx3OZls2D9eoTgxyak5ldpLSsQmVmF9V5Tqd2oZp9fx9H4wFopMxYBwAAAAAAAOC9vPKKLm8WERGhSZMmGR3DpRLvvEap63bVeCw0rqWuW/uC7JVVOvLBFzr078/dnM65Jo+J1w8n8jTz9T0XHyyp380rHTp/y+ZBWjN3lAID+FEBmIU31AEAAAAAAAB4D9699jJ5eXnavn27+vfvr6ZNmxodx+m6PXCjmraP1uePvXHBsTN7j+nD3veoPL9IQTHNNGLBkyo9m6fjn2w1IKnz/Pl3veTjY9Gf/rXbqedt17KJ1r15tdq3DnHqeQEYy+x1AAAAAAAAAN7FK7cu9GYnT57UjBkzdPLkSaOjOF2Xe8eq7TUDtH7SX1RZfOF9q8oLilWe/+OWfUUZZ5W8/Gu1GJDo7phOZ7FY9NQ9vbRszpVq3izAKeccd1U7fbNwrOJjeRMcMBsz1wEAAAAAAAB4HxpdMIXO94xR+xsu09oJz6gsr+b7TwU2D5MsFkmSLThArUf00Zl9x90X0sVuuLKd9n98k265Ju6np+mwmKggLX5xmD7625VqHhHo3IAAAAAAAAAAADgZWxfC4wXFNFP/p3+jvOOZGr3kaUlSZVmFVl/7hC596V6lrt2p1LU71fbagep02yjZKyplsfko5ZOt+uGDjdXnuW7tCwpsHi6/0GCN3/WGMrfs06Y/vGrQs6qfyPAALXxuqGbf30dvLk3SvGWHlZ1TctF5Q/vF6L4Jibp+WFv5+tL/BgAAAAAAAAB4Bhpd8HhFGWf1Xsy4Go9teeT16n9PevczJb37Wa3n+WTkdKdnM0r71iH664P9NPv+Pjp0PFe7DmRrT9IZ5eSVqbyiSoH+Popr01R9u0Sqd2KkmoX6Gx0ZAAAAAAAAAACH0ejyMv7+/urUqZP8/WlseAMfH6s6x4Wrc1y4plzXweg4ABoB6gAAAAAAAADMhEaXl2nfvr3ef/99o2MAAAxCHQAAAAAAAICZcDMeAAAAAAAAAAAAeCQaXV7m0KFDuuyyy3To0CGjowAADEAdAAAAAAAAgJnQ6PIydrtd5eXlstvtRkcBABiAOgAAAAAAAAAzodEFAAAAAAAAAAAAj2QzOgBQE1ugvyYdXWB0DIfYAv2NjgAAAAAAAAAAgFeh0YVGyWKxyDcowOgYAAAAAAAAAACgEaPR5WXatWunRYsWqVWrVkZHAQAYgDoAAAAAAAAAM6HR5WUCAgIUFxdndAwAgEGoAwAAAAAAADATq9EB4F4ZGRmaPXu2MjIyjI4CADAAdQAAAAAAAABmQqPLy+Tm5mrlypXKzc01OgoAwADUAQAAAAAAAJgJjS4AAAAAAAAAAAB4JBpdAAAAAAAAAAAA8Eg0ugAAAAAAAAAAAOCRaHR5mWbNmum2225Ts2bNjI4CADAAdQAAAAAAAABmQqPLy1itVvn6+spq5UsPAN6IOgAAAAAAAAAz4V0uL5Odna158+YpOzvb6CgAAANQBwAAAAAAAGAmNLoAAAAAAAAAAADgkWh0AQAAAAAAAAAAwCPR6AIAAAAAAAAAAIBHotHlZUJCQjR69GiFhIQYHQUAYADqAAAAAAAAAMzEZnQAuFerVq30zDPPGB0DAGAQ6gAAAAAAAADMhCu6vExpaalSU1NVWlpqdBQAgAGoAwAAAAAAADATGl1eJjk5WTfddJOSk5ONjgIAMAB1AAAAAAAAAGbC1oVolOx2uyqKPetqA1ugvywWi9ExAAAAAAAAAADwGjS60ChVFJdqYdxko2M4ZNLRBfINCjA6BgAAAAAAAAAAXoOtCwEAAAAAAAAAAOCRaHQBAAAAAAAAAADAI7F1oZdJSEjQ9u3bjY4BADAIdQAAAAAAAABmwhVdAAAAAAAAAAAA8Eg0urxMSkqK7rjjDqWkpBgdBXC7ysoqVdntkiT7//8T8DbUAQAAAAAAAJgJWxd6meLiYu3bt0/FxcVGRwFc7tukM/rkvye060C2du7PVvqpoupjGdnFGnbnp+qTGKGh/WJ09eWt5eND7x/mRx0AAAAAAACAmdDoAmAqFRVV+uCzY5q7+KC2fneq1nF2u/Tljgx9uSNDf5u/T7ExwbpnXILuGZ+giLAANyYGAAAAAAAAANQXjS6YSvSgLhq9bOZ5j5UXFivvWIaOLvlKB9/+VPbKKoPSwdUOHM3Rb576Sjv2ZTs890RGoZ58dZf+vmC/XvvjpbrpqvYuSAgAAAAAAAAAcCYaXTCl5OWblbp+p2SxKDAqTPHjh6j/zN8otEMrbX30DaPjwQVeWbBP0+fsUFl5wxqZp3NKNG7aRt189SWa9/QVCgrkxyQAAAAAAAAANFbckMbLxMTEaObMmYqJiTE6ikud3Z+sY0s36diSr7T/tZVafe0MFaSfVsdbrpR/RFOj48GJ7Ha7nnhlhx564ZsGN7l+btGaYxp172fKLyxz2jmBxsBb6gAAAAAAAAC8A40uLxMaGqqrr75aoaGhRkdxq4riUmXv/kEWq1VN27YwOg6caNYb3+q5t793ybm/3pOlsQ+sV0lphUvODxjBW+sAAAAAAAAAzIlGl5fJycnRRx99pJycHKOjuF1Iux8bXCU5+QYngbNs/Oak/jx3t0vX+HJHhp5+bY9L1wDcyZvrAAAAAAAAAMyHRpeXycrK0osvvqisrCyjo7iUT6C//JuFyD+iqcISYjXg2bsU0e0Snd5zRPnJmUbHgxMUFJXrzj9vcnjejkVjlbpuonYsGlvnOS++t1fffH/K4bWAxshb6gAAAAAAAAC8g1c0urKzs/XII48oLi5OAQEBatOmjaZNm6bi4mJNnjxZFotF8+bNMzomnKjn1PG6ef+7unnfO7r+i5eVePtopaz5Rhtve97oaHCSma/t0fGTBQ7Pi44MUusWwYqODKrznKoqu+6e+bXsdrvD6zUWdrtd3x8+q1X/PaFPvjyhPQezPfr5AAAAAAAAAIAk2YwO4Grff/+9Ro4cqaysLAUHB6tz585KT0/Xyy+/rOPHj+vEiROSpB49ehicFM50eOF6Ja/YLKvNR2GdYtXtD9crMDJMFSVl1WOGvjlNdrtd/73n5erH/MKa6Pov52jnM/N1bJnjVwvBPQqLyvXWskNuXXPvkRx9sT1Dwwe0dOu6DVVWXqn5K3/Qax8e1O6DZ8471q1DuO6bkKjf/KqDAvxNXw4AAAAAAAAAmJCpr+jKzs7WmDFjlJWVpenTpyszM1O7d+9WVlaWZs2apWXLlunbb7+V1WpV165djY4LJ8o/nqmMTXuV/sW32v/6Sm2Y8ldF9o7XoOfurh6z9bE31GJAotpff1n1YwOfvUuntifR5Grk/vPpUeXml118oJPNXXzQ7Ws2RG5+mUbf+7nunvn1BU0u6cfm3e9mb9GVd6/R2dxSAxICAAAAAAAAQMOYutH1wAMPKDU1VVOnTtXzzz+vJk2aVB978sknlZCQoIqKCnXs2FGBgYEGJnWfoKAgDRgwQEFBdd+2zQxO7z6iY0s36ZIbr1BU7w6SpNKcAm2eOlcD/nKXAluEq+21AxV9aRdtfewNg9PiYt5f9YMh6y7/IkV5Be5vsNVHWXmlbnx4vb7YkXHRsVu+PaXr/rBWJaUVbkgGo3lrHQAAAAAAAIA5mbbRdeDAAS1evFjNmzfXrFmzLjhusVjUp08fSedvW7hkyRLddNNNatu2rYKCgpSQkKAnn3xSBQWO3wuoMYqNjdWrr76q2NhYo6O43XdzlqiqolK9pk+sfix94x6lrNqqwf98QAOfu1tbpr2m0hxzfK3NqrKySrsOXHh1knvWtmtPkjFrO2rRp8e0cfvFm1w/2fLtKb27/IgLE6Gx8OY6AAAAAAAAAPMxbaNr4cKFqqqq0uTJk2v91Lq/v7+k8xtdL730knx8fPTss89qzZo1uvfeezV37lyNGjVKVVVVbsnuSpWVlSooKFBlZaXRUdwu/3imkldsVsshPdR8QGL14zue/rdC2kcrfeMepW3YbWBC1EVScq6KSoy78mjn/mzD1nZEfbZZnLv4oOx2uwvSoDHx5joAAAAAAAAA8zFto2vDhg2SpGHDhtU6Ji0tTdL5ja5PPvlEH374oSZNmqQhQ4booYce0quvvqotW7bo66+/dm1oNzhy5IiGDx+uI0e888qN719ZqqrKSvV6dEL1YxXFpSpIOaWcgycMTIa62n80x6vXr4uk5HPavu+0w/P2/ZBT4728YC7eXgcAAAAAAABgLjajA7hKSkqKJKldu3Y1Hi8tLdU333wj6fxGV1RU1AVjf9riMD09vd55OnToIKvV+X3FcePGOTT+1KlTkqQ1a9Zo165ddZpzww03OJyroXztVv1Z/R2el7l1v96Lqf3vJPdIuua3nlDr8Ybo2KGjyi2ef9VfY1fo10tqcn2Nx3YsGqvoyF++71B0ZGD1n6nrJtY6LjO7SP1uXnnB4x8s/lhr37ml7oENUGK7RGp6W73mjrx2ogLLk5ycqPHICJsqWUOVkZmh1q1bGx3HKcxaBwAAAAAAAOA9oqOjtXPnznrNNW2jq7CwUJJUUlJS4/EFCxYoNzdXERERatWq1S+e66uvvpIkJSQk1DtPRkbd75XjiJ+eZ10VFxdX/1nXuQ1p8NWXn8VHauH2ZRvkZMZJldnZCszlwttJTWo+FB0ZpNYtgut0GpuPtc5jf664uNiQ/yccEtxUalq/qWfPnJHyG/nza4iQSskqVVVWNv6vYx2ZtQ4AAAAAAAAAdWHaRld0dLTy8/O1c+dO9e3b97xjqampeuKJJyRJ3bt3/8XzpKen609/+pNGjhypXr161TtPTEyMS67oCg527I36n97UDAwMrPPcizUCXcHXbpU87OKoljEtuaLLDYp9g3S2lmOZ2UUXnR8dGSibj1UVlVXKzC6udVxt5woK9FG4Af9POKLc6qtTkmS3SxZL3Sb9/7FR4Tb5NW3cz68hMnx8VCXJ6uOjmEb+dawrs9YBAAAAAAAAeI/o6Oh6z7XY7Xa7E7M0Gvfdd59ee+01tWnTRuvXr1fHjh0lSdu2bdOUKVOUkpKi8vJyPfTQQ5ozZ06N5ygoKNDQoUOVmZmp7du3q2XLlg5lKCwsVJMmTarP5eibkXWxY8cOh8YnJSXp1ltv1fz58+t8hVq/fv3qE61ByotKtDBustvXbYhJRxfINyjA6Bimd+BojrrcsKze81PXTVTrFsFKyypUm6s+cHj+3x7pr6m3dqv3+u4yaPJKbfvesft0de/YTN9+dL0sdW2OeaDWIxYp/VSRWjUPUtr6m42O4xRmrQMAAAAAAABAXTj/EqNG4vHHH1d4eLhSU1PVpUsXdevWTfHx8Ro0aJASExN1+eWXSzr//lw/V1JSol/96ldKTk7W2rVrHW5yNVbx8fH6/PPPFR8fb3QUoF46tQtVcKBxF6P26Rxp2NqOuG9CYr3mmLnJhR9RBwAAAAAAAGAmpm10xcbGatOmTbrmmmsUEBCglJQURUVFad68eVqxYoWSkpIk1dzoKi8v169//Wtt375da9asUefOnd0d32VsNpvCw8Nls5l210qYnI+P1bBmk81mUa+ECEPWdtTE0XEaeWndt5sb3Cdav/lVBxcmQmNBHQAAAAAAAICZmLbRJUldunTR6tWrlZ+fr7y8PG3dulV33nmncnJylJGRIZvNdkETq6qqSlOmTNG6deu0cuVK9e/f36D0rpGWlqZp06YpLS3N6ChAvd16nTFXotwwvJ2aNvEzZG1H+fpateRvw3XVoItfjTq4T7SWvzJC/n4+bkgGo1EHAAAAAAAAYCambnTVZu/evZKkhIQE+fv7n3fs/vvv1+LFizVt2jQFBgZq27Zt1f+Y4U3BgoICbdq0SQUFBUZHAert5qvjFBbi/oZTfbYDNFJIsJ9W/3OU/j17sAZ0i7rgeO/ECL098wqtfWO0wpv613AGmBF1AAAAAAAAAGbi1Y2umrYt/PTTTyVJf/nLXzRo0KDz/pk3b55bcwKoWVCgTfeMT3Drmj06NdOQvtFuXdMZfH2tunVsB21bOFb7P75REaE/NrSaNwvQzg9+pTtu6MiVXAAAAAAAAAA8llfeoOOXGl3Hjx93cxo4U4+p49Xr0QlaPvRhnTuUet6xqD4dNei5uyVJFl+bTm0/qG/++I6qyiokSV1+N1bxvx4qi9Wi3KMntfmhf6ksr8jtzwF189Rve+qjtck6lpbv8rV8fCx6e+YVslgsLl/LlTrHhSvA/8emlq/N6vHPBwAAAAAAAAC8+oqu7t27G5wEztSsW3tF9emogrTTNR4/e+C4Prn6ca286lGtGDZVARGh6jRlpCQpZnB3xY8fotVjZmj5kId1dt9x9Xr8FnfGh4OCg3z1zjNXODwvM7tIaVmFysyuexPz8Tu6q0/nSIfXAgAAAAAAAAC4llc2urZs2SK73a5Ro0YZHcXtoqKi9OCDDyoq6sL79Xgyq59NA5+9S9sef0uy22scU1lcJntFpSTJx88mnwA/2auqJEnNOrdT1jcHVVFYIklKW79LceMGuyc86m1I3xg9+0Bfh+b0u3ml2lz1gfrdvLJO40cMbKmn7ulVn3hAo2TWOgAAAAAAAADv5JWNLm8WERGhSZMmKSIiwugoTtXr0Yk6tnSTClJP/eK4Jq2jNHbdi5q4/x2V5xfp8IL1kqQz3x1Vy8E9FBAZKkm65KbB8gsJkl9YE5dnR8M8fmd3/fG3PV1y7iF9o/Xx30dwDyuYilnrAAAAAAAAALwTjS4vk5eXp/Xr1ysvL8/oKE4T1aejInvGKem9zy46tiDttFZe9agWd79bPv6+anvNAElS5tb92v/GJxqxYIauXfWsirNyJElV5RUuzY6Gs1gsmnV/H/1zxqDq+085w5Qx8Vozd5SaBPk67ZxAY2DGOgAAAAAAAADvRaPLy5w8eVIzZszQyZMnjY7iNNGDOis0vpXGbZ+rcdvnKigmQlct+qNaDa99u7mKohIlL9+sS278v3s8HZq/VqtGP6bVY2Yoa9sBFaZnV29liMbv9xM769sPr9egHs0bdJ7oyECteGWE5j87RIEBNielAxoPM9YBAAAAAAAAeC/exYXH2/vP5dr7z+XV/z1u+1ytn/JXnTuUet64kLYtVJCeLXtFpay+NsVeM0BnD6ZUHw9sHqbiU+fkE+inntMnau/cFe56CnCSTu3DtOm9a7Vk3XHNXXxQX+3KrPPc9q1C9LtfJ+iumzopvKm/C1MCAAAAAAAAAJyFRhdM7dKX7lXq2p1KXbtT0Zd2UeffjpG9skoWm48yv96r7+csqR478oOnJKtVPr42HV3yXyW9s8bA5KgvHx+rJoy+RBNGX6J9R85q1Vep2nUgW7sOnFFKRoGqquySpMjwAPVOjFDvxAgN7Rujqwa1ktVqMTg9AAAAAAAAAMARNLpgOkv631f971seeb36348s2qgjizbWOm/F8GkuzQX369qhmbp2aFb933a7XRUVdtlsFlksNLUAAAAAAAAAwNPR6PIy/v7+6tSpk/z92ZoN3sdiscjXlwYXvBt1AAAAAAAAAGZCo8vLtG/fXu+//77RMQAABqEOAAAAAAAAwEysRgcAAAAAAAAAAAAA6oNGl5c5dOiQLrvsMh06dMjoKAAAA1AHAAAAAAAAYCY0uryM3W5XeXm57Ha70VEAAAagDgAAAAAAAMBMuEcXGiVboL8mHV1gdAyH2AL9jY4AAAAAAAAAAIBXodGFRsliscg3KMDoGAAAAAAAAAAAoBFj60IAAAAAAAAAAAB4JK7o8jLt2rXTokWL1KpVK6OjAAAMQB0AAAAAAACAmdDo8jIBAQGKi4szOgYAwCDUAQAAAAAAAJgJWxd6mYyMDM2ePVsZGRlGRwEAGIA6AAAAAAAAADOh0eVlcnNztXLlSuXm5hodBQBgAOoAAAAAAAAAzIRGFwAAAAAAAAAAADwSjS4AAAAAAAAAAAB4JBpdAAAAAAAAAAAA8Eg0uryM1WpVr169ZLXypQcAb0QdAAAAAAAAgJnwLpeXqaqq0p49e1RVVWV0FACAAagDAAAAAAAAMBMaXQAAAAAAAAAAAPBINLoAAAAAAAAAAADgkWh0AQAAAAAAAAAAwCPR6PIyISEhGj16tEJCQoyOAgAwAHUAAAAAAAAAZmIzOgDcq1WrVnrmmWeMjgEAMAh1AAAAAAAAAGbCFV1eprS0VKmpqSotLTU6CgDAANQBAAAAAAAAmAmNLi+TnJysm266ScnJyUZHAQAYgDoAAAAAAAAAM2HrQjRKdrtdFcWedbWBLdBfFovF6BgAAAAAAAAAAHgNGl1olCqKS7UwbrLRMRwy6egC+QYFGB0DAAAAAAAAAACvwdaFAAAAAAAAAAAA8Eg0ugAAAAAAAAAAAOCR2LrQyyQkJGj79u1GxwAAGIQ6AAAAAAAAADPhii4AAAAAAAAAAAB4JBpdXiYlJUV33HGHUlJSjI4CADAAdQAAAAAAAABmwtaFXqa4uFj79u1TcXGx0VEAuEFBUbm+TTqjXQeytf/oOZ3NLZUk5eSVas77+9QnMUK9EiMUEuxncFK4C3UAAAAAAAAAZkKjCwBMxm63a/OeLM1dfFBL1h1XeUXVBWOKSio19cVvJEk+PhbdMLyt7puQqKH9YmSxWNwdGQAAAAAAAADqhUYXTCV6UBeNXjbzvMfKC4uVdyxDR5d8pYNvfyp75YVv+gNm8W3SGd0982vt3J9d5zmVlXYtWXdcS9YdV/eOzTTv6cvVr2uUC1MCAAAAAAAAgHPQ6IIpJS/frNT1OyWLRYFRYYofP0T9Z/5GoR1aaeujbxgdD3C68vIq/eWtb/WXed+qosJe7/N8f/isBk7+RNNv76anf9db/n4+TkwJAAAAAAAAAM5Fo8vLxMTEaObMmYqJiTE6ikud3Z+sY0s3Vf/3ofc+1/Wb/q6Ot1yp3c8tUumZPAPTAc5VVFyhm6Zu0Geb05xyvqoqu557+3tt+/60Vv5jBPfvMhlvqQMAAAAAAADwDlajA8C9QkNDdfXVVys0NNToKG5VUVyq7N0/yGK1qmnbFkbHAZymtKxSv3pwndOaXD/35Y4MXX3fWhUVVzj93DCOt9YBAAAAAAAAmBONLi+Tk5Ojjz76SDk5OUZHcbuQdj82uEpy8g1OAjjP/c9u0fptJ112/s17snTnnzddfCA8hjfXAQAAAAAAAJgPjS4vk5WVpRdffFFZWVlGR3Epn0B/+TcLkX9EU4UlxGrAs3cpotslOr3niPKTM42OBzjFmk2pmrfssENzdiwaq9R1E7Vj0dg6z/ngs2Naui7Z0XhopLylDgAAAAAAAMA7eEWjKzs7W4888oji4uIUEBCgNm3aaNq0aSouLtbkyZNlsVg0b948o2PCiXpOHa+b97+rm/e9o+u/eFmJt49WyppvtPG2542OBjhFXkGZ7p75tcPzoiOD1LpFsKIjgxya97vZW3TmXInD6wEAAAAAAACAK5m+0fX999+ra9eu+tvf/qasrCx17txZZWVlevnllzV58mQdOnRIktSjRw+Dk8KZDi9cr89/PVPrbpmtHTPnq+RsngIjw1RRUlY9Zuib0zTkjannzfMLa6Jff/uWLrnxCndHBhzyzseHlX6qyG3rnc4p0esfJrltPWey2+1avy1dNz68XhnZxZKkzOxi3fPM1/ru0BmD0wEAAAAAAABoCFM3urKzszVmzBhlZWVp+vTpyszM1O7du5WVlaVZs2Zp2bJl+vbbb2W1WtW1a1ej48KJ8o9nKmPTXqV/8a32v75SG6b8VZG94zXouburx2x97A21GJCo9tdfVv3YwGfv0qntSTq2jHsSofGqqrJr7ocH3b7uG0uSVFFR5fZ1G+Joap56jPtYV/32M328IUVVVXZJUmWVXW8uOaSe45drzP1rlZtfdpEzAQAAAAAAAGiMTN3oeuCBB5SamqqpU6fq+eefV5MmTaqPPfnkk0pISFBFRYU6duyowMBAA5O6T1BQkAYMGKCgIMe2LfN0p3cf0bGlm3TJjVcoqncHSVJpToE2T52rAX+5S4EtwtX22oGKvrSLtj72hsFpgV/25Y4MHUnJc/u6qZmF+nRTqtvXra9jaXm67NZV2nsk5xfHrf4qVVfevUYFReVuSmYsb60DAAAAAAAAMCfTNroOHDigxYsXq3nz5po1a9YFxy0Wi/r06SPp/G0LN23apKuuukoxMTHy9/dX69atNWHCBB086P6rJ1whNjZWr776qmJjY42O4nbfzVmiqopK9Zo+sfqx9I17lLJqqwb/8wENfO5ubZn2mkpzCgxMCVzcFzsyDFv7y53Gre0Iu92um6ZuUNaZ4jqN33UgWw8+v83FqRoHb64DAAAAAAAAMB/TNroWLlyoqqoqTZ48udZPrfv7+0s6v9GVk5Ojrl276h//+IfWrl2r559/Xvv379egQYOUlpbmluyuVFlZqYKCAlVWVhodxe3yj2cqecVmtRzSQ80HJFY/vuPpfyukfbTSN+5R2obdBiYE6mbXgWwD1/aMe1p9tStT3yaddWjOglU/KDunxEWJGg9vrgMAAAAAAAAwH9M2ujZs2CBJGjZsWK1jfmpc/bzRNXbsWM2ZM0fjx4/XkCFDNGnSJC1dulS5ublaunSpa0O7wZEjRzR8+HAdOXLE6CiG+P6VpaqqrFSvRydUP1ZRXKqClFPKOXjCwGRA3e0+aFyzaU/SGdntdsPWr6u5ix2/CresvErvfHzYBWkaF2+vAwAAAAAAADAXm9EBXCUlJUWS1K5duxqPl5aW6ptvvpF0fqOrJpGRkZIkm63+f10dOnSQ1er8vuK4ceMcGn/q1ClJ0po1a7Rr1646zbnhhhscztVQvnar/qz+Ds/L3Lpf78XU/neSeyRd81tPqPV4Q3Ts0FHlliqXnBv4iV1SVvifJUvNP092LBqr6Mja770UHRlY/Wfquom1jsvMLlK/m1de8Hh+Yblat2knixr31UCZoQ9KPs0cnvf08+/qH3/6yAWJXMesdQAAAAAAAADeIzo6Wjt37qzXXNM2ugoLCyVJJSU1b0O1YMEC5ebmKiIiQq1atbrgeGVlpaqqqpSamqoZM2aoZcuWmjCh/g2SjAzX3Nfmp+dZV8XFxdV/1nVuenq6w7kays/iI7Vw+7INcjLjpMrsjfvNf5iBRWpWe9M8OjJIrVsEX/QsNh9rncbV5GRGllRVWq+5bhNilXwcn1ZcWmXIz7yGMGsdAAAAAAAAAOrCtI2u6Oho5efna+fOnerbt+95x1JTU/XEE09Ikrp3717j/CFDhmjz5s2SpPj4eG3cuLH6yq66CgoKUkFBgS6//HKdOnVKFoulHs/klwUHO/ZG9U9vagYGBtZ5bk2NQFfztVslD7s4qmVMS67ogluk2yskS80/vjOzi35xbnRkoGw+VlVUVikzu7jWcb90npYxzWVp5P+DZlnLVVGPeYH+UjMDfuY1hFnrAAAAAAAAALxHdHR0veda7J5ws5V6uO+++/Taa6+pTZs2Wr9+vTp27ChJ2rZtm6ZMmaKUlBSVl5froYce0pw5cy6Yf+jQIZ07d07Jycl68cUXdfr0aX399deKjY1191P5RTt27HBofFJSkm699VbNnz9fCQkJdZrTr1+/+kRrkPKiEi2Mm+z2dRti0tEF8g0KMDoGvED8tR/qaGp+veamrpuo1i2ClZZVqDZXfeDw/OjIQGVsvKVea7vT1Be3ac77+x2et+CvQzTp2ngXJHIds9YBAAAAAAAAoC6cf9OoRuLxxx9XeHi4UlNT1aVLF3Xr1k3x8fEaNGiQEhMTdfnll0uq/f5cnTp10oABAzRx4kRt2LBBeXl5euGFF9z5FFwiPj5en3/+ueLjPeuNXAD/p09nx64uNcvajrh3fKLDcyLDAzTuqvYuSNO4UAcAAAAAAABgJqZtdMXGxmrTpk265pprFBAQoJSUFEVFRWnevHlasWKFkpKSJNXe6Pq5sLAwxcfH64cffnB1bJez2WwKDw+XzWbaXSsB0+uTaGCjy8C1HdGxXagmjr7EoTnTf9NN/n71uLGXh6EOAAAAAAAAwExM2+iSpC5dumj16tXKz89XXl6etm7dqjvvvFM5OTnKyMiQzWZT586dL3qeU6dO6dChQ4qLi3NDatdKS0vTtGnTlJaWZnQUAPU0Zkgbr1zbUW/PvEKD+9Rtb997xifokd90c3GixoE6AAAAAAAAADMxdaOrNnv37pUkJSQkyN/f/7xjkydP1tNPP63ly5fryy+/1FtvvaWhQ4fKZrPp4YcfNiKuUxUUFGjTpk0qKCgwOgqAeuocF66h/WLcvm6/rpHq1zXK7evWV1CgTZ+/PkoP3NJZwYE1X73UIiJQL03rr9f+eKksFoubExqDOgAAAAAAAAAz8cp9i35qdNW0beHAgQM1f/58vfLKKyopKVGbNm00bNgwzZgxQ23btnV3VACo0X0TEvXljgy3rvm7Xzt+3yujBfjb9MrjgzTr/j56f9UP+ub70yosrlBoiK9GX9Za1w9vKz9f829XCAAAAAAAAJgVja7/cf/99+v+++93dyQ42VUfPKXAqDCpqkrlhSX65sm3dXb/8fPGRA/qohELZyjv6Mnqx1Zf96QqS8rcGxaohxuvbKvLerXQ5j1Zblmvd2KEJl8b75a1XKFpEz/9fmJn/X6i0UkAAAAAAAAAOJNXN7q6d+9ucBK4yn9/+zeV5RVJkmJH99Nlc+7TJyOnXzDu3OE0rRr9mLvjAQ3m42PVOzOvUI/xH6uktNKla/narHp31mD5+nrlbrcAAAAAAAAAGjGvfNdyy5YtstvtGjVqlNFR3C4qKkoPPvigoqI85z479fFTk0uSfEOCZK+yG5gGcI2O7UL1t2n9HZqTmV2ktKxCZWYXXXzw/zf7/j7q3rGZo/HQSHlLHQAAAAAAAIB38MorurxZRESEJk2aZHQMt7j8H39QzKVdJEnrbpld45jQuJa6bu0LsldW6cgHX+jQvz93Z0Sgwe6b2FkZ2cWa/ea3dRrf7+aVDp3/4Sld9Ojt3eqRDI2VN9UBAAAAAAAAmB+NLi+Tl5en7du3q3///mratKnRcVzq6wdelSTFjR+ivn+6VesnP3ve8TN7j+nD3veoPL9IQTHNNGLBkyo9m6fjn2w1Ii5Qb8/8vrcC/X305Ku7nHreP/62p575fW9ZLBannhfG8qY6AAAAAAAAAPPzyq0LvdnJkyc1Y8YMnTx50ugobnP0o/8q+tIu8g9vct7j5QXFKs//cfu2ooyzSl7+tVoMSDQiItAgFotFM+7uqS/fuUaXtA5p8PnaRAdr7RujNev+PjS5TMgb6wAAAAAAAADMi0YXTMe3SaACW4RX/3fs1f1VmlOg0pyC88YFNg+T/v+b+LbgALUe0Udn9h13Y1LAuYb0jdH3S27Q9Nu7KTTEz+H5IcG+emhyF+1bdqOuGtTKBQkBAAAAAAAAwLnYuhCm49s0SMPmPSJbgJ/sVXaVnMnThlv/Kkm69KV7lbp2p1LX7lTbaweq022jZK+olMXmo5RPtuqHDzYanB5omOAgXz3/cH/96Z5e+uCzY3pvxRHtOpit4pLKGscH+PuoV0KEbr0uXpOujVNIsOMNMgAAAAAAAAAwCo0umE7RyTNafc0TNR7b8sjr1f+e9O5nSnr3M3fFAtwqOMhXd97YSXfe2EkVFVVKSj6nfT/kqLC4QlVVdgUH+qprh3Altg+Try8X9wIAAAAAAADwTDS6vIy/v786deokf39/o6MAcBObzaquHZqpa4dmRkdBI0AdAAAAAAAAgJnQ6PIy7du31/vvv290DACAQagDAAAAAAAAMBP2qwIAAAAAAAAAAIBHotHlZQ4dOqTLLrtMhw4dMjoKAMAA1AEAAAAAAACYCY0uL2O321VeXi673W50FACAAagDAAAAAAAAMBPu0YVGyRbor0lHFxgdwyG2QH+jIwAAAAAAAAAA4FVodKFRslgs8g0KMDoGAAAAAAAAAABoxNi6EAAAAAAAAAAAAB6JK7q8TLt27bRo0SK1atXK6CgAAANQBwAAAAAAAGAmNLq8TEBAgOLi4oyOAQAwCHUAAAAAAAAAZsLWhV4mIyNDs2fPVkZGhtFRAAAGoA4AAAAAAADATGh0eZnc3FytXLlSubm5RkcBABiAOgAAAAAAAAAzodEFAAAAAAAAAAAAj0SjCwAAAAAAAAAAAB6JRhcAAAAAAAAAAAA8Eo0uL9OsWTPddtttatasmdFRAAAGoA4AAAAAAADATCx2u91udAgAAAAAAAAAAADAUVzRBQAAAAAAAAAAAI9EowsAAAAAAAAAAAAeiUYXAAAAAAAAAAAAPBKNLgAAAAAAAAAAAHgkGl0AAAAAAAAAAADwSDS6AAAAAAAAAAAA4JFodAEAAAAAAAAAAMAj0egCAAAAAAAAAACAR6LRBQAAAAAAAAAAAI9EowsAAAAAAAAAAAAeiUYXAAAAAAAAAAAAPBKNLgAAAAAAAAAAAHgkGl0AAAAAAAAAAADwSDS6AAAAAAAAAAAA4JFodAEAAAAAAAAAAMAj/T+Vn+RBuiSQZAAAAABJRU5ErkJggg=="
      },
-     "execution_count": 12,
+     "execution_count": 27,
      "metadata": {},
      "output_type": "execute_result"
     }
    ],
-   "execution_count": 12
+   "execution_count": 27
   },
   {
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2025-06-25T06:54:12.305720Z",
-     "start_time": "2025-06-25T06:54:12.301733Z"
+     "end_time": "2025-06-26T02:23:29.186455Z",
+     "start_time": "2025-06-26T02:23:29.182520Z"
     }
    },
    "cell_type": "code",
@@ -609,18 +221523,26 @@
     "    def __init__(self):\n",
     "        super(QCCNN, self).__init__()\n",
     "        self.conv = nn.Sequential(\n",
-    "            ParameterizedQuantumConvolutionalLayer(nqubit=4, num_circuits=3),\n",
-    "            nn.BatchNorm2d(3),  # 添加批量归一化\n",
+    "            ParameterizedQuantumConvolutionalLayer(nqubit=4, num_circuits=3),  # 恢复为4量子比特\n",
+    "            nn.BatchNorm2d(3),  # 恢复原始通道数\n",
     "            nn.ReLU(),\n",
-    "            nn.MaxPool2d(kernel_size=2, stride=1)\n",
+    "            nn.MaxPool2d(kernel_size=2, stride=1),\n",
+    "            nn.Conv2d(3, 6, kernel_size=1),  # 添加1x1卷积层增强通道间信息交互\n",
+    "            nn.BatchNorm2d(6),\n",
+    "            nn.ReLU(),\n",
+    "            nn.AdaptiveMaxPool2d((9, 9))  # 确保输出固定尺寸\n",
     "        )\n",
     "\n",
     "        self.fc = nn.Sequential(\n",
-    "            nn.Linear(8 * 8 * 3, 128),\n",
-    "            nn.BatchNorm1d(128),  # 添加批量归一化\n",
-    "            nn.Dropout(0.5),  # 增加dropout比例\n",
+    "            # 根据新的特征图大小调整输入维度：6通道、9x9特征图 => 6*9*9=486\n",
+    "            nn.Linear(6 * 9 * 9, 1024),  # 修改为正确的输入维度\n",
+    "            nn.BatchNorm1d(1024),\n",
+    "            nn.Dropout(0.5),\n",
     "            nn.ReLU(),\n",
-    "            nn.Linear(128, 10)\n",
+    "            nn.Linear(1024, 512),\n",
+    "            nn.BatchNorm1d(512),\n",
+    "            nn.ReLU(),\n",
+    "            nn.Linear(512, 10)\n",
     "        )\n",
     "\n",
     "    def forward(self, x):\n",
@@ -629,45 +221551,172400 @@
     "        x = self.fc(x)\n",
     "        return x"
    ],
-   "id": "ae9e76ee6bca6e2f",
+   "id": "689b98ee8ef5a139",
    "outputs": [],
-   "execution_count": 13
+   "execution_count": 28
   },
   {
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2025-06-25T07:01:17.413980Z",
-     "start_time": "2025-06-25T06:54:12.401415Z"
+     "end_time": "2025-06-26T03:24:45.098535Z",
+     "start_time": "2025-06-26T02:23:29.191976Z"
     }
    },
    "cell_type": "code",
    "source": [
     "# 修改QCCNN模型的训练参数\n",
-    "num_epochs = 300\n",
+    "num_epochs = 100\n",
     "device = torch.device(\"cuda\" if torch.cuda.is_available() else \"cpu\")\n",
     "\n",
     "model = QCCNN()\n",
     "model.to(device)\n",
     "criterion = nn.CrossEntropyLoss()\n",
-    "optimizer = optim.AdamW(model.parameters(), lr=3e-4, weight_decay=1e-5)  # 使用AdamW优化器和适当的权重衰减\n",
+    "optimizer = optim.AdamW(model.parameters(), lr=5e-4, weight_decay=5e-5, amsgrad=True)  # 优化学习率和weight_decay参数\n",
     "optim_model, metrics = train_model(model, criterion, optimizer, train_loader, valid_loader, num_epochs, device)\n",
     "torch.save(optim_model.state_dict(), './data/notebook2/qccnn_weights.pt')   # 保存训练好的模型参数，用于后续的推理或测试\n",
     "pd.DataFrame(metrics).to_csv('./data/notebook2/qccnn_metrics.csv', index='None')   # 保存模型训练过程，用于后续图标展示"
    ],
-   "id": "81c62294cae7da16",
+   "id": "2433c2ae15fb60e6",
    "outputs": [
     {
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "Train loss: 0.531  Valid Acc: 0.784:  16%|█▌        | 48/300 [07:04<37:11,  8.85s/it]"
+      "  0%|          | 0/100 [00:00<?, ?it/s]"
      ]
     },
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Early stopping at epoch 49 due to no improvement in validation accuracy.\n"
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.821  Valid Acc: 0.761:   1%|          | 1/100 [00:57<1:34:31, 57.29s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.618  Valid Acc: 0.788:   2%|▏         | 2/100 [01:55<1:34:14, 57.70s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.563  Valid Acc: 0.788:   3%|▎         | 3/100 [02:55<1:34:58, 58.75s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.531  Valid Acc: 0.805:   4%|▍         | 4/100 [03:55<1:34:56, 59.34s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.508  Valid Acc: 0.802:   5%|▌         | 5/100 [04:52<1:32:40, 58.53s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.489  Valid Acc: 0.807:   6%|▌         | 6/100 [05:53<1:33:02, 59.39s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.475  Valid Acc: 0.812:   7%|▋         | 7/100 [06:54<1:32:42, 59.81s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.463  Valid Acc: 0.824:   8%|▊         | 8/100 [07:54<1:31:49, 59.88s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.456  Valid Acc: 0.819:   9%|▉         | 9/100 [08:52<1:30:05, 59.40s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.446  Valid Acc: 0.826:  10%|█         | 10/100 [09:52<1:29:06, 59.41s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.442  Valid Acc: 0.813:  11%|█         | 11/100 [10:52<1:28:35, 59.72s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.431  Valid Acc: 0.830:  12%|█▏        | 12/100 [11:54<1:28:23, 60.26s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.422  Valid Acc: 0.819:  12%|█▏        | 12/100 [12:54<1:28:23, 60.26s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.422  Valid Acc: 0.819:  13%|█▎        | 13/100 [12:54<1:27:14, 60.17s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.419  Valid Acc: 0.828:  14%|█▍        | 14/100 [13:49<1:24:06, 58.68s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.411  Valid Acc: 0.835:  15%|█▌        | 15/100 [14:47<1:23:04, 58.64s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.411  Valid Acc: 0.831:  16%|█▌        | 16/100 [15:49<1:23:15, 59.48s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.406  Valid Acc: 0.835:  17%|█▋        | 17/100 [16:48<1:22:22, 59.55s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.400  Valid Acc: 0.839:  18%|█▊        | 18/100 [17:50<1:22:04, 60.05s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.398  Valid Acc: 0.831:  18%|█▊        | 18/100 [18:50<1:22:04, 60.05s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.398  Valid Acc: 0.831:  19%|█▉        | 19/100 [18:50<1:21:13, 60.17s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.394  Valid Acc: 0.839:  19%|█▉        | 19/100 [19:50<1:21:13, 60.17s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.394  Valid Acc: 0.839:  20%|██        | 20/100 [19:50<1:20:11, 60.14s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.390  Valid Acc: 0.838:  21%|██        | 21/100 [20:51<1:19:22, 60.28s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.388  Valid Acc: 0.838:  22%|██▏       | 22/100 [21:51<1:18:22, 60.29s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.384  Valid Acc: 0.839:  23%|██▎       | 23/100 [22:51<1:17:13, 60.17s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.378  Valid Acc: 0.832:  24%|██▍       | 24/100 [23:53<1:16:44, 60.58s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.382  Valid Acc: 0.842:  25%|██▌       | 25/100 [24:53<1:15:38, 60.51s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.375  Valid Acc: 0.842:  26%|██▌       | 26/100 [25:52<1:14:15, 60.22s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.370  Valid Acc: 0.844:  26%|██▌       | 26/100 [26:53<1:14:15, 60.22s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.370  Valid Acc: 0.844:  27%|██▋       | 27/100 [26:53<1:13:25, 60.35s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.371  Valid Acc: 0.842:  28%|██▊       | 28/100 [27:53<1:12:08, 60.12s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.369  Valid Acc: 0.843:  29%|██▉       | 29/100 [28:53<1:11:21, 60.30s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.361  Valid Acc: 0.849:  29%|██▉       | 29/100 [29:54<1:11:21, 60.30s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.361  Valid Acc: 0.849:  30%|███       | 30/100 [29:54<1:10:28, 60.41s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.360  Valid Acc: 0.848:  31%|███       | 31/100 [30:53<1:08:56, 59.95s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.358  Valid Acc: 0.843:  32%|███▏      | 32/100 [31:54<1:08:26, 60.39s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.353  Valid Acc: 0.847:  32%|███▏      | 32/100 [32:55<1:08:26, 60.39s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.353  Valid Acc: 0.847:  33%|███▎      | 33/100 [32:55<1:07:23, 60.35s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.355  Valid Acc: 0.842:  34%|███▍      | 34/100 [33:56<1:06:39, 60.60s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.350  Valid Acc: 0.849:  35%|███▌      | 35/100 [34:56<1:05:35, 60.54s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.350  Valid Acc: 0.844:  36%|███▌      | 36/100 [35:58<1:04:53, 60.83s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.327  Valid Acc: 0.853:  37%|███▋      | 37/100 [36:58<1:03:40, 60.65s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.326  Valid Acc: 0.849:  38%|███▊      | 38/100 [37:57<1:02:20, 60.33s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.322  Valid Acc: 0.852:  38%|███▊      | 38/100 [38:58<1:02:20, 60.33s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.322  Valid Acc: 0.852:  39%|███▉      | 39/100 [38:58<1:01:27, 60.46s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.316  Valid Acc: 0.852:  40%|████      | 40/100 [39:57<59:56, 59.94s/it]  "
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.315  Valid Acc: 0.850:  41%|████      | 41/100 [40:56<58:36, 59.61s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.314  Valid Acc: 0.851:  42%|████▏     | 42/100 [41:57<57:56, 59.93s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.311  Valid Acc: 0.849:  43%|████▎     | 43/100 [42:56<56:43, 59.72s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.298  Valid Acc: 0.859:  44%|████▍     | 44/100 [43:56<55:50, 59.83s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.293  Valid Acc: 0.855:  45%|████▌     | 45/100 [44:57<55:10, 60.19s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.295  Valid Acc: 0.853:  46%|████▌     | 46/100 [45:58<54:24, 60.45s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.292  Valid Acc: 0.857:  47%|████▋     | 47/100 [46:58<53:25, 60.49s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.293  Valid Acc: 0.857:  48%|████▊     | 48/100 [48:00<52:48, 60.94s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.292  Valid Acc: 0.853:  49%|████▉     | 49/100 [49:02<51:53, 61.06s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.286  Valid Acc: 0.856:  50%|█████     | 50/100 [50:02<50:43, 60.86s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.281  Valid Acc: 0.861:  51%|█████     | 51/100 [51:03<49:45, 60.94s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.283  Valid Acc: 0.860:  52%|█████▏    | 52/100 [52:04<48:40, 60.84s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.281  Valid Acc: 0.854:  53%|█████▎    | 53/100 [53:05<47:38, 60.82s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.282  Valid Acc: 0.858:  54%|█████▍    | 54/100 [54:06<46:39, 60.86s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.279  Valid Acc: 0.858:  55%|█████▌    | 55/100 [55:06<45:33, 60.75s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.276  Valid Acc: 0.859:  56%|█████▌    | 56/100 [56:08<44:47, 61.09s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.278  Valid Acc: 0.856:  57%|█████▋    | 57/100 [57:09<43:47, 61.10s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.271  Valid Acc: 0.860:  58%|█████▊    | 58/100 [58:10<42:45, 61.07s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.274  Valid Acc: 0.861:  59%|█████▉    | 59/100 [59:13<42:10, 61.71s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.271  Valid Acc: 0.860:  60%|██████    | 60/100 [1:00:14<40:56, 61.42s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Train loss: 0.271  Valid Acc: 0.860:  60%|██████    | 60/100 [1:01:15<40:50, 61.26s/it]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Early stopping at epoch 61 due to no improvement in validation accuracy.\n"
      ]
     },
     {
@@ -678,13 +393955,13 @@
      ]
     }
    ],
-   "execution_count": 14
+   "execution_count": 29
   },
   {
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2025-06-25T07:01:22.540624Z",
-     "start_time": "2025-06-25T07:01:17.490806Z"
+     "end_time": "2025-06-26T03:24:51.526326Z",
+     "start_time": "2025-06-26T03:24:45.349899Z"
     }
    },
    "cell_type": "code",
@@ -702,17 +393979,485 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Test Acc: 0.797\n"
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Input shape: torch.Size([64, 1, 28, 28])\n",
+      "Unfolded shape: torch.Size([64, 1, 9, 9, 3, 3])\n",
+      "Reshaped shape: torch.Size([64, 3, 9, 9])\n",
+      "Test Acc: 0.852\n"
      ]
     }
    ],
-   "execution_count": 15
+   "execution_count": 30
   },
   {
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2025-06-25T07:01:22.666238Z",
-     "start_time": "2025-06-25T07:01:22.660994Z"
+     "end_time": "2025-06-26T03:27:54.738714Z",
+     "start_time": "2025-06-26T03:27:54.730708Z"
     }
    },
    "cell_type": "code",
@@ -732,7 +394477,7 @@
     "    vgg_block(1, 32, 2),  # 增加通道数和调整卷积层数量\n",
     "    vgg_block(32, 64, 2),\n",
     "    nn.Flatten(),\n",
-    "    nn.Linear(64 * 4 * 4, 256),  # 调整全连接层大小\n",
+    "    nn.Linear(64 * 7 * 7, 256),  # 调整全连接层大小\n",
     "    nn.BatchNorm1d(256),  # 添加批量归一化\n",
     "    nn.ReLU(),\n",
     "    nn.Dropout(0.5),  # 增加dropout比例\n",
@@ -746,19 +394491,19 @@
    ],
    "id": "f72e03c426bd658b",
    "outputs": [],
-   "execution_count": 16
+   "execution_count": 33
   },
   {
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2025-06-25T07:05:52.523565Z",
-     "start_time": "2025-06-25T07:01:22.782579Z"
+     "end_time": "2025-06-26T04:00:50.810478Z",
+     "start_time": "2025-06-26T03:27:58.648817Z"
     }
    },
    "cell_type": "code",
    "source": [
     "# 修改VGG模型的训练参数\n",
-    "num_epochs = 300\n",
+    "num_epochs = 100\n",
     "device = torch.device(\"cuda\" if torch.cuda.is_available() else \"cpu\")\n",
     "\n",
     "vgg_model = VGG\n",
@@ -775,31 +394520,17 @@
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "Train loss: 1.563  Valid Acc: 0.855:  28%|██▊       | 85/300 [04:29<11:22,  3.17s/it]"
-     ]
-    },
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "Early stopping at epoch 86 due to no improvement in validation accuracy.\n"
-     ]
-    },
-    {
-     "name": "stderr",
-     "output_type": "stream",
-     "text": [
-      "\n"
+      "Train loss: 1.510  Valid Acc: 0.925: 100%|██████████| 100/100 [32:52<00:00, 19.72s/it]\n"
      ]
     }
    ],
-   "execution_count": 17
+   "execution_count": 34
   },
   {
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2025-06-25T07:05:55.080402Z",
-     "start_time": "2025-06-25T07:05:52.668397Z"
+     "end_time": "2025-06-26T04:01:03.995494Z",
+     "start_time": "2025-06-26T04:01:01.239374Z"
     }
    },
    "cell_type": "code",
@@ -817,17 +394548,17 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Test Acc: 0.894\n"
+      "Test Acc: 0.920\n"
      ]
     }
    ],
-   "execution_count": 18
+   "execution_count": 35
   },
   {
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2025-06-25T07:05:55.325058Z",
-     "start_time": "2025-06-25T07:05:55.198007Z"
+     "end_time": "2025-06-26T04:01:05.563661Z",
+     "start_time": "2025-06-26T04:01:05.416700Z"
     }
    },
    "cell_type": "code",
@@ -878,19 +394609,19 @@
       "text/plain": [
        "<Figure size 1200x500 with 2 Axes>"
       ],
-      "image/png": "iVBORw0KGgoAAAANSUhEUgAAA+kAAAHUCAYAAABGRmklAAAAOnRFWHRTb2Z0d2FyZQBNYXRwbG90bGliIHZlcnNpb24zLjEwLjAsIGh0dHBzOi8vbWF0cGxvdGxpYi5vcmcvlHJYcgAAAAlwSFlzAAAPYQAAD2EBqD+naQABAABJREFUeJzs3XV4FMcbwPHvxT0hBgFiQBKcBIK7E9xdAoEfWrQUKQVatHix4gkUpxQpGijurgGCRJBAIO52+/tjyUGIQoAg83mee7jbnd2dvRy39+7MvKOQJElCEARBEARBEARBEIQ8p5bXFRAEQRAEQRAEQRAEQSaCdEEQBEEQBEEQBEH4QoggXRAEQRAEQRAEQRC+ECJIFwRBEARBEARBEIQvhAjSBUEQBEEQBEEQBOELIYJ0QRAEQRAEQRAEQfhCiCBdEARBEARBEARBEL4QIkgXBEEQBEEQBEEQhC+ECNIFQRAEQRAEQRAE4QshgnThu6ZQKHL0OHbsWK6OM3nyZBQKxQdte+zYsY9Sh9wc+++///7sx/4QN27coHfv3tjb26Ojo4OBgQHly5dn1qxZhIaG5nX1BEEQhBwQ1+ac2717NwqFAjMzMxISEvK0Ll+jyMhIpk2bhqurK0ZGRmhra2NnZ0efPn24cuVKXldP+I5p5HUFBCEvnT17Ns3rKVOmcPToUY4cOZJmecmSJXN1nL59+9KkSZMP2rZ8+fKcPXs213X41q1cuZJBgwbh5OTE6NGjKVmyJElJSVy6dIlly5Zx9uxZduzYkdfVFARBELIhrs05t3r1agBCQ0PZuXMnnTp1ytP6fE0ePnxIo0aNCA4OZsCAAfz6668YGBjg7+/P1q1bqVChAuHh4RgbG+d1VYXvkEKSJCmvKyEIXwp3d3f+/vtvoqOjsywXGxuLnp7eZ6pV3jl27Bh169Zl27ZttG/fPq+rk6mzZ89Ss2ZNGjZsyM6dO9HW1k6zPjExkQMHDtCyZctcHysuLg4dHZ0Pbn0RBEEQ3o+4Nmfs+fPnWFtbU6tWLc6cOUPNmjXx9vbO62pl6Ev726SkpODi4kJAQACnT5+mdOnS6crs37+f2rVr57rekiQRHx+Prq5urvYjfF9Ed3dByEadOnUoXbo0J06coFq1aujp6dGnTx8AtmzZQqNGjbCyskJXV5cSJUowduxYYmJi0uwjoy51dnZ2NG/enAMHDlC+fHl0dXUpXrw4a9asSVMuoy517u7uGBgY8ODBA5o2bYqBgQHW1taMGjUqXXe3J0+e0L59ewwNDTExMaFbt25cvHgRhUKBl5fXR3mPbt26RatWrciXLx86Ojo4Ozuzdu3aNGWUSiVTp07FyckJXV1dTExMKFu2LH/88YeqzMuXL/nf//6HtbU12traWFhYUL16dQ4fPpzl8adPn45CoWDFihXpAnQALS2tNAG6QqFg8uTJ6crZ2dnh7u6ueu3l5YVCocDb25s+ffpgYWGBnp4eW7ZsQaFQ8N9//6Xbx59//olCoeDGjRuqZZcuXaJly5aYmpqio6ODi4sLW7duzfKcBEEQhMyJazOsXbuW5ORkRowYQdu2bfnvv/8ICAhIVy48PJxRo0ZRpEgRtLW1sbS0pGnTpty9e1dVJiEhgd9++40SJUqgo6ODmZkZdevW5cyZMwD4+/tnWrd3r6mp7+uVK1do3749+fLlo2jRooB8PezcuTN2dnbo6upiZ2dHly5dMqz306dPVb8JtLS0KFiwIO3bt+fFixdER0djYmJC//79023n7++Puro6s2fPzvS927lzJzdv3mTcuHEZBugAbm5uqgDd3d0dOzu7dGUy+gwpFAqGDBnCsmXLKFGiBNra2qxatQpLS0t69OiRbh/h4eHo6uoycuRI1bLIyEh+/PFH7O3t0dLSolChQgwfPjzdZ1j4donu7oKQA0FBQXTv3p2ffvqJ6dOno6Ym39+6f/8+TZs2Zfjw4ejr63P37l1+//13Lly4kK5bXkauX7/OqFGjGDt2LPnz52fVqlV4eHhQrFgxatWqleW2SUlJtGzZEg8PD0aNGsWJEyeYMmUKxsbGTJw4EYCYmBjq1q1LaGgov//+O8WKFePAgQMftTvcvXv3qFatGpaWlixcuBAzMzPWr1+Pu7s7L1684KeffgJg1qxZTJ48mQkTJlCrVi2SkpK4e/cu4eHhqn316NGDK1euMG3aNBwdHQkPD+fKlSuEhIRkevyUlBSOHDlChQoVsLa2/mjn9bY+ffrQrFkz/vrrL2JiYmjevDmWlpZ4enpSv379NGW9vLwoX748ZcuWBeDo0aM0adKEypUrs2zZMoyNjdm8eTOdOnUiNjY2zU0BQRAEIee+92vzmjVrsLKyws3NDV1dXTZu3IiXlxeTJk1SlYmKiqJGjRr4+/szZswYKleuTHR0NCdOnCAoKIjixYuTnJyMm5sbJ0+eZPjw4dSrV4/k5GTOnTtHYGAg1apVe696pWrbti2dO3dmwIABquDS398fJycnOnfujKmpKUFBQfz5559UrFgRHx8fzM3NATlAr1ixIklJSYwfP56yZcsSEhLCwYMHCQsLI3/+/PTp04cVK1Ywa9asNF3Sly5dipaWluqmTUZSexy0bt36g84tOzt37uTkyZNMnDiRAgUKYGlpiZ+fH8uWLWPJkiUYGRmpym7atIn4+Hh69+4NyL0OateuzZMnT1Tnfvv2bSZOnMjNmzc5fPiw6M33PZAEQVDp1auXpK+vn2ZZ7dq1JUD677//stxWqVRKSUlJ0vHjxyVAun79umrdpEmTpHf/u9na2ko6OjpSQECAallcXJxkamoq9e/fX7Xs6NGjEiAdPXo0TT0BaevWrWn22bRpU8nJyUn1esmSJRIg7d+/P025/v37S4Dk6emZ5TmlHnvbtm2ZluncubOkra0tBQYGplnu5uYm6enpSeHh4ZIkSVLz5s0lZ2fnLI9nYGAgDR8+PMsy73r+/LkESJ07d87xNoA0adKkdMttbW2lXr16qV57enpKgNSzZ890ZUeOHCnp6uqqzk+SJMnHx0cCpEWLFqmWFS9eXHJxcZGSkpLSbN+8eXPJyspKSklJyXG9BUEQvkfi2pzeiRMnJEAaO3as6jzt7e0lW1tbSalUqsr99ttvEiAdOnQo032tW7dOAqSVK1dmWsbPzy/Tur17TU19XydOnJjteSQnJ0vR0dGSvr6+9Mcff6iW9+nTR9LU1JR8fHwy3fbhw4eSmpqaNH/+fNWyuLg4yczMTOrdu3eWx23SpIkESPHx8dnWUZLkv62trW265Rl9hgDJ2NhYCg0NTbP8xo0bEiCtWLEizfJKlSpJFSpUUL2eMWOGpKamJl28eDFNub///lsCpH379uWozsLXTXR3F4QcyJcvH/Xq1Uu3/NGjR3Tt2pUCBQqgrq6OpqYmtWvXBuDOnTvZ7tfZ2RkbGxvVax0dHRwdHTPs9vUuhUJBixYt0iwrW7Zsmm2PHz+OoaFhusQ4Xbp0yXb/OXXkyBHq16+frhXb3d2d2NhYVQKgSpUqcf36dQYNGsTBgweJjIxMt69KlSrh5eXF1KlTOXfuHElJSR+tnrnRrl27dMv69OlDXFwcW7ZsUS3z9PREW1ubrl27AvDgwQPu3r1Lt27dAEhOTlY9mjZtSlBQEPfu3fs8JyEIgvCN+Z6vzakJ41JbixUKBe7u7gQEBKQZirV//34cHR1p0KBBpvvav38/Ojo6WbY8f4iMrp3R0dGMGTOGYsWKoaGhgYaGBgYGBsTExKT52+zfv5+6detSokSJTPdfpEgRmjdvztKlS5Fep9jauHEjISEhDBky5KOey/uqV68e+fLlS7OsTJkyVKhQAU9PT9WyO3fucOHChTTv/Z49eyhdujTOzs5pfjc0btz4i5hRQPg8RJAuCDlgZWWVbll0dDQ1a9bk/PnzTJ06lWPHjnHx4kX++ecfQE4wlh0zM7N0y7S1tXO0rZ6eHjo6Oum2jY+PV70OCQkhf/786bbNaNmHCgkJyfD9KViwoGo9wLhx45gzZw7nzp3Dzc0NMzMz6tevz6VLl1TbbNmyhV69erFq1SqqVq2KqakpPXv25Pnz55ke39zcHD09Pfz8/D7aOb0ro/MrVaoUFStWVF1sU1JSWL9+Pa1atcLU1BSAFy9eAPDjjz+iqamZ5jFo0CAAXr169cnqLQiC8C37Xq/NUVFRbNu2jUqVKmFhYUF4eDjh4eG0adMGhUKhCuBBzvVSuHDhLPf38uVLChYsqBou8LFk9Pfp2rUrixcvpm/fvhw8eJALFy5w8eJFLCws0ry/Oak3wLBhw7h//z6HDh0CYMmSJVStWpXy5ctnuV3qTZhP9dsho3MH+abK2bNnVfkAUm/uv32D5sWLF9y4cSPd7wZDQ0MkSRK/G74TYky6IORARmN/jhw5wrNnzzh27JjqDj2QZox1XjMzM+PChQvplmcV9H7IMYKCgtItf/bsGYBqfJmGhgYjR45k5MiRhIeHc/jwYcaPH0/jxo15/Pgxenp6mJubs2DBAhYsWEBgYCC7d+9m7NixBAcHc+DAgQyPr66uTv369dm/fz9PnjzJ0UVdW1s7w/lkMxv7ntnYr969ezNo0CDu3LnDo0ePCAoKUo0pe/vcx40bR9u2bTPch5OTU7b1FQRBENL7Xq/NmzZtIjY2lgsXLqRrrQXYsWMHYWFh5MuXDwsLC548eZLl/iwsLDh16hRKpTLTQD31xsO7186scsa8+/eJiIhgz549TJo0ibFjx6qWJyQkEBoamq5O2dUb5Bbr0qVLs3jxYgwMDLhy5Qrr16/PdrvGjRuzYsUKdu7cmaYumdHR0cnwd0NmAXNmvxu6dOnCyJEj8fLyYtq0afz111+0bt06zd/R3NwcXV3ddMkK314vfPtES7ogfKDUL+B3s4kvX748L6qTodq1axMVFcX+/fvTLN+8efNHO0b9+vVVP4retm7dOvT09KhSpUq6bUxMTGjfvj2DBw8mNDQUf3//dGVsbGwYMmQIDRs25MqVK1nWYdy4cUiSRL9+/UhMTEy3PikpiX///Vf12s7OLk32dZB/2GU3vc+7unTpgo6ODl5eXnh5eVGoUCEaNWqkWu/k5ISDgwPXr1/H1dU1w4ehoeF7HVMQBEHI3PdwbV69ejWGhob8999/HD16NM1j9uzZJCQksGHDBkDOUO7r65tlwjw3Nzfi4+OzzCqfP39+dHR00l07d+3alaM6g/y3kSQp3d9m1apVpKSkpKvT0aNHczQkbOjQoezdu5dx48aRP39+OnTokO02rVq1okyZMsyYMYNbt25lWObgwYPExsYC8u+G4OBgVQ85kKd3PXjwYLbHelu+fPlo3bo169atY8+ePTx//jzdMIPmzZvz8OFDzMzMMvzdkFGWeeHbI1rSBeEDVatWjXz58jFgwAAmTZqEpqYmGzZs4Pr163ldNZVevXoxf/58unfvztSpUylWrBj79+9XXVRy2rXt3LlzGS6vXbs2kyZNYs+ePdStW5eJEydiamrKhg0b2Lt3b5qMqy1atKB06dK4urpiYWFBQEAACxYswNbWFgcHByIiIqhbty5du3alePHiGBoacvHiRQ4cOJBpK3SqqlWr8ueffzJo0CAqVKjAwIEDKVWqFElJSVy9epUVK1ZQunRp1TjBHj168MsvvzBx4kRq166Nj48PixcvTpMdNidMTExo06YNXl5ehIeH8+OPP6Z7T5cvX46bmxuNGzfG3d2dQoUKERoayp07d7hy5Qrbtm17r2MKgiAImfvWr823bt3iwoULDBw4MMPx+NWrV2fu3LmsXr2aIUOGMHz4cLZs2UKrVq0YO3YslSpVIi4ujuPHj9O8eXPq1q1Lly5d8PT0ZMCAAdy7d4+6deuiVCo5f/48JUqUoHPnzigUCrp3786aNWsoWrQo5cqV48KFC2zcuDHH521kZEStWrWYPXs25ubm2NnZcfz4cVavXo2JiUmasr/99hv79++nVq1ajB8/njJlyhAeHs6BAwcYOXIkxYsXV5Xt3r0748aN48SJE0yYMAEtLa1s66Kurs6OHTto1KgRVatWZeDAgdStWxd9fX0CAgL4+++/+ffffwkLCwOgU6dOTJw4kc6dOzN69Gji4+NZuHBhupsLOdGnTx+2bNnCkCFDKFy4cLp8AcOHD2f79u3UqlWLESNGULZsWZRKJYGBgXh7ezNq1CgqV6783scVvjJ5m7dOEL4smWWQLVWqVIblz5w5I1WtWlXS09OTLCwspL59+0pXrlxJlwE1swyyzZo1S7fP2rVrS7Vr11a9ziyD7Lv1zOw4gYGBUtu2bSUDAwPJ0NBQateunbRv3z4JkHbt2pXZW5Hm2Jk9Uut08+ZNqUWLFpKxsbGkpaUllStXLl0G2Llz50rVqlWTzM3NJS0tLcnGxkby8PCQ/P39JUmSpPj4eGnAgAFS2bJlJSMjI0lXV1dycnKSJk2aJMXExGRZz1TXrl2TevXqJdnY2EhaWlqSvr6+5OLiIk2cOFEKDg5WlUtISJB++uknydraWtLV1ZVq164tXbt2LdPs7u9mWH2bt7e36v3w9fXNsMz169eljh07SpaWlpKmpqZUoEABqV69etKyZctydF6CIAjfM3FtfmP48OESIF27di3TMmPHjpUA6fLly5IkSVJYWJg0bNgwycbGRtLU1JQsLS2lZs2aSXfv3lVtExcXJ02cOFFycHCQtLS0JDMzM6levXrSmTNnVGUiIiKkvn37Svnz55f09fWlFi1aSP7+/plmd3/58mW6uj158kRq166dlC9fPsnQ0FBq0qSJdOvWrXTXX0mSpMePH0t9+vSRChQoIGlqakoFCxaUOnbsKL148SLdft3d3SUNDQ3pyZMnmb4vGQkPD5emTJkilS9fXjIwMJA0NTUlGxsbqXv37tLp06fTlN23b5/k7Ows6erqSkWKFJEWL16caXb3wYMHZ3rMlJQUydraWgKkn3/+OcMy0dHR0oQJEyQnJydJS0tLMjY2lsqUKSONGDFCev78+Xudo/B1UkjS63SIgiB8N6ZPn86ECRMIDAzM0RhuQRAEQRA+LXFt/jCJiYnY2dlRo0YNtm7dmtfVEYSPQnR3F4Rv3OLFiwEoXrw4SUlJHDlyhIULF9K9e3fxI0AQBEEQ8oC4Nufey5cvuXfvHp6enrx48SJHCeAE4WshgnRB+Mbp6ekxf/58/P39SUhIwMbGhjFjxjBhwoS8rpogCIIgfJfEtTn39u7dS+/evbGysmLp0qXZTrsmCF8T0d1dEARBEARBEARBEL4QYgo2QRAEQRAEQRAEQfhCiCBdEARBEARBEARBEL4QIkgXBEEQBEEQBEEQhC/Ed5c4TqlU8uzZMwwNDVEoFHldHUEQBEFAkiSioqIoWLAgamri/vnHIK73giAIwpfkfa71312Q/uzZM6ytrfO6GoIgCIKQzuPHj8X0Sx+JuN4LgiAIX6KcXOu/uyDd0NAQkN8cIyOjPK6NIAiCIEBkZCTW1taqa5SQe+J6LwiCIHxJ3uda/90F6ald3oyMjMRFWxAEQfiiiG7ZH4+43guCIAhfopxc68XAN0EQBEEQBEEQBEH4QoggXRAEQRAEQRAEQRC+ECJIFwRBEARBEARBEIQvxHc3Jl0QhG+HJEkkJyeTkpKS11URhCypq6ujoaEhxpwLgiAIgpAtEaQLgvBVSkxMJCgoiNjY2LyuiiDkiJ6eHlZWVmhpaeV1VQRBEARB+IKJIF0QhK+OUqnEz88PdXV1ChYsiJaWlmihFL5YkiSRmJjIy5cv8fPzw8HBATU1MdpMEARBEISMiSBdEISvTmJiIkqlEmtra/T09PK6OoKQLV1dXTQ1NQkICCAxMREdHZ28rpIgCIIgCF8ocStfEISvlmiNFL4m4vMqCIIgCEJOiF8MgiAIgiAIgiAIgvCFEEG6IAiCIAiCIAiCIHwhRJAuCIIgfDReXl6YmJjkdTUEQRAEQRC+WiJIFwRB+ExatGhBgwYNMlx39uxZFAoFV65cUS3bvn079erVI1++fOjp6eHk5ESfPn24evVqmm0TExOZPXs25cuXR19fH2NjY8qVK8eECRN49uxZhsdzd3dHoVBk+fgQnTp1wtfX94O2TXXs2DEUCgXh4eG52o8gCIIgCMLXSATpgiAIn4mHhwdHjhwhICAg3bo1a9bg7OxM+fLlARgzZgydOnXC2dmZ3bt3c/v2bVasWEHRokUZP368aruEhAQaNmzI9OnTcXd358SJE1y+fJlZs2YREhLCokWLMqzLH3/8QVBQkOoB4OnpmW5ZqsTExBydo66uLpaWljkqK3xfli5dir29PTo6OlSoUIGTJ09mWX7JkiWUKFECXV1dnJycWLdu3WeqqSAIgiDkMek7ExERIQFSRERErvf1z5XHUuP5x6Xp+3w+Qs0EQcipuLg4ycfHR4qLi1MtUyqVUkxCUp48lEpljuqdlJQk5c+fX5o8eXKa5TExMZKhoaG0aNEiSZIk6ezZsxIg/fHHHxnu5+3jzZgxQ1JTU5OuXLmSbdmsANKOHTtUr2vXri0NHjxYGjFihGRmZibVqlVLkiRJmjt3rlS6dGlJT09PKly4sDRw4EApKipKtZ2np6dkbGysej1p0iSpXLly0rp16yRbW1vJyMhI6tSpkxQZGZlpXY4ePSoBUlhYWIbrQ0NDpR49ekgmJiaSrq6u1KRJE8nX11e13t/fX2revLlkYmIi6enpSSVLlpT27t2r2rZr166Subm5pKOjIxUrVkxas2ZNjt6j3Mroc5vqY16bvkSbN2+WNDU1pZUrV0o+Pj7SsGHDJH19fSkgICDD8kuXLpUMDQ2lzZs3Sw8fPpQ2bdokGRgYSLt3787xMb/191QQBCG3bj0Nl3yefV3fkS8i4yT3Neel9ef887oq7+19rktinvRciE5I4e7zKGzNxDzNgpDX4pJSKDnxYJ4c2+e3xuhpZf91qqGhQc+ePfHy8mLixImqLuXbtm0jMTGRbt26AbBp0yYMDAwYNGhQhvt5uyv6pk2baNiwIS4uLtmWfV9r165l4MCBnD59GkmSAHkasYULF2JnZ4efnx+DBg3ip59+YunSpZnu5+HDh+zcuZM9e/YQFhZGx44dmTlzJtOmTfugerm7u3P//n12796NkZERY8aMoWnTpvj4+KCpqcngwYNJTEzkxIkT6Ovr4+Pjg4GBAQC//PILPj4+7N+/H3Nzcx48eEBcXNwH1UPIuXnz5uHh4UHfvn0BWLBgAQcPHuTPP/9kxowZ6cr/9ddf9O/fn06dOgFQpEgRzp07x++//06LFi0+a90FQRC+RX6vYmiz5AwowHt4LezM9T9oP5IkcfZhCD5BkXSuZIOBds7Cy8RkJYuPPiAuMZm+NYuQ30gnR9tN3Hmbo/decsz3JXZm+lQvZv5B9c5IUEQcc719cbA0oKVzQayMdT/avt+XCNJzwVhXE4CIuKQ8rokgCF+LPn36MHv2bI4dO0bdunUBuat727ZtyZcvHwC+vr4UKVIEDY03X9Hz5s1j4sSJqtdPnz7F2NgYX19f6tSpk+YYbdq04dChQwCULVuWM2fOfFBdixUrxqxZs9IsGz58uOq5vb09U6ZMYeDAgVkG6UqlEi8vLwwNDQHo0aMH//333wcF6anB+enTp6lWrRoAGzZswNramp07d9KhQwcCAwNp164dZcqUAeQAL1VgYCAuLi64uroCYGdn9951EN5PYmIily9fZuzYsWmWN2rUKNPPZkJCAjo6aX+w6erqcuHCBZKSktDU1Mxwm4SEBNXryMjIj1B7QRCEb9OsA3dJTFECMHWvD6t6VXzvfZx5+Ir5h3y56B8GwLqzASzo7Ex5m3xZbhcak8jA9Zc57xeq2s69mh0Dahcln75WptsdvP2cA7efAyBJMGzzNfYNq4GlYc4C/KwolRLDN19T1WnmgbtUtjeltXMh3MpYqeK+z0UE6blg8vqPFR4rgnRByGu6mur4/NY4z46dU8WLF6datWqsWbOGunXr8vDhQ06ePIm3t3eacu+2gPfp04eWLVty/vx5unfvrmrZzqjs0qVLiYmJYeHChZw4ceIDzkiWGsi+7ejRo0yfPh0fHx8iIyNJTk4mPj6emJgY9PUzvgtvZ2enCtABrKysCA4O/qA63blzBw0NDSpXrqxaZmZmhpOTE3fu3AFg6NChDBw4EG9vbxo0aEC7du0oW7YsAAMHDqRdu3ZcuXKFRo0a0bp1a1WwL3war169IiUlhfz586dZnj9/fp4/f57hNo0bN2bVqlW0bt2a8uXLc/nyZdasWUNSUhKvXr3Cysoq3TYzZszg119//STnIAiC8C25HBDK/lvPUVOAmkLB4TvBHL0XTF2n9DllJEkiMi6ZuKQU+ZGYwvPIOJYff6QKaLU01DDW1SQwNJYOy84yrL4Dg+oURUM9ffqzu88j6bv2Ek/C4jDQ1qCYpQHXHoez/MQjNp4PpF+tIvStaZ+uh2JkfBITd90CoE91e848fMXd51EM23SN9X0ro66W9rfQzScR3HgazuPQOB6HxvI4LJaIuCR+aVaSBiXTXo8AvM74c94vFD0tdUoXNOaCfyjnHsmPibtus7CLC01KF/jg9/x9iSA9F1LvqESKlnRByHMKhSJHXc6/BB4eHgwZMoQlS5bg6emJra0t9evXV613cHDg1KlTaVoMTUxMMDEx4cmTJ2n25eDgwN27d9MsSw1gTE1Nc1XPd4PugIAAmjZtyoABA5gyZQqmpqacOnUKDw8PkpIy/x58t9VToVCgVCo/qE5v35x4d3nqzYq+ffvSuHFj9u7di7e3NzNmzGDu3Ln88MMPuLm5ERAQwN69ezl8+DD169dn8ODBzJkz54PqI+TcuzeT3v6bveuXX37h+fPnVKlSBUmSyJ8/P+7u7syaNQt19Yxvio0bN46RI0eqXkdGRmJtbf3xTkAQBOEr4f8qhtkH71GyoBGD6hRN810rSRLT9so3tTu6WmOgrcGqU35M+deH6kXN0dJ4E1iHxybyv78uc+F1MP4uLXU1ulSyZmCdYuhqqfPLzlvsvv6MeYd8OeH7kpENHbEw1MZETwsTPU2O3g1mxJZrxCSmYGumx6qerhSzNODYvZfMOniPO0GRzDvky94bQazoWQFbsze/Q2YfuMeLyATszPT4qYkTT8JsaLn4FGcfhfDHYV9GNnIC5G78M/bdwdvnRYZ1HrTxCmt7V6JqUTPVsocvo/n9gPxbanzTEnSvYsuTsFh2X3/GrqvP8A2Oomxh4w/8a3yYPM3uPmPGDCpWrIihoSGWlpa0bt2ae/fuZbvd8ePHqVChAjo6OhQpUoRly5Z9htqmZ6L3uiVdBOmCILyHjh07oq6uzsaNG1m7di29e/dOcwHt0qUL0dHRWXYhf7vsoUOH0k3L9ilcunSJ5ORk5s6dS5UqVXB0dMx0irdPpWTJkiQnJ3P+/HnVspCQEHx9fSlRooRqmbW1NQMGDOCff/5h1KhRrFy5UrXOwsICd3d31q9fz4IFC1ixYsVnPYfvjbm5Oerq6ulazYODg9O1rqfS1dVlzZo1xMbG4u/vT2BgoKpHhrl5xuMPtbW1MTIySvMQBEH4kkXEJRGbmJyjsi+jEth++QnDNl+l9uyj/LDpKg+Co9KUkSSJDecDcPvjJHtvBjH74D1m7r+b5gb3gVvPuRIYjq6mOiMbOjK0gQPmBlo8ehWD52k/VbnQmES6rjyvCtA11BQYamtgbqCNjake3avYcGx0HX5tVZoCxjoY62ryR2dn5ncqh4G2BpcCwui66jwN55+g4rTDOPy8n//9dZmYxBSqFTVj56DqOOQ3RKFQULe4JXt/qMHiri5YGGpz70UULRef5oTvS0Bu+V9/Xp4ZZ3rbMuhoqlPM0oAZbeVhbYuOPmDPjWf89q8PDecdx9vnBepqCmo7WuBezY4JzUqwvEcFGpbMT2Kykn7rLnHraQQAKUqJH7ddJyFZSU0Hc7pVtgGgcD49BtUpxsERtTj2Yx0Kmnze8el52ux0/PhxBg8eTMWKFUlOTubnn3+mUaNG+Pj4ZNpt0s/Pj6ZNm9KvXz/Wr1/P6dOnGTRoEBYWFrRr1+6z1j+1JT02MYXEZGWaO0+CIAiZMTAwoFOnTowfP56IiAjc3d3TrK9atSqjRo1i1KhRBAQE0LZtW6ytrQkKCmL16tUoFArU1OTvmxEjRrB3717q1avH5MmTqVmzJvny5cPX15f9+/dn2ur4IYoWLUpycjKLFi2iRYsWnD59+pPeJL1582aabvIAzs7OtGrVin79+rF8+XIMDQ0ZO3YshQoVolWrVoA8bt7NzQ1HR0fCwsI4cuSIKoCfOHEiFSpUoFSpUiQkJLBnz540wb3w8WlpaVGhQgUOHTpEmzZtVMsPHTqk+ptlRlNTk8KFCwOwefNmmjdvrvrsC4Ig5CVJkrj7PIqDt59z8PYLwmMTWdXLlVIFs25xfRIWy8HbLzh46zmXAkLR0lCjjUth3KvZ4VTgzTVPkiRuP4vk4O3nHLkbzO1nafNsBITEsufGM1qWK8jQ+g4YamswZvsNjt6TA9viBQy5+zyK5Sceoa2hxshGTiQmK1Utxv1qFcHydbK2n5oU56e/b7Dwv/u0cSmEQqGg+6rz3HsRhbmBFuv7VqZ4gexvfCoUCtq4FMbV1pSpe3249zyKsNgkIuOTSL1P0LOqLb80L4nmO13h1dQUNC9bkIp2pvT/6zLXHofj7nmBHxs7sePKUyQJOlQoTLWib27UtnIuxLlHoWy6EMiQjW8aK+oVt2R80+IUs0z7G6K2owW91lzgvF8ovdZcYNuAqhy8/YKrgeEYamvwe7uyGfbwertF/3PJ0yD9wIEDaV57enpiaWnJ5cuXqVWrVobbLFu2DBsbGxYsWABAiRIluHTpEnPmzPnsQbqhjiYKhZy4ICIuCQtD7c96fEEQvl4eHh6sXr2aRo0aYWNjk279nDlzqFSpEn/++aeqRTF//vzUqlWLs2fPqloJdXR0+O+//1iwYAGenp6MGzcOpVKJvb09bm5ujBgx4qPV2dnZmXnz5vH7778zbtw4atWqxYwZM+jZs+dHO8bbMroOSJKEp6cnw4YNo3nz5iQmJlKrVi327dun6lafkpLC4MGDefLkCUZGRjRp0oT58+cDcsA4btw4/P390dXVpWbNmmzevPmT1F94Y+TIkfTo0QNXV1eqVq3KihUrCAwMZMCAAYDcVf3p06equdB9fX25cOEClStXJiwsjHnz5nHr1i3Wrl2bl6chCMI3JiQ6gfvB0YTHJhIak0RYbCIJyUq6VbbJNNt4XGIKi47cZ8+NIAJDY9Os8/C6xK4h1TPc9rjvS2YfvMutp2mD7fgkJZsuBLLpQiDViprRtnxhfF4H50/D084+UqaQMbUdLShb2JjtV55w8PYLdl17xr/Xn6GvpUFUQjJaGmr81NiJPtXtWXvWn1//9WHhkQdoaahhoK2Bf0gs5gba9K/1Jqlq+/KF2XA+kOuPw/l55y0evYzm4csYLA212divCsUsDd7rfbU21WN5jzd5bVKUkirRtmkWieEA8hvpsKV/FX7ZeYutl54w64Dcy9rcQIufm6W/qT6pRUmuPQ7nTlAkxQsY8nOzEtR0sMhw3zqa6qzq5UqXlee49TSSbqvOExKdCMAvLUp+9tbyrCikzAb45YEHDx7g4ODAzZs3KV26dIZlatWqhYuLC3/88Ydq2Y4dO+jYsSOxsbHpxj5mlO3V2tqaiIiIj9IVruzkg0TGJ3N4ZO33/gALgvBh4uPj8fPzw97ePl0GaEH4UmX1uY2MjMTY2PijXZu+REuXLmXWrFkEBQVRunRp5s+fr7oR4+7ujr+/P8eOHQPkBIFdu3bl3r17aGpqUrduXX7//XecnJxyfLzv4T0VhO9BVHwScUkpHyWDdypJkth04TG/7blNfFL6HCklrIzYPrBqulw3kiQxaMMV9t+Sh+9oa6hR08GCRiXzs+LkIx4ER1OmkDFb+ldJs+3Wi48Zt+MmKUoJNQVUtDOlSekCNCpVgKdhcXid8ePAreco34nKdDTVqONoScOS+anlaJGuQfDW0wgWHL7P4Tvy+OtSBY2Y38kZx/xvWpCXH3/IjP13VfVNSFYyrU1pulW2TbOva4/Dab3ktOq1lbEOG/tVwf4Dp2bLLUmSWH8ugF//9SFZKfFHZ2daORfKsGxUfBK3nkZSyd40XQK5jLyKTqDjsrM8ehUDQP3ilqzq5ZqraWtz4n2uS19MkC5JEq1atSIsLIyTJ09mWs7R0RF3d3fGjx+vWnbmzBmqV6/Os2fP0mV8nTx5cobZXj/WRbvWrKMEhsayfWBVKtjmLkmTIAg5I4J04Wv0vQfpn5t4TwXh63fc9yVDNlxBoYCDI2rleN7qx6Gx/LLrFvkNdehZzTZNF/SI2CTG7bjBvptyoF04ny75jXTIp6eJiZ4WR+8GExKTSLOyVizu4pImcFv0333mHvJFS12Nme3K0LhUAfRfzwseGBJL66WnCY1JpEmpAiztVh6FApYcfcAcb18A2pYvxM9NS2BmkL737dPwOP46G8Bx35eUsDKkcakC1HKwQFcr+2Frt55G4PsiiuZlC2Y4/Da13gDFLA04MKxmhpnXR2+7zrbLTyicT5dN/apgbaqX7bE/tbvPIwmKiM8w83xuPAmLpduq8ySnSOwYVE3V9f9Tep/r0heTCnnIkCHcuHGDU6dOZVs2owyxGS2HT5/tVcyVLgiCIAiCIAgf19oz/vz6721V6/Ly44+Y3LJUttvdex5Fj9XnCY6Se9JuufSYSvam9K5mh5mBNiO2XONpeBya6gp+alwcjxr2qL3V+nrBL5SuK8+x90YQJa2MGFy3GACHfF6oAt0prUvRtnzhNMe1MdNjRY8KdF15ngO3n/P7gbvEJaWw7qyc8GxQnaKMbuyUaWttIRNdxroVZ6xb8fd7o4DShYwpXSjzsfA/1HdATU3B5ouBTG1dOsMAXT6v0lQtakZNh/St9nmleAGjHI2Hf1+F8+lxZFQdklKU6LzHVLqfyxcRpP/www/s3r2bEydOqBLEZKZAgQIZZojV0NDAzMwsXXltbW20tT/dh0wE6YIgCIIgCILwcSSnKPltj48quK1axIyzj0LYdCGQwXWLZRk8Xg4IpbfnRSLjk3HMb4BTASP23wzigl9ommnEbM30WNTFhbKFTdLto5K9Kb+2KsXPO24xx/seJawMsc6nx4gt1wDoVdWWThXT55IBcLUzZVb7sgzfco3lJx4BoFDApOYlca9u/4HvyMcxuG4x1Q2HzOhoqqe7+fAtU1dToK725QXokMdTsEmSxJAhQ/jnn384cuQI9vbZf3irVq3KoUOH0izz9vbG1dU13Xj0z8E4dRq2WBGkC4IgCIIgCMKHiopPorfXRdadDUChgLFuxdnYrzIuNiYkJCtZdfJRptsevRtMt1XniYxPpoJtPrb2r8qiLi6cGlOPIXWLqRKWtXYuyJ4famQYoKfqVtmWrpVtkCQYtukaHmsvEZ2QTJUipkxoXjLLc2jtUoih9R0AeR7xRV1c8jxAF74+edqSPnjwYDZu3MiuXbswNDRUtZAbGxujqyuPOXk34+uAAQNYvHgxI0eOpF+/fpw9e5bVq1ezadOmPDkH0ZIuCIIgCIIgCNmTJCnL5Fy/7LzFyfuv0NVUZ34nZ5qULgDAD/WK0cfrEn+dC2BA7aLkeydD+K5rTxm19TrJSok6Thb82a2Caix3AWMdfmzsxJB6xXgWHoe9uX6OEoRNblGK+y+iuOgfRlRCMoVMdFnStXy6qcMyMqKBA075DbE316dkQZETQ3h/eRqk//nnnwDUqVMnzXJPT0/VvMFBQUEEBgaq1tnb27Nv3z5GjBjBkiVLKFiwIAsXLvzs06+lMtEVLemCIAiCIAjCly0sJpHtV57wPCKel9EJvIxKICQ6kYr2+fi5ackME5TFJ6Ww4PB9HgRHU6qgEeWsjSlb2ATzDBKfZeVlVAITd93ivF8oi7u6pJnrOtX5RyHsvPYMhQL+8qiEq92bhMx1nSwpVdCI288iWXPaj1GN3sz0sO9mECO2XEMpya3kszuUyzCQ1tFUp4hFzmdi0tJQY2m3CrT78wxhsYms6Fkhw4RvGVEoFDQra5V9QUHIRJ4G6TlJLO/l5ZVuWe3atbly5conqNH7S21JjxQt6YIgCIIgCMIXKDFZSacVZ/F9EZ1u3b0XUdx8EsHKnq5pMly/iIznf+sucf1JBIBqmi+Qp+cy0E4bRpjqa9GuQmFalC2YJuDfc+MZv+y8RdjrBq3hm6+xf1jNNAFvcoqSSbtvA9Clkk2aAB3koPeHesUYsP4KXqf96VuzCMa6mhy5+4Khm66ilKCja2Fmti2bJglcblkYanNoZC0SkpUY6Xz+YbXC9+uLSBz3NTNJHZMugnRBEARBEAThC7Tq1CN8X0STT0+TDq7WWBhoY2GojYTEr//6cP1JBK2WnGZVL1dKFTTm+uNw/vfXJV5EJmCip0m/mkV4GBzNjacRPHwZTVBEfIbHOe8XytQ9PnRwtaZFuYKsPPGIvTeDAHnu8YTkFB69jGHM9hus7PlmXuoN5wO5+zwKEz1NRr/VSv62RiUL4JjfAN8X0fx11p/yNvkYsP4KyUqJFuUKMuMjB+iptDXU0db4MpOLCd8uEaTnkhiTLgiCIAiCIHypHofGsvC/+wBMaFaSdhXSZu8ub5OPPl4Xefgyhg7LztKzqh2ep/1ISFbiYGnAql6u2Jrpq8pHxSfh+yKKxOQ3PWIlJG48iWDD+QAeh8ax+pQfq0/5AXIG7cF1izGkbjEeBEfTeslpDt8JZt3ZAHpVs+NVdAJzve8B8GMjp3TjzVOpvd7PsM3XWHHiEclKicRkJQ1K5Gdex3Kof4IAXRDySp5md/8WGOvKXyThsYl5XBNBEITPz8vLCxMTE9XryZMn4+zsnOU27u7utG7d+pPWSxAE4WsTl5jCkqMPaPfnGbZffpKjYaHZkSSJX3bdIj5JSdUiZrQtXyhdGVszff4ZVJ2aDubEJqaw7PhDEpKVNChhyT+DqqUJ0AEMdTSpYGtK1aJmqke1ouYMqF2U4z/WxbN3RRqUsEShAMf8BuwcVJ2RDR3R0lCjZEEjxjWV5wGftu8Od4IimX3gHpHxyZQqaESXShlPbZaqedmC2JvrExmfTGxiCjUdzFnc1SVHydwE4WsiPtG59KYlPTmPayIIwpeuRYsWNGjQIMN1Z8+eRaFQpMm3sX37durVq0e+fPnQ09PDycmJPn36cPXq1TTbJiYmMnv2bMqXL4++vj7GxsaUK1eOCRMm8OzZswyPt337dtTV1dMk5nxb8eLFGTp06Huf448//sh///333tu97dixYygUCsLDw3O1H0EQhK9BcoqSzRcCqTPnKLMP3uNyQBijtl2ny8pzPAhOP4b8fey7+Zxj916ipa7G1DalM81qbqyryRr3ivSsaouWuhoD6xRleQ9XDN9zHLaamoK6Tpas6lWRa780Yv+wWpQpbJymjHs1O+oVtyQxWUnftZfYcukxAL+1KpVta7i6moLhDeTpzSrZmbK8RwV0NEVXdOHbI4L0XEqdJz0iLvGj3PEUBOHb5eHhwZEjRwgICEi3bs2aNTg7O1O+fHkAxowZQ6dOnXB2dmb37t3cvn2bFStWULRoUcaPH6/aLiEhgYYNGzJ9+nTc3d05ceIEly9fZtasWYSEhLBo0aIM69KyZUvMzMxYu3ZtunWnT5/m3r17eHh4vPc5GhgYYGZm9t7bCYIgfG9SlBIHbj3H7Y+TjP3nJi8iEyhkoot7NTt0NNU49ygUtz9OMNf7HvFJKRnu40FwFEuOPqDl4lPUm3uMhf/dJzhSHi8eGZ/Er//KydgG1ClK0Wwym2uqq/Fbq9Lc/q0xY5oUz3X3cWM9zQz3oVAomN2+LBaG2jwNjwOgXfnCVLA1TVc2I62cC3FoRC029KuMnpYYuSt8m8QnO5dSp2BLSpGIS0oRXxaCkFckCZJi8+bYmnqQgzlXmzdvjqWlJV5eXkyaNEm1PDY2li1btjB9+nQAzp07x6xZs/jjjz/StGbb29tTu3btNDcE58+fz6lTp7h06RIuLi6q5cWKFaNx48aZ3jzU1NSkR48eeHl5MWHChDStK2vWrKFChQqUK1eOefPm4enpyaNHjzA1NaVFixbMmjULA4OMf+xNnjyZnTt3cu3aNQBSUlIYPXo0a9asQV1dHQ8Pj1zf0AwLC2PYsGH8+++/JCQkULt2bRYuXIiDg9y6EhAQwJAhQzh16hSJiYnY2dkxe/ZsmjZtSlhYGEOGDMHb25vo6GgKFy7M+PHj6d27d67qJAjCt+NFZDy3n0Xg8yySO0FR3HkeSSETXX5pXhLH/Ia53v+TsFi2XnrCtkuPVQnYTPQ0GVK3GD2q2qKtoY5HDXsm7rrF0XsvWXTkAStOPKKgiS4FjHSwMtHBUFuDUw9e8fBlTJp9zzvky8L/7tO4lDy/eHBUAvbm+gyqUzTH9fscXcfNDLSZ17EcPVZfwFBHgzFuGSeLy4zDR/g7CMKXTESUuaSnpY6GmoJkpUREXJII0gUhryTFwvSCeXPs8c9ASz/bYhoaGvTs2RMvLy8mTpyoCoy3bdtGYmIi3bp1A2DTpk0YGBgwaNCgDPfzdkC9adMmGjZsmCZAz6zsuzw8PJg3bx7Hjx+nTp06AMTExLB161ZmzZoFgJqaGgsXLsTOzg4/Pz8GDRrETz/9xNKlS7M9X4C5c+eyZs0aVq9eTcmSJZk7dy47duygXr16Odo+I+7u7ty/f5/du3djZGTEmDFjaNq0KT4+PmhqajJ48GASExM5ceIE+vr6+Pj4qG4q/PLLL/j4+LB//37Mzc158OABcXFxH1wXQRC+LYv+u8/cQ77plj96GUPzhafkacDqFM02kA2NSWTj+QDVtGOp7gdHc/L+S1LvVZroadK1kg39axdVDaEEsDbVY417RQ7efs6v//oQFBGP36sY/F6lDco11RVUL2ZO41IF0NZQY8P5QC4HhKkyqgNMaVX6i+wSXtPBgh2DqmGoo4mloU72GwjCd0RElLmkUCgw0dPkVXQi4bFJWBnr5nWVBEH4gvXp04fZs2dz7Ngx6tatC8gt123btiVfvnwA+Pr6UqRIETQ03nxFz5s3j4kTJ6peP336FGNjY3x9fVUBdqo2bdpw6NAhAMqWLcuZM2cyrEvJkiWpXLkynp6eqn1s3bqVlJQUunTpAsDw4cNV5e3t7ZkyZQoDBw7McZC+YMECxo0bR7t27QBYtmwZBw8ezNG2GUkNzk+fPk21atUA2LBhA9bW1uzcuZMOHToQGBhIu3btKFOmDABFihRRbR8YGIiLiwuurq4A2NnZfXBdBEH4tpzwfakK0B0sDShV0IiSBY0oZmnAhnOB/Hc3mLmHfNl/6zmz2peldCHjdPtIUUpsuhDIHO97hMdmPvNPtaJmdK5kQ6OS+TMNoBUKBU1KW9GgRH6ehcfzLCKOoIg4noXHExqTSNnCxtQtbplm/u625QtzJyiS9ecC2HcziFbOhajhYJ7Ld+bTcbHJl9dVEIQvkgjSPwIjXTlIF9OwCUIe0tSTW7Tz6tg5VLx4capVq8aaNWuoW7cuDx8+5OTJk3h7e6cp924LeJ8+fWjZsiXnz5+ne/fuabqMv1t26dKlxMTEsHDhQk6cOJFlfTw8PBg+fDiLFy/G0NBQdcMgNWP70aNHmT59Oj4+PkRGRpKcnEx8fDwxMTHo62fdeyAiIoKgoCCqVq2qWqahoYGrq+sHd3m/c+cOGhoaVK5cWbXMzMwMJycn7ty5A8DQoUMZOHAg3t7eNGjQgHbt2lG2bFkABg4cSLt27bhy5QqNGjWidevWqmBfEITvV3BUPCO3XgOgexUbprYuk2Z9XSdLdl17xuR/b+MTFEmrJaepUsSUyvZmVLI3xdnahDtBkUzcdZubTyMAKF7AkDpOlmn2Y6ijQfOyVukypmdFQ10NGzM9bMxydq0pYWXEtDZlmNamTPaFBUH4Iokg/SNIHZee1R1TQRA+MYUiR13OvwQeHh4MGTKEJUuW4Onpia2tLfXr11etd3Bw4NSpUyQlJaGpKX+/mJiYYGJiwpMnT9Lsy8HBgbt376ZZZmVlBYCpafZJeDp37syIESPYsmULderU4dSpU/z222+APLa7adOmDBgwgClTpmBqasqpU6fw8PAgKSlvvu8yC+4lSVLdrOjbty+NGzdm7969eHt7M2PGDObOncsPP/yAm5sbAQEB7N27l8OHD1O/fn0GDx7MnDlzPudpCILwBVEqJUZtvc6r6ESKFzBkQrOS6cooFApauxSiejFzJu2+xb6bzzn9IITTD0IA0FJXIzFFCYChtgYjGznSo4otGmJqMEEQPoD45vgIUscQRYqWdEEQcqBjx46oq6uzceNG1q5dS+/evdO0hnfp0oXo6OgcdSnv0qULhw4dSjctW04ZGhrSoUMHPD09WbNmDUWKFFF1fb906RLJycnMnTuXKlWq4OjomOmUbhkxNjbGysqKc+fOqZYlJydz+fLlD6oryF30k5OTOX/+vGpZSEgIvr6+lChRQrXM2tqaAQMG8M8//zBq1ChWrlypWmdhYYG7uzvr169nwYIFrFix4oPrIwjC12/FyUecvP8KHU01FnVxyXL8toWhNku7VeDQiFpMaVWKZmWtsDDUVgXo7coX5siPdehd3V4E6IIgfDDRkv4RmOhpARAel5jHNREE4WtgYGBAp06dGD9+PBEREbi7u6dZX7VqVUaNGsWoUaMICAigbdu2WFtbExQUxOrVq1EoFKipyT/+RowYwd69e6lXrx6TJ0+mZs2a5MuXD19fX/bv34+6evbJgjw8PKhZsyY+Pj78+OOPqhsGRYsWJTk5mUWLFtGiRQtOnz7NsmXL3utchw0bxsyZM3FwcKBEiRLMmzcvx/Of37x5E0PDtBl8nZ2dadWqFf369WP58uUYGhoyduxYChUqRKtWrQB5HL2bmxuOjo6EhYVx5MgRVQA/ceJEKlSoQKlSpUhISGDPnj1pgntBEL4vVwPDmHPwHgCTW5TKcdZwh/yGOOQ3pEdVOyRJwj8kFg01BdamOR/+JAiCkBlxi+8jSG1JF2PSBUHIKQ8PD8LCwmjQoAE2Njbp1s+ZM4eNGzdy9epVmjdvjoODAx06dECpVHL27FmMjIwA0NHR4b///mPs2LF4enpSo0YNSpQowfDhw6levTo7d+7Mti41atTAycmJyMhIevXqpVru7OzMvHnz+P333yldujQbNmxgxowZ73Weo0aNomfPnri7u1O1alUMDQ1p06ZNjratVasWLi4uaR4Anp6eVKhQgebNm1O1alUkSWLfvn2qoQEpKSkMHjyYEiVK0KRJE5ycnFS9ErS0tBg3bhxly5alVq1aqKurs3nz5vc6J0EQvn7yHOVBDNl4lWSlRLOyVnSqaP1B+1IoFNib64sAXRDe18Oj4HsQcjk167dIIeV2wtqvTGRkJMbGxkRERKh+5OZW6pyU3SrbiCQdgvAZxMfH4+fnh729PTo6YtoW4euQ1ef2U1ybvnfiPRUyEpuYzLZLT1hz2o+AkFgAbEz12DO0Rpos6YIgfGIXVsK+H+XnTs2gxQIwsMxyk08u5CEcmgjFGkAFdznf0Uf0Ptcl0d39IzARLemCIAiCIAifxYvIeNac8iOfvhb25voUtZBbsbU1Mh/ek5yiZPUpP5Yee6j6vWasq0n3Kjb0qW4vAnRB+JzOLoWD4+TnCjW4txcCz0LzeVAqZ73tsqVMAWUyaGjnrHxyAmztBS9uwt09cGc3tFwExoU/Tn3ekwjSPwLR3V0QBEEQBOHTi4xPoufqC9x7EZVmuZpCnnrMo4Y9LcoVRPOtpG13giL56e8bqqnRbM308KhhT/sKhdHTEj+FhW/A3b1yt/HG00HbIK9rk7XTf8it1QA1R0GptrBjgBwcb3OHO/9Cs7mgm+/DjxH9EpbXgqhnoGUAeqagayoH3HXGQYHS6bc5Ol2ug7YxpCTAwyOwtBq4/Q7lOn/0VvXsiDHpH4GJngjSBUEQBEEQPqWkFCWDN1zh3osoLAy1aVmuIGUKGaOvpY5SgtvPIhm59Tp1Zh/D87QfEXFJzD/kS4tFp7j5NAIjHQ1mtS/LkVF16FnVTgTo37qnVyAs4PMc694B2D8G4sI/z/HeFuYP2/vClbVwYfnnP/77ODHnTYBeeyzU+0UOmPsdgVqjQaEOt7bDhg6QFP/hxzm9QA7QARKjITwQgq7JLeRezeDZOzPiBJyRbx4AtF4CA05BIVdIiICdA2BzV4h68eH1+QDi2+kjEC3pgiAIgiAIn44kSfyy8xYn779CT0sdT/eKlC5krFr3IjKBHVefsvqUH0/D4/j1Xx+m7PFB+TrzUuNS+ZnSqjSWRiKPyXfBZxds7Sl3pS7REqr9AIVdP82x4sLgn//JAd2zq9BjB2jpf5pjvUuSYO8oSJLzK3BxNVQbBuofMcS7uw9e3YNqQ0Et+xljVPV6egWCrsKrBxByH17dh/DXN03qToDao9+U19CCehPA0Q3Wt4EnF2HPcGj95/u3YEc9h4ur5Ocd/4L8pSA2FOJC5ZsETy7A2lbQ4x/5MxEfCf/0ByRw7g4lWsjb9jkIZxbKLez39kHpdlCm/fvVJRdEkP4RpLakh8eKIF0QBEEQBOF9xSYm8zIqgZdRCcQmplCyoBHmBm/Gki47/ojNFx+jpoBFXVxUATrI2dULGOswsE5Rele3Y/uVJyw//ojA0FjM9LX4tVUpmpWxUk0vKXwGr+7D9U1gXxuK1P68x44JgT0j5eeSEnx2yg+balB9KDg2+bhdl88ukQN0gMfn5VbXLltA8yPdEIoLh6Q4MLJKv+7WdnhwGNS1QFMPIp/KrcWlWn+cYycnwD/95NZog/zg3DXr8inJ8ljuM4vg2ZX06xXqUH8i1Bie8faFK0AHL1jfXv78WJaA6sPer84n50FyPFhXlgNuhQLMisrrbKvBho4QeAbWtYbu2+GyF0QEgoktuM18sx91Dag5Ehwaye9z6XbvV49cEkH6R2D0uiU9Mj4JpVJCTU1cBARBEARBELLy8GU0Y/6+wZ2gSGISU9KttzXTo7xNPiyNtFl+/BEAE5uXpH6J/JnuU0dTnW6Vbenkas2lgDCKFzDERE/rk52D8BZJkpN/nVkktzwCnFkMvffLwdfnsn80xL4Cy5LQeqmcRfzGVjkwCzwDlfrL44yzCtQToiDkwetW4AdgYAGuHum3iQmBc3/Kz2uOgnPL4NEx+LsPdFwL6rlMSJgUByvryl3aG8+Ayv3f1CE2FA6MfX3sHyElEU7OgQsrPl6QHnBGDtBBblEu3S7jRGyJsXBlHZxbInctB9DQgSJ1wNwBzIqBmQNYFAd9s6yPWbQeNJkp/x0PTQJzJ3BqkrP6RjyFy57y87o/p/97aRtC979hYyfwPwnrWsoBvUIN2q6Q17+rQOmMx7B/YiJI/whSu7tLEkTFJ2OsJzKECoIgCIIgZOaSfyh9111K0wtRV1MdSyNtNNQUPHoVQ0BIrGqaNIDe1e1wr26fo/1rqKtRpUg2wYCQnjIFIh7LrYrv09rsdxIOT4Knl98sM7aW97W5K/zvKBgV/Pj1fdedf+VWT4U6tFoCBV3kQL3eL3IAeWaRPG5bmQxN54DaW+m5lEq4tBpOLYDIJ+n3nZwIVQelXXZmoRzEFigrH8O+tjye+t5e2DkQ2qxIe4z3dWElhMo3qDgwBl7ekeutrim/3zEv5SC2xnCIeQWn5kPAaXh+Ewp8hGmh7x968zziMVzyhCoD0pZJSYYN7eXjAuiZQaX/QcW+oG/+Ycet1A+CfeSAe7sH9D0st6onxcvd1uMj5dbxd2+CnJwj36ywrQH2tTLet5Y+dN0Km7vIN1QAqg8HmyofVtdPRATpH4G2hjq6murEJaUQEZckgnRBEARBEIRM7L8ZxLAt10hMVlKusDGzO5SjkIku+tpvfpZGxidxLTCcywFhXHscjr25PhOalczDWn/jXvrCtQ1wYwtEBcljc1v8kbOxzQFnYX07OSO2ujY4d4GqQ8CwAKxqKAeWm7uC+z7Q0vt05xAb+qabe/VhUKj8m3VGVtBoqtySu2uIHIwrk6D5H3IQHf4Ydg0Gv+NvttG3kFt/dYzA9wAc+gUKOstdpgGig+VWa3jTalukttyCvqU73Nwmd0FvvuDDAvX4CDg1T37u6CbX4bKXPJd35f5yyzXIfycNbTAuBCVbwu0dcH45tFqcdn8RT+H2P3LXeF1T0Msn/2vumHlG+Pve8r/FGsjd6k/MBpduaVucj82QA3RtI2j4K5TrApq673++b1MooOlsuReD/0lYUVdelvTmph2mRaD1MrCpLL8OC4Arf8nP647P+iaTlh502QwHxslBfZ1xuavvJyCC9I/EWFeTuKQUwuMSseETfgEJgiAIgiB8pdac8mPKXh8kCRqUsGRhF5cMs6wb6WhSy9GCWo4WeVDL74Qkyd3AL66UE3W97dp6uct4e8+sA+uX92BTZzlAd2wCLRfLXcNTddkEK+vJCdV2DYb2az7dVFYHxkJMsByI1xmbcRmX7qCmIbdyX1kn9xywqQoHx0NCJGjoQoPJ8pRbuibyNpIkZ0+/9bc8RVj/E/INiNN/yEFjoQrg2PjNMZzcoM3yNxnXJSW0WPj+gfqZRXJSOnMn6LxBDpL/9pCDVv+TcpkK7mBb9c02lfrLQfrNbdDwN3nqMZC77a9tLt+AeZexNQy5lH4MfaifnPBNoS73CFjdEEIfynOc1xkjl3l0DE7OlZ+3WPBxx22ra0LHdfLnJ8zvzXKFurwu9BGsaSwnBaz7s9yKrkySu9jbVc9+/5q6cp2/UGIKto9ETMMmCIKQe8eOHUOhUBAeHg6Al5cXJiYmWW4zefJknJ2dP3ndBEHIndkH7/LbHjlA717FhmXdK4hp0PKKMgX2jYYd/5MDdIU6ODSWg6KOf8njiX0PwLpWcgt1RqKeywm+4sOhcEU5oDd456aKqT10+ksOjG//I2fX/thiQl6PO98ijy1utTTjcdOpynWGtivlc762AXYPkQP0wpVg4Gm5O3dqgA7yTYWWC8GiBES/gG295Zb31AziGbXalmkvj3FWqMHVv+QbFMr0eRcyFR0sJ6QDqP+LnFXdsTH0PSQPRQA5kVuDX9NuZ1NF7nqfHP+mpf3lPfBqKgfopkWhZGuwqwn5S8t/54jHcrK3dz04/HqfVeVx5PUmyK/PLJLf8+hgOas9EpTv9WkSq+mZQv/j8hRtQ6/C2ECYGAKj7kG5rvKxzyyE5TXh6gZ5mzrjP3498oAI0j+S1ORxIsO7IAjZefz4MR4eHhQsWBAtLS1sbW0ZNmwYISEh6co+ePCA3r17U7hwYbS1tbG3t6dLly5cunQpTbmjR4/StGlTzMzM0NPTo2TJkowaNYqnT58Cb4Lf0qVLk5KS9oeCiYkJXl5eqtd2dnYoFArOnTuXptzw4cOpU6dOhud0+fJlFAoFp06dynB948aNadmyZXZvTTqdOnXC19f3vbd7m7+/PwqFgmvXruVqP4IgfDiv034sOfoQgJ+aODGlVWk01MXP0DyRnCAnNru4ElDI81OP9IFuW6FkK7nLdM9doGMiT1e1pglEvDNGOyFKHnsdESgHfl22ZN7iblcDmr1ubT06VQ7UlcoPr78yBa5vgZ2DYFEFmF0E9v0or6v2Q86S1JVpD+1XyzcP1LXkYLfPgTdZwN+lpQ+d1oOWoZx8bnXDNxnEi9bPeJuyHaHdKvlmwPWNsGOAPH47J07MftNKX7z5m+WWJaDfUbl7dre/095MAPlmQeX+8vOLq+Sx6V7N5JsL+UuDh7fcHd99j3xDosbr4QGXPNPXIbWru0ND+d+SreUbAIlRcqv1jv7yfi1KyInePhUdY/l9MC0iP1co5PNu8yd03igPS3jlC1KK3C0/tfv7V058O+bGy3twaQ08OIyJmCtdEIQcePToEa6urvj6+rJp0yYePHjAsmXL+O+//6hatSqhoW9aLC5dukSFChXw9fVl+fLl+Pj4sGPHDooXL86oUaNU5ZYvX06DBg0oUKAA27dvx8fHh2XLlhEREcHcuXPTHP/hw4esW7cu23rq6OgwZsyYHJ9XhQoVKFeuHJ6e6S/0jx8/5vDhw3h4eOR4f6l0dXWxtLR87+0EQfh0noTFctjnBeGxiTkq7337Ob/u8QFgdGMnBtUpJqZDUyrh4VH4d5g8zvhziY+Uk3z57AQ1TTlQrTdB7r79NpsqctBqWFCeI/uPcrDIFTZ2Bu8JsKkLPL8hB0jd/84+Y3cFdzmABjgyBTZ2kFtjP8ShiXIPgGsb5DHLIHcJr/bD+7WilmoDgy/ILbQ1hmc/B7h5MTkwhDfdxjPKIP620u2gg6d8M+DmVnk6s/jIrI8T5v8maG4wOf3+9c3k7vxWZTM/pq6p3EK+sr6cXK5AWej1b/pEbuV7yDcRAs9A8N03y5PiwO+E/Nyhkfyvmho0mCQ/P7cUHh6Rhwd0yGZIxKdUvBkMOg9lOkA+O7mL/zdCBOm54XsA9oyAa5tUGd5FkC4IeUOSJGKTYvPkIUlSjus5ePBgtLS08Pb2pnbt2tjY2ODm5sbhw4d5+vQpP//8s+p83N3dcXBw4OTJkzRr1oyiRYvi7OzMpEmT2LVrFwBPnjxh6NChDB06lDVr1lCnTh3s7OyoVasWq1atYuLEiWmO/8MPPzBp0iTi4+OzrGf//v05d+4c+/bty/G5eXh4sHXrVmJiYtIs9/LywsLCgmbNmrF+/XpcXV0xNDSkQIECdO3aleDg4Ez3mVF395kzZ5I/f34MDQ3x8PDI9lyyk5CQwNChQ7G0tERHR4caNWpw8eKb8ZlhYWF069YNCwsLdHV1cXBwUN2MSExMZMiQIVhZWaGjo4OdnR0zZszIVX0EIS/tuxmEh9dFxv1zE8/Tfpx+8IrgyHiuBIYx++Bdmiw4QY3fj9J33SWqzTzC1D0+BEXEZbq/q4FhDN18FUmCLpWsGVQnk5bK70XoIzgyDf4oC3+1lgP0f4elzaL9qUS9kLs9+50ALQPoti3rLsqWJeSW1wJl5GzoIffBd7/c3dn/pJwUresWuYUzJxpOkcesa+jIXamX1YDAc9lv9zbfg3D2dUK0qkOg6zb4yQ+GXJATw73v3ORmRcG4cM7Ll2ghZwIHuct4ZhnE31aylTyMQE1T7vI/x1HuJv7oeMY9Co7OeD22um7O9v8uTV35pgjIuQIKukCv3W/Gp7/NqKCcSwDS3izyPy33FDAqJH8OUhWtL593Krff067PC/pmco+FYdchf6m8rctHJAYC5Ubql1LoI0wKiSBdEPJSXHIclTfmTRen813Po6eZ/V3k0NBQDh48yLRp09DVTZv5tECBAnTr1o0tW7awdOlSrl27xu3bt9m4cSNqGSSbSQ1ct23bRmJiIj/99FOGx3w3wB0+fDjr169n8eLF/Pjjj5nW1c7OjgEDBjBu3DiaNGmSYR3e1a1bN0aPHs22bdtwd3cH5JsNXl5e9OrVCw0NDRITE5kyZQpOTk4EBwczYsQI3N3dc3wzYOvWrUyaNIklS5ZQs2ZN/vrrLxYuXEiRIjn8kZiBn376ie3bt7N27VpsbW2ZNWsWjRs35sGDB5iamvLLL7/g4+PD/v37MTc358GDB8TFyUHJwoUL2b17N1u3bsXGxobHjx/z+PHjD66LIOQVSZJYceIRM/bfzbasmgIKGOnwLCKeVaf8WHvWn9bOhehWxZYiFvoY6ci/iQJCYui79hLxSUrqOlkwpVXp77cFPSUJ9o58M04Y5K67ZsXkact29IcBp+Us5G+TJHm8tZo6VHz/3khp7Bosd3/Wt5C7Shd0zn4bE2vofxIin76eN/y+/G/EE3marELvMf+5QiG33BYqD1t7yUG/Z1N5WrNSbcHKOevkapHP5IRvAJUHQuNpOT/2x1R/EtjXhILlc54Er3gz6Po6m/grX3n8/I0tYGwj91pQvD5vKQVu/v36OBMz3192KvWTx8KbO8pdwt/tFv82197ylHHXN8ot5Zq6abu6v32OCgU0mgKezaB0Gyjf88PrKGRJBOm58VaQblzsdZAuxqQLgpCJ+/fvI0kSJUpkfNe5RIkShIWF8fLlS+7fvw9A8eLFs92nkZERVlZWWZZLpaenx6RJkxg/fjz9+vXD2Ng407ITJkzA09OTDRs20KNHj2z3bWpqSuvWrfH09FQF6ceOHePRo0f06dMHQPUvQJEiRVi4cCGVKlUiOjoaA4NMpoB5y4IFC+jTpw99+/YFYOrUqRw+fPiDW9NjYmL4888/8fLyws3NDYCVK1dy6NAhVq9ezejRowkMDMTFxQVXV1dAvoGRKjAwEAcHB2rUqIFCocDW1vaD6iEIeUmplJi69w5rTssZlLtUssZMXxvfF1HcD44mICQGfS0NajlZ0KCEJXUcLTHR0+S470uWHX/IuUehbLv8hG2X5XHLJnqaWOfT42VUAiExiZQuZMTiruW/zDHoEU/k7r7vBsfv68FheOEjt2DqGKVdlxgjB6UPDgEKKFoPnLu+GWu8qgG8uCl34e6x802365Qk2D1UDp5ADm4LunxY/UIevjl+r3/fr/VToZBbm40Ly5mzcyt/KXne9D0j5CzkZxbJD4MCcnI0p6ZQrH7aObCVKXLrc2yI3HW74a+Z7/9TU1OTxz6/r2INYHB9+abMtQ1wc7s8pv9mYPqyJVunnULufRkVlJOrKdSyv5FQtJ58syAiEG7vlKfQUwXpjdKXL+gC457I+/1eb7p9BiJIz418dvK/8eFYaMitKuFxORufJQjCx6Wrocv5rufz7NgfQ2q3eS0tLdXz7FqdJEl675YpDw8P5s2bx++//8706dMzLWdhYcGPP/7IxIkT6dSpU4733ahRIx48eECxYsVYs2YN1atXx8nJCYCrV68yefJkrl27RmhoKMrXXf0CAwMpWTL7OZDv3LnDgAED0iyrWrUqR48ezVH93vXw4UOSkpKoXv3NdC2amppUqlSJO3fuADBw4EDatWvHlStXaNSoEa1bt6ZaNXmeXHd3dxo2bIiTkxNNmjShefPmNGqUwY8aQfhCJSSnMGrrdfbckMfY/ty0BP1qFUlXRl2hSBdk13GypI6TJVcCw1hx/BEX/UMJiUkkPDaJ8NgIAAqZ6LKmV8U0c6B/Eg+PyF3JK/TOfmxxqlA/WFpF7tZrVU6ei9qpidyi+z7fqyEP5THaKYlyV+xG0+TEZAqFnBl9Qwd4eun1+F0v+Rhv6+AJy2vL3dBPzoPao+XAfpv7m2AJ4PRCueyHuLRG/tehUd53TwZ5nu22K+Xu4zf/lv9+0c/lKcuurAWjwlBloNxSq2MkJ5vzPyl30+/glXX29i+ZQgGFXeVH4+ny0Nl3k/Kpa0PZDrk/Vk7/H6ipQ4WecGQqXPYE60rylGdqmmBfO5NtvsAbbt8YEaTnhpY+GFpBVBBWKc8A0d1dEPKKQqHIUZfzvFSsmJwsycfHh9atW6dbf/fuXSwsLDAxMcHR0RGQg9KsphdzdHQkIiKCoKCgHLema2hoMHXqVNzd3RkyZEiWZUeOHMnSpUtZunRpjvbdoEEDbG1t8fLy4qeffuKff/5h8WJ5/GBMTAyNGjWiUaNGrF+/HgsLCwIDA2ncuDGJiXlzgzOzmyFv3/xwc3MjICCAvXv3cvjwYerXr8/gwYOZM2cO5cuXx8/Pj/3793P48GE6duxIgwYN+Pvvvz/7uQjC+5AkiUsBYcw5eI/zfqFoqiuY06EcrZwLpSurrZH1j/3yNvlY1kPu9hydkMyTsFgCQ2J5ERlP3eKWWBq95zjh95UQBZu7Q1IM+J2Up77KSRDne1AO0AGCrsuP4zPleaPbLJOzkmdHkuSpzFIS5Rb56Bfwz+v5sWuOhP1j5O7NOibyGHDrSun3Ye4gZz/fOQCOTYf8JeHUfHlqNA1dOUnY4UlysrewSW8aiXIqMRaurpefV+z7ftt+SgrF62zyreSM8/4n4d4B+Twjn4D3z3D8d3n9tdfTazWbl3kG9q+Npq6cvO5L4NxdHgv/+DycXiAvs60G2tn3cBM+DXEbJLded3m3SJKnORJTsAmCkBkzMzMaNmzI0qVLVWOaUz1//pwNGzaouok7OztTsmRJ5s6dq2ptflvqPOLt27dHS0uLWbNmZXjM1HLv6tChA6VKleLXX7PuMmhgYMAvv/zCtGnTiIzMJiMtcrDbu3dv1q5dqxpP37FjR0C+CfHq1StmzpxJzZo1KV68eJZJ4zJSokSJdFPDvfv6fRQrVgwtLa00U8clJSVx6dKlNMMSLCwscHd3Z/369SxYsIAVK1ao1hkZGdGpUydWrlzJli1b2L59e5os/YLwJQmJTmDliUc0mHecDsvOct4vFH0tdTzdK2UYoL8vA20NihcwolGpAvSoakfhfJ/h5untHXKADnKAt6F99hm0AR4dk/+tPhxaLZG7n2vqyVmxN3aCJ5ey2lp2dw88/E+exmvAKag7QU6M5n8S/mojB+hGhaDPwYwD9FTOXaBcF5CUsLmrHKDrmMgJv2oMl7skS0o4m7Mbpmnc/keey9zEVu5G/iXS0Ja7gzebA8NvQYuF8njqhEh5bLWkBOduUC5nvbqE92RkBU7ykC9V3oSMuroLn41oSc8tU3sIOI1J/BOgAJGiJV0QhCwsXryYatWq0bhxY6ZOnYq9vT23b99m9OjRODo6qrKxKxQKPD09adCgAbVq1WL8+PEUL16c6Oho/v33X7y9vTl+/DjW1tbMnz+fIUOGEBkZSc+ePbGzs+PJkyesW7cOAwODdNOwpZo5cyaNGzfOts7/+9//mD9/Pps2baJy5eyT8/Xu3ZvffvuN8ePH07lzZ/T19QGwsbFBS0uLRYsWMWDAAG7dusWUKVPe492DYcOG0atXL1xdXalRowYbNmzg9u3bOUocd+/evXTLSpYsycCBAxk9ejSmpqbY2Ngwa9YsYmNjVVPGTZw4kQoVKlCqVCkSEhLYs2ePKoCfP38+VlZWODs7o6amxrZt2yhQoEC6hH2C8LmdefiK8f/cJCQmEU11NTTUFGiqqxEcFU9SityDRFdTnRblrPhfrSIUszTM4xrnwtXXrawlW8GD/+Ru415Nodt2MMyf8TYpyeD/+uZcqdbyOFuX7nI3802d5X2sbwfue6FA6Yz3kRgD+8fKz6sPk1vA85eU58c+MBbu7ZOnBuvxT84yiDedIwfnIQ/kwL77P2D5Oi9JtaFyl/Crf8kt6xll6s7MxVXyv659ct4FOi9p6kCFXuDSQ+7uf36ZPIWZW8Y3o4WPxLW3fNMplQjS85QI0nPrdUu6YcxjwJVwEaQLgpAFBwcHLl68yOTJk+nYsSPBwcFIkkTbtm3566+/0NN70+pUqVIlLl26xLRp0+jXrx+vXr3CysqKatWqsWDBAlW5QYMG4ejoyJw5c2jTpg1xcXHY2dnRvHlzRo4cmWld6tWrR7169fD29s60DMhjtKdMmULXrl1zdI42NjY0aNAAb2/vNIniLCws8PLyYvz48SxcuJDy5cszZ84cWrZsmaP9AnTq1ImHDx8yZswY4uPjadeuHQMHDuTgwYPZbtu5c+d0y/z8/Jg5cyZKpZIePXoQFRWFq6srBw8eJF++fICcI2DcuHH4+/ujq6tLzZo12bx5MyD3NPj999+5f/8+6urqVKxYkX379uUoG74gfCq3nkbwv3WXiU5IznB92cLGdK5oQ4tyVhjqaGZY5rNRpuQsuVVmXj2Ax+fkfTT5Xe5uvqG9nMV8dUPouTPjKcKeXYHEKNDNJyciS6WlD503ya3gTy7I//beL8+R/a4Tc+Ru2cY2UOOt79p8ttBlE7z0BRObnE8Lpm0gB+Y3toJLNzn5V6oideSp0J7flIPu2u/M6PHqPoQHyFNkvf1ePr0Mz67K45xdsk8A+kVRU5PH7787hl/4NIrUk3tbhAfI/5o75HWNvmsK6X0m+P0GREZGYmxsTEREBEZGRtlvkJ3bO2CbO8mFKlLs4QgA7k9zQ/NLzGAqCN+I+Ph4/Pz8sLe3R0fnE491/AwmTZrEvHnz8Pb2pmrVqnldHeETyepz+9GvTcJ3+54GhMTQ7s8zvIpOpEoRU6a2Lo1SgqQUJckpEka6mtib6+d1NeWx3Je95LHW1pXlaaLUP+CGweFf4dQ8udWv2zZ5WegjObgO85fndHbfk36747Pg6LQ3c1i/Ky4c1jaXg2KjwtDngDwdWapX92FpVXk+684b5Sm2PrUb2+Tx7nrmMOKWPKY5dfmuwfKc2JX6Q5OZbxJ77Rwkj+cu2xnaLv/0dRS+bmeXwsFx8hCQvMyg/416n+tSnkaSJ06coEWLFhQsWBCFQsHOnTuz3WbDhg2UK1cOPT09rKys6N27NyEhIZ++spl5fXdWPdxftUgkjxME4X38+uuvLFy4kPPnz2c4/lwQBCEnXkYl0GP1BV5FJ1LCyogVPV0pZmmIY35DShU0ppy1yZcRoEc+k7uS7xkO8RFyl+bDk99/P8oUuC73asG525vlpkWg5y5AIY8ND/NPv23qePTMphTTNYHuO8DMQW4tX1pFDvyPzZS7ne/7UQ7QHRrJU4Z9DqVay0ntYl/B9U2gVMJ/v8mBe0qCXObCctjeR07EFhsKt7bLy7+khHHCl6vKQPA4DHXH53VNvnt5GqTHxMRQrlw5Vebf7Jw6dYqePXvi4eHB7du32bZtGxcvXlTNl5sn8tkDoIh5iZWOnJ1YJI8TBOF99e7dm+HDh4tu0oIgfJCo+CTcPS8QGBqLtakua3tXxCivu7K/S5Lg+hY54H34n5xgrXxPed3ZxXLvxPfx8ChEPZO7rKcmvUqVzw6KvJ4+6sbWtOsSouHxBfl5VvN+G1jIwb65IyRGy8H5sRlysP7omNyF3O33zzdXtLomVBkkPz+zGLb2gJOvc47UGCFPaaamKb+PG9rLY7mT4+Xu/IVdP08dha+bQgHWFb/eKe6+IXk6Jt3NzQ03N7fsC7527tw57OzsGDp0KAD29vb0798/06zGn4WOEehbQMxLSuiEEBRvJVrSBUEQBEH4bJRKiUEbrnD7WSRm+lqs61P500979iH2j5FbegEKVYDWy8DCUQ6yT/8BOweDZUmwcMrZ/q69nlasTMeMg4pyXeRg+vomqDX6TTAdeFZuBTexUTW2ZMq4EAw6By9uy9NTPb4g/xseAPV+zni8+6dUvqc8TVzoQ/mhrg0tF73Jeq5vDlt6yInv/E7Iyyr2/Xw3EgRB+Ci+qiabatWq8eTJE/bt24ckSbx48YK///6bZs0yHweUkJBAZGRkmsdH9/oL2lHjJQARcXkz368gCIIgCN+f1af8OHn/Fbqa6nj2rvhldGl/l//p1wG6AupNgD7ecoAOUG+iPHY8KQa2dJfnPc9ObCjc3Ss/d+mWcZnizUFTXx6j/uTim+WpXd3ta+cseFVTB6uyUKkftFsJw2/AhGA5o/vnpm0AFfvJz/Ut5ezzb09LVrSePAZfz/x1eWMo0/7z11MQhFz56oL0DRs20KlTJ7S0tFTT3CxatCjTbWbMmIGxsbHqYW1tnWnZD/Y6SLdXewGIMemCIAiCIHwed59HMvugPL3gxBYlKVvYJG8rlJGUJHkMN8hTa9UaDepvdeZU14D2a8DQSp5XfOcgeHol7SPmVdp93toOKYmQvwxYlcv4uNoGUPL17BHXNr5Znt149JzIy+7AdcZBe095XnbriunXF3QBD29wbAJNZsgZ6wVB+Kp8VUG6j48PQ4cOZeLEiVy+fJkDBw7g5+fHgAEDMt1m3LhxREREqB6PHz/++BV73VXKmueAGJMuCIIgCMKnl5Ccwogt10lMUVK/uCWdK36ChoiP4cJKCPaRu7XXn5RxGQNLOcu6mibc2Q0r66Z9zHGEjZ3AZzckJ8LV113dnbOZGrLc66kXb/8DSfEQ/RJe3JKX2df+OOf3ualrQOm2mc8BD2BWFLpuybyXgSAIX7Svap70GTNmUL16dUaPHg1A2bJl0dfXp2bNmkydOhUrK6t022hra6Ot/Ynvdr5uSbdKCQJES7ogCIIgCJ/e/EP3uRMUiam+FjPblUXxJY47jnoOR6fLzxtMBj3TzMtaV4I2y+TkbEnxb5ZLKRAVBL4H5IduPogLAzUNKNsx6+Pb1ZKnUIt8Im+rfD13fP4ycmI4QRCEL9BXFaTHxsaioZG2yurq6gDkxXTvUYlRvIp7ha6eCQUA88SngGhJFwRBEATh07roH8ryEw8BmN6mDBaGX2g2Zu8JkBglJ4pz6Zl9+TLtMx5D/fKe3GX9+maIlnsu4thETpSWFTU1OZA/NU/eNrV8ka+0FV0QhO9CnnZ3j46O5tq1a1y7dg0APz8/rl27RmBgICB3Ve/Z880XeosWLfjnn3/4888/efToEadPn2bo0KFUqlSJggULfvb677i/g5Y7WzIvUE5cYpj0El3iiRQt6YIgCIIgfCKR8UmM3HoNSYL2FQrTpHSBvK5SxvxOws1tgAKazZUD5g9l4QQNf4URt6Hb31BjJLjlcHaf1C7vDw7JremQu/HogiAIn1ieBumXLl3CxcUFFxcXAEaOHImLiwsTJ04EICgoSBWwA7i7uzNv3jwWL15M6dKl6dChA05OTvzzzz95Un8tdS0AkhQK0DEBwEYRTLgI0gVBED7IsWPHUCgUhIeHA+Dl5YWJiUmW20yePBlnZ+dPXjdByGsvoxKY632P2rOO8jg0jkImukxqUTLnO7jzL+wYCFc3pE/E9rG9nSzOtY+czOxjUNcAh4bQYJI8PVpOWDhBwfJyV/eYl/K4d5uqH6c+giAIn0CeBul16tRBkqR0Dy8vL0D+cXbs2LE02/zwww/cvn2b2NhYnj17xvr16ylUKIdf0h9ZapCeqExUjUu3U7wQY9IFQcjS48eP8fDwoGDBgmhpaWFra8uwYcMICQlJV/bBgwf07t2bwoULo62tjb29PV26dOHSpUtpyh09epSmTZtiZmaGnp4eJUuWZNSoUTx9Kg/DSQ1+S5cuTUpKSpptTUxMVN+7AHZ2digUCs6dO5em3PDhw6lTp06G53T58mUUCgWnTp3KcH3jxo1p2bJldm9NOp06dcLX1/e9t3ubv78/CoVC1WtLEL42919E8dPf16k+8wiLjjwgLDYJa1NdlnYrj6GOZs52kpwAuwbD9Y2waxDMLgarGsLJeRB8Bz72sMETs+HlXdAzk6dcy2vlurx5bl1JzvwuCILwhfqqsrt/aTTV5AtjYsqbIN1W8VwE6YIgZOrRo0e4urri6+vLpk2bePDgAcuWLeO///6jatWqhIaGqspeunSJChUq4Ovry/Lly/Hx8WHHjh0UL16cUaNGqcotX76cBg0aUKBAAbZv346Pjw/Lli0jIiKCuXPnpjn+w4cPWbduXbb11NHRYcyYMTk+rwoVKlCuXDk8PT3TrXv8+DGHDx/Gw8Mjx/tLpauri6Wl5XtvJwjfgqfhcYzaep1GC06w9dITElOUuNiYsLRbeY79WJdy1iZvCt/aLmdRzyzYvn8I4iPkpGsFygISPLkA//0KS6vAH+Vg/1h5erKUXP6O8TsBx193RXeblXWyuM+ldDu5BR1EV3dBEL54IkjPBVVLekralnSROE4QPj9JklDGxubJ430SVw4ePBgtLS28vb2pXbs2NjY2uLm5cfjwYZ4+fcrPP/+sOh93d3ccHBw4efIkzZo1o2jRojg7OzNp0iR27doFwJMnTxg6dChDhw5lzZo11KlTBzs7O2rVqsWqVatUw4dS/fDDD0yaNIn4+Ph0dXtb//79OXfuHPv27cvxuXl4eLB161ZiYmLSLPfy8sLCwoJmzZqxfv16XF1dMTQ0pECBAnTt2pXg4OBM95lRd/eZM2eSP39+DA0N8fDwyPZcspOQkMDQoUOxtLRER0eHGjVqcPHiRdX6sLAwunXrhoWFBbq6ujg4OKhuRiQmJjJkyBCsrKzQ0dHBzs6OGTNm5Ko+ghARm8SMfXeoO+cY2688QZKgUcn8bB9YlR2DqtO0jBXqam9lco8Nhe395O7lgWcz3umNLfK/Lt1hwEl5bHezuVCsIahrQ3gAnP8T1rWSW9mPz4KE6PevfMwruS5I8rEySgKXF/TN5DnaNfWhVNu8ro0gCEKWvqrs7l8aLbXXY9KVSaog3Ubxgsi4JCRJ+jKnQhGEb5QUF8e98hXy5NhOVy6j0NPLtlxoaCgHDx5k2rRp6OrqpllXoEABunXrxpYtW1i6dCnXrl3j9u3bbNy4EbUMki2lBq7btm0jMTGRn376KcNjvhvgDh8+nPXr17N48WJ+/PHHTOtqZ2fHgAEDGDduHE2aNMmwDu/q1q0bo0ePZtu2bbi7uwOohjD16tULDQ0NEhMTmTJlCk5OTgQHBzNixAjc3d1zfDNg69atTJo0iSVLllCzZk3++usvFi5cSJEiRXK0fUZ++ukntm/fztq1a7G1tWXWrFk0btyYBw8eYGpqyi+//IKPjw/79+/H3NycBw8eEBcXB8DChQvZvXs3W7duxcbGhsePH/P48eMProsg/Hv9GRN23lL1yqtsb8q4piVwfrvV/F33D8nTlAGcXwa21dKujwt/kzCtzOspy4wLQ8W+8iMhWm5Bv7cf7h+Ux20fnQbnl0OtH+Ux5Ro5yB6vVMKOAXL2dXOnnCd2+1yazpEf4veZIAhfONGSngsZtqSrvSAxRUlcUkpWmwqC8B26f/8+kiRRokSJDNeXKFGCsLAwXr58yf379wEoXrx4tvs0MjLCysoqR3XQ09Nj0qRJzJgxg4iIiCzLTpgwAT8/PzZs2JCjfZuamtK6des0Xd6PHTvGo0eP6NOnDwB9+vTBzc2NIkWKUKVKFRYuXMj+/fuJjs5Zi92CBQvo06cPffv2xcnJialTp1Ky5HskznpHTEwMf/75J7Nnz8bNzY2SJUuycuVKdHV1Wb16NQCBgYG4uLjg6uqKnZ0dDRo0oEWLFqp1Dg4O1KhRA1tbW2rUqEGXLl2yOqQgZGrvjSCGbb5KRFwSTvkNWePuyub/Vck6QAfw3f/m+Z09EPEk7fo7uyElESxKQIEy6bfXNoASzaH1EhjlC+3XgGlRiH0FB8bCogpwe0f2J3BuiZxBXV0bOniCln7223xOCoUI0L8gkiQx5ewUtt7bmtdVEYQvjmhJz4WMEscVJAQtkoiIS0JPS7y9gvC5KHR1cbpyOc+O/TGkdpvX0tJSPc+uR86H9Nrx8PBg3rx5/P7770yfPj3TchYWFvz4449MnDiRTp065XjfjRo14sGDBxQrVow1a9ZQvXp1nJycALh69SqTJ0/m2rVrhIaGolQqATnYzUmwfefOHQYMGJBmWdWqVTl69GiO6veuhw8fkpSURPXq1VXLNDU1qVSpEnfu3AFg4MCBtGvXjitXrtCoUSNat25NtWpyS6W7uzsNGzbEycmJJk2a0Lx5cxo1avRBdRG+b0fuvmDY5qsoJejoWpgZbcum7dKemeREePCf/NywIEQ9g4ur5eznqW68DoLKdsw+SFVTk8dvl2gJV9fD8d8h4jFsc4fol1D5fxlv9/QyHJ4sP28yA/KXyr7uwhcvIiGCKeemEJ0UzYjyI3Aydfoo+5UkifmX57PVdytqCjUqFqiIvbH9R9m3IHwLREt6LqRJHKdvDlqGqCkkrBXBYly6IHxmCoUCNT29PHnkNEguVqwYCoUCHx+fDNffvXsXCwsLTExMcHR0BFAFiplxdHQkIiKCoKCgHL9XGhoaTJ06lT/++INnz55lWXbkyJHExcWxdOnSHO27QYMG2Nra4uXlRWRkJP/8848qYVxMTAyNGjXCwMCA9evXc/HiRXbskFvnEhMTc1z/jymzmyFv3/xwc3MjICCA4cOH8+zZM+rXr68aKlC+fHn8/PyYMmUKcXFxdOzYkfbtv5AxuF+YpUuXYm9vj46ODhUqVODkyZNZlt+wYQPlypVDT08PKysrevfuneEMCN+CMw9eMWD9FZKVEi3KFcx5gA4QeAYSIkHfQg6OAa6shaTXuRoinoD/6/f6fcaHq2uCa28YehWqDJKX7R8NZ5ekL3vvAGzsLE9xVqKl3D1e+OoFRgbSfV93Dvof5PTT03Ta04m5l+YSmxSb632vvrUaz9tyr6tJVSeJAF0Q3iGC9FxQzZOekiTfmTaVv2BsFS+4FBCWl1UTBOELZGZmRsOGDVm6dKlqTHOq58+fs2HDBtVYbmdnZ0qWLMncuXNVrc1vS51HvH379mhpaTFrVsZjP1PLvatDhw6UKlWKX3/9Ncs6GxgY8MsvvzBt2jQiIyOzPkHkYLd3796sXbtWNZ6+Y0d5DOzdu3d59eoVM2fOpGbNmhQvXjzLpHEZKVGiRLqp4d59/T6KFSuGlpZWmqnjkpKSuHTpUpphCRYWFri7u7N+/XoWLFjAihUrVOuMjIzo1KkTK1euZMuWLWzfvj1Nln4BtmzZwvDhw/n555+5evUqNWvWxM3NjcDAwAzLnzp1ip49e+Lh4cHt27fZtm0bFy9epG/fvp+55p/e5YBQ+q67RGKykoYl8zOvY7mcB+ggB8gADo2heHMwtobYEDnbO8DNv+V/bauDic37V1BTFxpPh5qvZ5Q4OB5OLZCfx0fK07pt6gQxwWBZElouEl3KvwEXn1+k676u+Ef6U0C/AHWt65IipeB124vWu1pzNPBopklTJUkiMDKQA/4HuBNyJ125zXc388eVPwD40fVH2jqIRH6C8C7RHzsXUhPHJSpftwCZFoHnN7BTvMDzlB/dKtmg9j4XWkEQvnmLFy+mWrVqNG7cmKlTp2Jvb8/t27cZPXo0jo6OqmzsCoUCT09PGjRoQK1atRg/fjzFixcnOjqaf//9F29vb44fP461tTXz589nyJAhREZG0rNnT+zs7Hjy5Anr1q3DwMAg3TRsqWbOnEnjxo2zrfP//vc/5s+fz6ZNm6hcuXK25Xv37s1vv/3G+PHj6dy5M/r68rhUGxsbtLS0WLRoEQMGDODWrVtMmTLlPd49GDZsGL169cLV1ZUaNWqwYcMGbt++naPEcffu3Uu3rGTJkgwcOJDRo0djamqKjY0Ns2bNIjY2VtUDYOLEiVSoUIFSpUqRkJDAnj17VAH8/PnzsbKywtnZGTU1NbZt20aBAgXSJez73s2bNw8PDw9VkL1gwQIOHjzIn3/+mWE2/HPnzmFnZ8fQoUMBsLe3p3///pnejPpaXXscjrvnRWITU6jpYM7iri5oqmfSfiJJ6YNfSXozHt2pCahrQEUPudv5+WXg3DVtV/cPpVBAvV/kKcyOz4TDk+Rs8PcPyV3hUUDVwfJ86JofZ/iPkHd23N/Bb+d+I1mZTBnzMiystxBzXXNOPDnB9PPTeRr9lKFHh2KoZYidkR12RnbYGtmirqbO9ZfXufHyBqHxb25UOuZzpFXRVjQr0owzz84w7fw0APqX7U+vUr3y6jQF4YsmgvRcSJM4DlTj0h00g1nzKoYjd4NpUDJ/XlVPEIQvkIODAxcvXmTy5Ml07NiR4OBgJEmibdu2/PXXX+i9lSW+UqVKXLp0iWnTptGvXz9evXqFlZUV1apVY8GCBapygwYNwtHRkTlz5tCmTRvi4uKws7OjefPmjBw5MtO61KtXj3r16uHt7Z1lnTU1NZkyZQpdu3bN0Tna2NjQoEEDvL29VQnjQG6N9vLyYvz48SxcuJDy5cszZ84cWrZsmaP9AnTq1ImHDx8yZswY4uPjadeuHQMHDuTgwYPZbtu5c+d0y/z8/Jg5cyZKpZIePXoQFRWFq6srBw8eJF++fICcI2DcuHH4+/ujq6tLzZo12bx5MyD3NPj999+5f/8+6urqVKxYkX379uUoG/73IjExkcuXLzN27Ng0yxs1asSZM2cy3KZatWr8/PPP7Nu3Dzc3N4KDg/n7779p1qxZpsdJSEggISFB9TonPT/y0iX/UNw9LxKdkEwlO1NW9HBFW0M9fcHEWFhZDwwsoPsOORBP9fIehPmDuhYUqSsvK98Ljs2E5zfg0moIvi2vL9kqdxVWKKDuOFDTgKNT4dIaeXk+O2j9Z/qM8sJXIS45jjshd7j16ha3Q25zO+Q2AZEBADS2a8zU6lPR0dABoFbhWlQsUJEVN1aw9vZaohKjuPnqJjdf3Uy3X001TYqZFONh+EN8w3yZfWk28y7PU63vVqIbg50Hf56TFISvkEJ6nwl+vwGRkZEYGxsTERGBkZFRrvb1POY5Df9uiIaaBld7XIUrf8HuITwyrkK9F0OpbG/Klv5VP1LNBUFIFR8fj5+fn2p869du0qRJzJs3D29vb6pWFd8Z36qsPrcf89r0pXn27BmFChXi9OnTqoR7ANOnT2ft2rUZ9nAA+Pvvv+nduzfx8fEkJyfTsmVL/v77bzQ1NTMsP3ny5AyHb3yJ7+m5RyH08ZJb0KsUMWV1r4roa2fSbvLgMKxvJz9vMhOqDHyz7tR8udW8WAPovv3N8l1D4Opfcsu3MknuBt85Z7M05MjphXJCubKdoOFvcnZ44ashSRLXX15n+/3tHPQ/SFxy2uFX6gp1+pXtx8ByA1FTZHzDMT45nsdRj/GP9CcgMgD/CH8SUhIobV6achblKGlWEi11LSISIjjof5BdD3Zx49UNAFoWbcmU6lMy3bcgfKve51ovWtJzITVxXLIyGaWkRO31mHQbnqOhpuC8Xyg3n0RQprBxXlZTEIQv3K+//oqdnR3nz5+ncuXKohVW+CZllZzvXT4+PgwdOpSJEyfSuHFjgoKCGD16NAMGDFBNjfeucePGpek5EhkZibW19cc7gY/k1P1X9F13kfgkJTUdzFnRwxVdrQxa0FMFvpVz4eh0KNUGDAvIr1PHozs2SbtN5f5ykK58ncS2bM5mZ8ix6kOh2g9i7PlXJioxip0PdvLP/X94EP5AtdxC14JS5qUobVaa0ualKWVWChMdkyz3paOhg0M+BxzyOWRZzljbmI5OHeno1JFH4Y/wi/SjduHaIkAXhGyIID0XUru7AyQpk9B+3d1dI/IxLctY8M/1YFadesQfnV3yqoqCIHwlevfunddVEIRPwtzcHHV1dZ4/f55meXBwMPnzZzwkbMaMGVSvXp3Ro0cDULZsWfT19alZsyZTp07Fysoq3Tba2tpoa2t//BP4iI77vqTf6yRxdZ0s+LN7BXQ0swjQAQLOyv+qa8tZ3A9NhLYrICYEnlyQ170bpBcoIyeKCzgNOsbg8AmmBRQB+meTpEwiMiH98A09TT101HWynWFEKSnZ/XA38y/PV40V11HXobFdY9o5tsPZwvm9p/L8EEVMilDEJPv8IYIgiCA9V94O0hNTEtE2KAAaupAcxyjz8+yhMHtvBDGmSXEKmohEKoIgCML3R0tLiwoVKnDo0CHatGmjWn7o0CFatcp4nHRsbCwaGml/oqiry8Hs1zpKLyE5hVFbr6myuC/u6pLxGPS3JSfA00vy85aLYEd/uLEFyveE8McgKSF/GTDJoMdArR9h/Tlw9QDNr39Y0LcmWZnMzVc3OR90HkMtQ2oXrk1hw8JpygREBrD13lZ2PthJZGLGORY0FBroa+ljoGlAfr38VLGqQvVC1SllVgp1NXXuht5l2rlpXHt5DQA7Izu6l+hO0yJNMdQy/NSnKQjCBxJBei6kdneH18njtAyhoAsEnqHQ6Z+5qGfMusS6/HNUjyFt6uRZPQVBEAQhL40cOZIePXrg6upK1apVWbFiBYGBgQwYMACQu6o/ffqUdevWAdCiRQv69evHn3/+qeruPnz4cCpVqkTBggXz8lQ+2L/Xg3gVnYiVsQ5LupZHSyMH3X2DrkNyPOiZydnZA8/CZU/Y+yOYFZXLODXJeNui9WBsAGjqf7yTEHIlJimGA34HOP3sNOeenSMqKUq1buaFmRQ1Lkpt69oUMS7Cnkd7OBeU/fSSyVIyEQkRRCRE8DT6KVeCr7D0+lKMtY0paVqS88/Po5SU6GroMrDcQLqX6I6mesZ5HQRB+HKIID0X1BRqaKhpkKxMJil13FfnDXI21YtrMI56xg8aO0m+vpsktV5oNp0p7mYLwkf0tbaoCd+n7/nz2qlTJ0JCQvjtt98ICgqidOnS7Nu3D1tbWwCCgoLSzJnu7u5OVFQUixcvZtSoUZiYmFCvXj1+//33vDqFXJEkCc/TfgD0qGqbswAdIOB19nubqnL38voTwWcXvLwjPwCc3DLfXlu0lH4pXsW9wuOgB48iHqmWGWsbU8WqCqHxoVx5cYWHEQ95GPFQtV6BgpqFa9LJqRPVC1ZHXe1NzwtJkohLjiMqMYqYpBiikqJ4GP6QU09Pce7ZOSISIjgbJA+VaGLXhFGuoyigX+DznbAgCLkisrvnUuUNlYlNjmVvm73YGNm8WZGShPLuXq7/MxuXlFvysgJloeM6eJ1gThCED5OSkoKvry+WlpaYmZnldXUEIUdCQkIIDg7G0dFR1XU71bec3T2vfEnv6QW/UDouP4uOphpnx9Ynn/6b4XKkJINCDTJKGLmxszwPeqNpUG2IvOzKOtj9g/zcID+MvJvxtsJnFZkYSXRiNAUN0vf0CIkLweOgBw8jHmKpa0l7x/ZpuqQDRCREcPrpaY49OcbD8IfULFST9o7t03WBz4nUrvTXg69T2rw0rgVcc31+giDknsju/hlpqWsRmxz7Zq70VOqaqJVqjU+0C/N2beQPzSWYPr8BK2pDm+VZ3/kWBCFL6urqmJiYEBwcDICent5nSXojCB9CkiRiY2MJDg7GxMQkXYAufPvWnJJb0du4FE4boIc8hOW1oWRLaL007UZKpdy9HeSW9FTO3eHyWnmsumNjEaC/h6SUJELiQz5qi7IkSex4sIM5l+YQnRhNy6It+cHlB/Lry0kRQ+ND6evdVw7Q9SzxauyFtVH6HALG2sY0LdKUpkWa5rpOGmoauFi64GIpEhcLwtdKBOm5pKUmX2wTlYkZru9QwZqDt+vTzLcQS7UW4hJ/HzZ1Btc+oKUPYf4QFgDhAVCwvNxdXkuMHxOE7BQoIP/ISg3UBeFLZ2JiovrcCt+Px6GxePvIme17V7dLu/LKWkiMgmsboc5YMHmrR97LuxAfDpp6YFX2zXI1NejgCeeXQ9Uhn7z+34oUZQoDDg/gwvMLuFi60LVEV+rb1E+TX+h9BUQG8OvZX7n4/KJq2a6Hu/AO8Ma9lDuti7VmyJEhPAh/gKWuJWsar8kwQBcEQXiXCNJzKTX5RrqW9Ne0NNRY1dOVEVs06HjzF8ZrbKS3xgG4tCZ94UdHYWtP6LIZMkrqkZIMKYmgpfcxT0EQvkoKhQIrKyssLS1JSkrK6+oIQpY0NTVFC/p36q9zASglqFHMHMf8b40RVyrh5vbXLyS5G3u9CW/Wp7aiF3ZN/5vAxAYaT/uk9f7WeN324sJzecq6q8FXuRp8lfx6+elcvDMdHDtgrG2c430lKZPwuuXFsuvLSFQmoqOuwxCXIZSzKMfcS3O59vIaf17/k2XXlyEhYaFrwarGq7A1sv1Up/fVkpRKUChEbzhBeIcI0nMpdRo2VeK4jMpoqLGwiwvjtTX49VJPzipLMMb6DkXti4CJLeSzA2UybO8LDw7DzkFyl/i3u7DdPwS7hsjTrfQ58CarqyB859TV1UXwIwjCFyk2MZnNF+SEeOla0R+fh8gnb15f+Qtqj3kTkKu6ulf79BX9xt0Nvcvia4sBGFVhFDHJMWy9t5UXsS/448of/OXzF9NrTKd6oerZ7uvyi8tMPTeVB+EPAKhWsBq/VPlFNXZ8nds6DgUcYt7leTyNfoqZjhmrGq/C3vjz5iNKevqUAPfeGDVtiuWI4Z/12DklJSfz7KcxaNnbY/GD6BUiCG8TQXouqbq7Z9KSnkpdTcHMdmUw1NFg1SnwDqhIz4K2THAt+SbLa8d1clf4m1tB3xwaT4ekWPCekLblfUt36HtYdIsXBEEQhC/Y9itPiYxPxs5Mj7pOlmlX3vpb/rd0e/A7DtHPwfcAlGghLw98Pf2WbVWED5eQksC4k+NIViZT17ouvUr1QqFQ0K9MPw76H2TVzVU8injEgMMDcC/lzlCXoRlOURYWH8b8y/PZ8WAHAPm08zG64miaF2mephVYoVDQyK4RdazrcOzxMcpalM2TrOphW7eR9PgxIStXYtyqJdpFinz2OmRFSkri6agfifL2Bk1NjJo1Q7uISKwsCKlEtpFcSm1Jzy5IB/mL++dmJRjd2AmAdWcD6Lj8LM/C4+QCjo3eJI45txT2jYZlNd4E6BX7yplcg31g91D4vhLzC4IgCMJXQ6mU8Ho97Zp7NTvU1N7qzpuSBLd3ys+du4BzN/n5JU/53/DHEPEYFOpQSGTmzgmlpCRZmZxu+aIri3gQ/gBTHVMmVZ2kCqi11LVoUbQFW5pvoZNTJ0DuEt9zf08eRz4mJimGe6H3+C/wP1beWEnLnS1VAXo7h3b82+ZfWhRtkWk3bS11LRrZNcqTAF2SJCL375dfKJW8WrI06w0+kaSgIFKiY9ItlxITeTJiBFHe3ig0NSn8xx8iQBeEd4iW9FxKTTiSWeK4dykUCgbXLUYJK0NGbLnOtcfhNF90ioWdXajhYA7lOkNsCBwcDxdXyhsZFZKD9yJ1oHQ7WNtCvgNf2BWqDHyz84gncHQGhPlBy0WiS7wgCIIg5JGTD17x8GUMhtoatHd9J1nYo+MQ+wr0zMG+DpgWgdML4OEROaHs49eJyKzKgbbB5634V+h5zHOGHhmKX4QfNQrVoJ5NPWpb1+Ze6D3W+awD4Ndqv2Kmm37KTh0NHSZUmUDVglWZeHoit0Ju0Xxnc5SSMl1Zh3wOTKwyEWdL5099SrkSf9uHpMBAFJqaSElJRO7bh1n//6Hj6PjZ6hB75QoBPXuhpq2NSYcOmPbsgWbBgigTE3k6dBjRx46h0NKi8OJFGNSq9dnqJQhfCxGk59L7tKS/rV7x/Oz5oQYD1l/m9rNIeqw5T88qtjQpbYVrpYFoxoXDyTlQpgO4zQJdE3lD22ryfKkHxsDBn+W5163KwqkFcHYxJMfL5TzdoOcusCzx0c5VEARBEIScSW1F7+BqjYH2Oz+3Uru6l2oN6hpykF6krpxA9so6iAuT19uIru7Z8Q3zZeDhgQTHyjN9HA48zOHAw2goNNDR0EFCop1DO+pY18lyP/Vt6lPKrBRjTozhSvAVQJ4WrbBBYQoZFMK1gCvtHdvnKhv85xK5fx8ABvXqgSQR5e3NqyVLKfzHgs9Wh5cLF0FyMsrkZEK9vAj96y+M3NxICQsj5vRpFNraFF6yBIMa2ecBEITvkQjScyknieMyY22qx/aB1Zi8+zabLz5m7dkA1p4NwEBbg+rFGlO/UQfql7HFTFc77YaV+8vzo97cJmeDV6hBzOtpqGyqQUIkvLgFnk2hxw4o6Pxm24RouLQaHl+QE9S8Pa2LIAiCIAi55vcqhqP3XqJQQM+q72T0ToqDO3vk52U6vFnu2lsO0q+uB20jeZkYj56lc0HnGHF0BNFJ0f9n777Do6q2Bg7/pieTXkhIIPTQOyhdmoCIIoJSvIgI2LABVvRaLhawgCiKlYAFEBBBPkUUQToqRUAE6ZBOek+mnu+PkwzEBAgwYUhY7/PMg3PmlD1DcLLOWnttGgQ04JnrnmFXyi7Wx67naNZR8mx51PatzdPXPV2h89X0qcmCmxaQkJdAoCkQX2PVq2JQFIXcH9cA4D9wIMYG9cldu5bcn36i6J9/8GratNLHULB7NwW//QYGAxGvTCN75XcU/PYbOd+rP/cab2+iPpyLT+fOlT4WIaoqCdIvU0Ubx52Ll0HHjGGt6dc8nB/2JbHxcCrp+VZ++vs0P/19Gt3/HeWG6FBua1uLfs3D8THpQaOBW9+F0wcg5W/1RMENoN8r0HSQegf+q2GQuBs+Hwyjv4Gw5mr5/LY5ajk9qGV1t38EzW+7vA9BUSDlIPzzA/zzf5B+DHzDwC8S/CPAP1It1W/QWx27EEIIUY19sf0kAL2bhFEv9F9NXo/8rK6NHhAFta8/s73JzeATBnmn1QdA1LUXxCiKQmphKj4GH8x6c7lzvhVF4fvj3/PithexO+10CO/Au73fJcAUQNdaXXm03aOczD7JH8l/0DWyK2ZDxZeu1Wg0rk7tVVHR3r3YEhPRmM349rwBrbc3/gMHkrN6Nalz3ifqg/crfQxpcz8EIHDIENej8O+/yZi/gKJ/DlLzxRfxuf76C5xFiGubBOmX6ULrpFdU32bh9G0WjtOp8FdCNhsOpbL2YDL7E3L49VAqvx5Kxdug44bGobSuHUjLWgG0HryAoO3TIaoTdBwHevWGAeZgtdR90XB1CZcvhoDeBIUZ6uvBDcC3JsRuUzPxvabCDU+XXvINoDBLneeekwg5Ceqflhw1c49GDbjtRWqwn3G89LEZeaW3bX1X/WWk91QJ1oUQQlRbeRY73+xUl1a7p2u9sjv8tUz9s+XQ0t+7OgO0Gw1bZqnPQ6LBt0blDvYqklaYxndHv+PbI98Sm6suW2fQGgg0BRJgCkCr0ZJnzSPPlke+LR+H4gDgpno38Vr311yVjSXqBdSjXkC9K/02PK6kYZxf795ovb0BCH3kYXLWrCFv3ToK9/+Nd8sWlXb9wn37yN+yBXQ6Qu6/z7Xdu0ULar39VqVdV4jqRoL0y+TKpFewcdyFaLUa2kQF0iYqkMdvjOZoSh6r9iTw3d5ETqUXuDLsJSIDRtMkz49G6UeJDvOjUbgvjcP98PXyh9HL4eu74PgGsOVDcEO44akz5XVrX1C7yG+YrnaMv+FpSNilrt0au71s4H0+OqMafDe7Rb1pUJBeHNwnQvoR2LcU4v+AL29XMwO9nlWz61dDsO50wu8fgtYA1993dYxJCCFElfTt7nhyLXYahJrpofkL0uueaeRalA2Hf1b/u+UdZQ/ucA9seQdQoE71y6L/fPJnfo37FR+DD4GmQAJNgZgNZjbFb2Jj3EbsSunu7DanjdTCVFILU8ucS6/Rc0+Le3is/WNoNbJYEYDidJKz5icA/G8e6NpuatCAgFtvIfu7VaTOeY86H39caWMoyaIHDB6MMSrqAnsLIc5FgvTL5JqT7rj4OekV0SjMlyn9mzC5X2P2xmfz+/F09ifm8HdCNsfT8knMLiIxu4hfD535AjPqtAxpF8mEHg1oPGoJ/PaBWlbXYqjaoKbETdPVMvjvJ8OB79THv5lD1HJ1/1rqn16BgKKWuJd0Po1sB9H9wOR37jfS+3m1ud3OGIj7Db4cAvV6QN+XIOo6N3xSxfLT1XL/0EYV29/pgFWPwp6FxRsUdc6/EEIIcZGcToXPt50E4NU6u9AufEV9IbQxNL4JtHpwWNTnNVuVPUFQPWgyEA6tVr9Xr1J2p52dp3ei0+jwMfjgZ/DDx+hDgDEAnVZXZn+H08GsXbNcndbPpXWN1twRfQcD6g0AINuSTZYli0yL2kiv5Dq+Bl/8jf546b3c/+YukuJwULh3H3kbN6LR6wmd+BAaXdnPwK3XVBSK9u7F1KSJK1sOULh7N/bTp9H6+uLTo0epY0Ifeojs738gf+MmcjdswK9XL7ePq/Dvv8nbsAG0WkIfuN/t5xfiWiJB+mVydXd3Uyb9XDQaDW2jAmkbFejalltk42BSLkdScjlyOo+jKXkcPp1LSq6FpTvjWboznp6Na3Bfj3vo1iik/LU8298NodGw/D61+Vytjurd+zqdofZ1Z7rKXy6/mjBwBnR7XM0S7JoPJzfDvBuhySDo+8L5O9EXZcOJTZCbrM539w1XH96BkLhHbbZzfAMk7QMUtWKg9/Pnz4o77LDyIfhr6Zlta6ZCeAuo190979sTnM6yUxeEEEJUui3Fy675mvR0yvrxzAtph9VHiZZ3nPv76faPIPFPqN+zcgd7Gd7d/S4L/l5QZnuQKYhRzUYxqskoAr0CAci15vLUpqfYmrAVgBFNRhBgCnAF4NmWbBoFNmJo9FCig6JLnc9sMBPhG1HZb+eiKYpC3q8byF27lryNG3FkZLheMzaoT8CgQZV6/dPTp5P5xZeYohtRe+5cV8Y6Z3VxqXvfvmiNpcv/jfXqETRiOJmLFpMwaTJ1Pv0E83VuTJIA6R99BID/oEEY69Vz67mFuNZoFEVRPD2IKyknJ4eAgACys7Px9/e/7PPN2jmL+X/PZ0zzMTx13VNuGOHl23Uqg083neCnA8mU/O3WD/Xh1tYRDG4bSaOwcjLeiqJmlXVX6L5NVixseAP2LirOyGugQU+1JD+oHgTVBe8giP0Njq6D+B1QPP+swloNh9veV+fj/5vDBt/eD39/q2Y2hs1TG9/9tVRdt/b+DRBYxcq0inLUqoBDq9Vf7prfpjYSNAerryuK2tQv/g91GkKrO9TPWgjhce7+bhKe+UzHL9jBun9SmNTByKS/71B7uDz8ByT/BYfXwOGfAAUe3AKBda7ImNztePZxhn03DLtip55/PQpsBeTacim0F7r28dZ7Myx6GH3q9GHa9mmczDmJl86LV7u/6sqSXwnWU6fQ16yJ1lTO7wGXIWPRIk5Pe8X1XOvnh6F2bSwHD+LVvDn1ln9TbmIk4/PPyVqxkvDnppbbOE1RFDIWfE7WkiXUmDQJ/5vKflbZ3/9A4pNPup7rAgOp9d67mDt04EjPXjjS0oj6+CN8e5a9yaNYrcQ9+ij5Gzeh9fGhzoIFeLdqWWafwv37UaylK0RNjRqiDw0952dSdOgQJ24bAhoNDb7/P0wNG55zXyGuVRfzvSRB+mWa8+ccPtn3CSObjOT5zs+7YYTucyo9n/lbT7J0ZxwF1jMBbtOafgxoUZNQPxNmgw6zUYe3UUeor4naQd4EeBvKz7pXhtRDsP5VOLjqwvuGREONJpCfqmbU806rjev8IqFhb3WOe/2ecOQn+L9JalBftxuM+OpMoApgzYcVD6rX1BrgzgXqXHprAcT0V3+ZimgL49aAwbv8sRTlqAH+3yugIAOcdnBY1eDf6KsGwO3uvnJNf1L+gSWj1fn/Z9PooH4PtWdA/I4za+8CGMzqdIPr75fMu7iwpH1qv4p2d4PB8yWm1Y0E6e53pT/TU+n59Hp7A4oCu3vtI/i3Gep30j1nfb85HeoN0yt1Q9zNFEXhoXUPsTVhK71q92JO3zmu12xOG+tOrSNmfwwHMw6WOi7cHM57fd6jeUjzKzbWvM1biLvvPvxvvplas2a67byK08mxmwZii43F/5ZbCLzjDswd2uPIy+No7z4oRUXUWTC/zPJilhMnOH7rYLDb0RiNRL79Fv79+5c6b8obb5DxefGUAK2WyDfeIODWW1z7FB0+zMkRI1EKCwm6axSF+/6iaP9+0OsJHDqUrKVL0QYE0HjzJjT/yqSXcBYVEXf/AxT88Qe6gADqfPkFXo0b48jOJnPJUjK//BJ7atkeAOdb11yxWomb+DD5W7bgN/Amar/zzqV8tEJUexKkn4e7v7Q/3vsx7+95n2HRw3i568uXP8BKkGex88uB06zam8imw6nYnef/K/cz6akdbKZOsDe9m4QxsGUEAWZD5Q7y9N9q07rMk5B5CrJOqUF4zdbQ6EZo2EfNrp9NUcBWoAab/76pcGw9LL1H7UYf0gg6jIXk/ZC0Ry05VJxq4Dr8S2hy05njMk/BJ73UTvhtRsGQD9VzK4oaiMf9oa5he2Cleu3z0Rqg+WC1835EG7DkqsG9JVcN6EMagV94xT8jS64aZPvXLh1UH/gOVk4Ea57aO+Dmt9RGgAe+U284nE1nUnsIOO2QsFPdFtVZrTgILV1meFVSFPWmSG4S5CVDfpo67sj20vCvMh3+GZberd4Ua/sfuO0D+bzdTIJ097vSn+kr3x9g3pYT9GpSgwWFj6v/Hx48B9qPqfRrXykb4zbyyPpH0Gv1rLxtJXX965bZR1EUtiduZ97+efyR/AdtarRhdu/ZhHqfOwtbGWIfeID8jZtAo6Hh2rUYa9cqs48jL4/Ep59BFxhIjccfxxAedsHz5m/bRuy48Wh9fYnetBGt+czybsnTXiFz0SJ8evSgzqeflDou/tHHyF27Fo3ZjFJQAFotNV98kaCRI1CsVhKfe961jrh3u3YU/vknaLVEvP4agUOG4MjN5eQdd2I9dQqfrl2I+vRTFKuVpOefd5W5AwTcMYzIV18973tw5OUTO24cRfv2oasRin+//mStXKmOC9AFB6MPCS61vz0pCY3RSO0575XK0jvy8kl4/HHyt24Fg4H6y7/Bq3HjC36OQlyLJEg/D3d/ac/fP59Zu2Zxa4Nbeb3H624YYeXKKrCyZn8yf5zMIN9ip8DqoNDqIN/qIDXXQlqepcwxBp2GXk3CGNK2Fh3rBVFgdZBbZCO3yE6exY6PUU+Qj4Egs5FgHyNORSmeH5/HkdO5HE3Jo9DmQKfVqA+NBh+TngEtanJj8zBM+kpqsHL6ACy8E3Liy74WUAdufUe9AfBvxzeqXegVhxrIOx3ll9qHNlaXywlroWZFdEY1ME87rM65j99x4TH6hEHNlmoDodDG6tx9vwh1iTyTr3rj4vgG9RG/Ux2H3qu4qqCxer19X6vnqtcD7phfOnuffkwtsdTo1AZ94a3UpfqcTtgVA2tfUoN7nQl6PKHeUPD0kj95qepNkCM/q6sEWAvU1Qms+epNDmc5TRqD6kPLYeoj/MplaipFUY56EyInUb0JUZB25k+jL3SfDD6X8cuu0wnJ+9SbG0afC++/f7k6NcR5VtflQbPguvGXPgZRhgTp7nclP1OHU6HdtJ/JKbKz9PYArv9xkPqd8ORhdepWNWB1WLn9u9uJzY3l3pb3MqXDlAsek5yfTA3vGuU2k6tMtqQkjva9Uf3/HRBy//2ETZlcZr+0jz8htTjrqzWbCX30UYJH/weN4dyJifhJk8lds4agu0ZR88UXS71mjYvj2ICbwOmk/qrvXMFqwa5dnPrPaNBqqb/iWzIXLiJrqdoPJ+ShByna95ca5Or1RE5/Hf9Bg0j+3zSyliwBjYaIV6aRu2EDeb+sQx8RQf1vl6MPUn+uFEUh7cMPSXtPrWqoEzMPn65dL/gZObKyODXmHiyHz/RKMDVpQsi4e/EfOLBUJl6xWomfMoW8X9ahMRio9e67+PXpjT09nbgHHqRo/340ZjO1330X3x5VuKePEJVMgvTzcPeX9sKDC5nxxwwG1BvA2z3fdsMIPavQ6iA+s4C4zAIOJObw/b4k/knOrbTrBZkNDGlXi+Edo2gWUQm/ROUmq4GoJUctYY9sq2a1/Wqe/7g/PoU1z5YOTEANkloOVUt+a193/mxi0l7YOV9dfs6WrwbKXv5qF3yNVs3ac5H//LT6smMC6Poo9H354ksos2LVqQHH1hWf36DOY+8wVi3TLK8MXlHU5fmOrYdTW9XMtsMKdov6p94ELW5XM65nTzM4n8Is+Od7NSA8vvHC/QfMIeqNDO8gSNxduqohJBqCGxTf8KipNhisfR1EtK7YWEo4bOrnfTEZ47xUdcpBrQ7l90L4N0VRqzN2LVBv6uQmqTdNzse/Noz4Qr3GxbLkqQH3oR/Un+Vmt0LrEVD/Bijvl+id89XVH1DURldhzWD9K+rPyb2rIeqsOZUOu9oUct8StR9Cl4cr/vdfmfLT1RtT51t94iogQbr7XcnP9GSaWupu0ms52GsH2i0z1aaooxZV6nWvpJKkRKh3KN/f/j0+hgrc5POQ1LlzSXtvDlp/f5w5OeiCg2m04ddSzdScRUUc7XsjjvR0DJGR2BITATBFR1PzxRfKbapmT0vjSK/eYLdTf+UKvJo2LbNPSRAfMGQIkTOmoygKJ0eOpGjvPgLvvJOIV6apgfWcOa7lyoAyQa6iKJx+5VUyF535GdIYDNRd+BXerct+n+Vt3YotNpbAkSMrPGXRnpZGwhNPojEZCR5zDz7dup7zWMVmI+HJp8j96SfQ6wmf+iyZX3yJ9dQpdEFBRH38UbnjEkKcIUH6ebj7S3vpoaW88tsr9I7qzXt93nPDCK8+/yTnsGpPIqv2JhKfWYivSY+flx5fkx4fk54Cq53MAhuZ+VZXKX2or5HoMD8ah/vSKNwPfy89TkXB7lBwKgqn0gv4dncCyTlFrus0relHj+hQekTX4Pr6wXgZruyd9zIKMtTgT6tXgxKtTs086i6y9N9hUwNrvVfpgM+aDykH1ZL00/vVUv/cZDVQK0hX9zGHqg31SubbB9RW90s7rM7nz4pVl+lpMrCcC1eQosBf38Bvc9WAt0RgXXWJPpPfmUdBuprVz4678Hl1xcF6x3FqMPfvL367FY7+olYCHFqjLktUIrKdumRgaLQ6ncHoo/5p8lODbv1Zc+2s+XDoRzXAP7K2/Ew7QMO+cMOTUPc8GQanU10pYGeMek6TnxrcR7RRb/LUbKXeADj7Z0BR4NQ22DkPDqxSr+9fS814n2v+dlG2evNm53xI+bvs66YA8I8Anxpq1twcqt6Y2P8NpB9VM3SDZpYuo81LUT+DlAPQbLBaJXL2Z54VC4tHqT9r/+YXAdH91ZUTvIPV4DrtCGwuvvHYcRzcPFM939Ixaj8Hvwh4YJN6TPoxNfgvmUIBYPBRs+1dH1X3qQwpB9WfSZN/8Q2w4v+nn9qmrgZxYhOkHlT/7fV8Gro8Wvpn53yu8CoJEqS735X8TNceOM19X+ykeU0/VvOIOmXrjhi1uqcaSCtM45YVt5Bvy+eVbq8wpNEQTw/pnBSnk2P9+mNLSCDi9ddJffdd7KdPE/nWW6Xmd2csXMjpV17FEBlJwzU/kr1qFSlvz8SRlQVAxGuvEThsaKlzp33yKamzZuHVpjX1lywp9/qF+/ZxcvgIMBho9MtaCnfvJmHyFDRmMw3X/Igh7Mz/DzMWLeL0K6+iCwwsN8hVFIWUGTNc89RrvvwyQSNHuONjuiSK3U7i08+Qs3q1a5shMpKoeZ9hql/fY+MSoqqoMkH6pk2beOutt9i1axdJSUmsWLGCIUOGnPcYi8XCtGnT+Oqrr0hOTqZ27do8//zzjBs3rkLXdPeX9sqjK3lh6wt0q9WNj2786LLPd7VzOhW02nPcZVUU8ix2HE6FQPOFfxF2OBU2HU5l6c44fjl4GpvjzI+iUa+lfZ1Agn2MaDUa9FoNOq1WXRvUrpboF9oc2BwKrWoFMKRdLdrUDrhyDe8qm92izj/3CbuyTd2S9sHuz9UA0pJz7v20BnWZvgY9IbCeGvjoTOqfmSfV4DN535n9vYPBP/JMdhuN2k2/8MyyNdRoqmZrWw6FkEvsCluYqWamc5Mg97T6Z3YcHPv1THa+bjfoNkntrKw41BsoToca0O2ar47/fLR6tbw+tLHaJ+HYekj958zrRt8z2XDfmuqyg1Gd4PRf6g2Z5L/Uz7mkE7Leu7hUf6h6Y8Q/4txl6EXZsOIhNRMO0P4edZrDviXqOM6uQIhspy5F2ORm9TNZ8h+16aJPGIxcqN5c2Pc17P8WirLO/X67T4G+L54J+C258GlfSDukfpYth8LPL6g3tEz+alB+8P/O/P3rvdQxBNVVb17411J/FnxC1Z8Lo/nc1y6Pw67eJPj9I7WR3cWo0RRumQ11u5T/urVAXR3hr2XqqhJhTaH/a+rPeSWTIN39ruRn+sGvR3nrp0M83iSTyaceVm9SPXX04n++r0I2p43/bvkvq0+spmVISxYOWohWc/U2G3XNGffzI3rzJtJjYkh7bw7e7dtTb9FCQM0KHxtwE7bERMJf+C/B//kPAPbMTFLeeJPslSvReHlRf9lSTNFqvxbF6VSPiYsrN4A/26nRd1OwcydBY+4m79cN2OLiCH3kEWo88nCZfa2nTqELCEAXGFjuuRRFIXv5chRFIfCOOzz+e45it5P43HPkrPo/TI0bE/XppxWayy+EqEJB+o8//sjWrVtp3749w4YNq1CQftttt3H69GleffVVGjVqREpKCna7na4VmH8D7v/SXn18Nc9sfobra17PvAHzLvt816qMfCubj6Sy5UgaW46mkZRddOGD/qVeiJnBbWvRp2kYGsBid2KxOyiyOUnNtRCXWUB8ZiHxmQVkF9poUzuQrg1D6NYolMjAc3Rxv1ZZ89WMeX6aGpRZctWgXWdUg8K6Xc4/n1lRIGG3mpHe/43acKw8vuHQ6k5oPVxtElhZv3xknICts2HPIrUk/3xMAdBmpJqlVhzqtIWkvWpgffpvderCvxnM6vvoOE4NBPd8BZvfKb8fQokaTdX9W48A78CKvxenE7bMUldF+Pd0iVodIbyFGmSWTAGo0VSdnuCwqpUAIxeXXl7QblErEJL2qjc5CjPUKhJrPrQZAddNKDuG1MPwaR+wnjUVpl4PtdFiYJT693/kZ9j4Zunsenn0Xuq0Bf9aapVD/RvUG0Bnl6cXZas3T46ugx2fQU6Cul1rUIP/kqaMJTc+Qhur56l/A9TtDkfXwk/Pq/P6Qe0lUaerWvXgKH4k/6UG/+VNN2h6C/SbVvrmUUGG+vNQkA4thpz/PVaABOnudyU/08lL9rDizwS+a7iKNglfq0uADvu0Uq9Z2ZLzk/nm8Dd8e+RbUgvVbt9f3fwVbWq08fDIzi9hyhRyVv/omjNuS0nhaJ++pUrUs1auJOnZqehCQmi07he0XmcqnhSnk7j7HyB/yxZM0Y2ot3QpWm9v8rZsJW7CBDX437QRrfe5f2/I/fVX4h+a6Hqur1GDhj+tKdVkripTnE6K/voLU+PG5/0chBClVZkg/WwajeaCQfqaNWsYOXIkx48fJzj40uY6uvtL+5dTvzB5w2Ta1mjLlzd/ednnE+pd42Op+eyOzcRid+JwOLE7FRzFpfTeRh1eBvWhKArrDqaw9sBpCm0XuY76WeqH+tA80h//4jJ+X5OBQLOBvs3CqB1UPb5UPcaSp5Z+5iadKecvylGzk/V7XdmliHISYdsctSy8ZM65Vqf2CwiopQZvLYed+waEoqjnSDusloNnHFO79LceDl4Bpfe1W2HvItj2vpqprtmq+NH6TKPAy7kpcfQXdSlBo48a6LcaDqGN1Nfy02D7B/DHJ2eCzqa3wNBPKtYsriIO/p+67J/OBDe+BJ0eKlv1oShwcovaADEnUQ2ucxLU/y5IL7+/Aqh/H5Ft1VUYMk+WXjoQ1GkAHcepj7P7S9it6rSJ8uafF2TALy/B7i/O/74C66p/n40HqpUGO+apN2u0BnVpxYJ0NTgvuVHgFQjPnLzsG0wSpLvflfxMb5mzmQMJWRwMmITJkgZ3LYXGV2498IpyKk7icuNIK0xzPdIL07H/69/iyZyTbIrfhKO4OifEK4SH2jzEiKbuLbVWnE7S3v8Ar5Yt8OvT57LPZ8/M5OgNPVFsNup/uxyv5moj0fjJk8n9cQ2BI0ZQ86UXOX7rYKzHjlFjyhRC77+v7HnS0zk+ZAiO1DQChw8nYtr/iH/scXJ//pmg//yHmi/894Lvq+QaADVfmUbQnXde9vsTQlRt1TZInzhxIocPH6Zjx458+eWX+Pj4MHjwYF555RW8z3Enz2KxYLGcmeuak5NDVFSU2760N8Vv4uF1D9M8pDlLbil/fpKofPkWO78cPM3KPxPYn5iDUafFpNdi1GsxGXSE+BiJCvKmdpCZ2kHeeBt17DiZwZaj6fwVn8W5VqXTazUMbhvJQz0bEh1e8cZTFruD7EIbOYU2sgtthPiYqBti9niZmqhmFOX8wWFBhtqUzugD193n/qkTSXvVLHhgnYs/VlHUGwgFGWr2PvUwnNwEJzarN3X+zafGmRUVWgy99LXaY3+D7e+DrUjtLaAzqAG4bxg0H1K2f0LKP/Dz8+pNkX8LrAPhLWHop+pqDJdBgnT3u1KfqdOp0PylNXRw7GWhcbr6b+KJwxXvf3CFnMo5xVMbnyqzhvn5dAzvyIimI+gb1RfDxfZjqYC8jRuJe+BBty3dlfH555yePgOv5s2p/+1y1/b83/8g9p570JjN1HzxBZKenYrW15dGv65H51f+d3v+tm3Ejp8AikL4c89x+s031Wz8d9/h1eTC48xasZKkqVMxRUdTf+UKNDoP99kRQnjcxXwvXcEU1uU7fvw4W7ZswcvLixUrVpCWlsbEiRPJyMggJiam3GOmT5/O//73v0obk0GrfmlZL1RCKyqVj0nPbW1rcVvbsuugnkuvJmE8NQCyC238cSKD+MwC8oqXlcu12DmakscfJzL4dncC3+5OoH/zcEZ1qkMNX5OabffS423QcSItn33x2fyVkM1fCVkcT82nwFo2qx9oNtCmdiBtowJpFuGH1aGQU6guZZdTZMNs0BEd7kt0uB91g83odRcOqBRFwWJ3upbTK7A6UFCoHWTG11T2n7fd4eRURgGZ+VZa1w7EqL965xWKCrjQTR9zMPS48DJJlyziMspeNZozDQmD6qpz6NsUZ+myYtV59AZvCKqnZrcvMwh2qdNZfVRUWFMYvRyO/KI2FQyurwbmYc3KVk+Ia1J8ZiGKrYjhxs3qhua3XXUB+o8nfuTlbS9TYC/AqDVS06cmod6hhHiHEOwVjJeu9E0vs8HMgHoDaBh4if1BKij/t+K+EjYbSc89T72vF6PRX9qvpoqikPWNGpgH3nlHqdfM11+HsWFDrMeOkfziSwAE3XXXOQN0AJ+uXQm5/37SP/6Y06+rS+x6t21boQAdIGDIbegC/PFq0VICdCHERatSQbrT6USj0bBw4UICAtRfjmbNmsUdd9zBBx98UG42ferUqUyZcuaX1JJMursYdeoXse1cHaXFVS/A20C/5uHlvrY3Lou5G47y09+n+fmA+qgojQb8vQz4eelJybGQVWBj4+FUNh5OveCxRr2WBqE+NKzhS/1QH+qF+lA/1AejTsvfidn8nZjD/sRsDiblUGRzlnuOUF8T9ULM1AkxU2h1cDQlj5Pp+a4GfXVDzEzp15hbW0eesxmgp8RnFrDhUCptowJpWUsCoWtOYJ1Ly85Xpugb1YcQADlJ6tSZ1IPUSD7MQVMCWk1xSVarq6es2eKw8OYfb7L0sLomd4fwDrzR4w3Cfcr/zrvS8n//zfXfRfv3kz5/PqH3lS0/r4iiffuwHDmCxmTCf9CgUq9pNBqCRo3i9KuvolitaEwmgu8Zc44znVHj0Uco2LGDwt3qyieBIype7q/RaNxSwi+EuDZVqSA9IiKCWrVquQJ0gGbNmqEoCvHx8UQXd+A8m8lkwmSqwHrFl8ioVYN0yaRXT22iAvn47o4cTcnl443H2R2bSZ7FTl6RnfzibLm/l57WtQNpVTuAVrUCaFrTjxAfE35eelfwa7U7OZiUw974LPbEZXEsJQ9vow4/L4MrkM8psnHkdB5HUnIpsjn5Jzn3otao9zJoMRv1KIpCZoGNtDwLaXkWdp4qPZ/XbNSh12o4lV7A41/v4eONx3n6pib0bFyj3HL89DwLm4+ksfFwKun5VjrVD6Zn4xo0j/CvcHCvKAo5hfbi5n0FpOVZaVjDl1a1A0pl/P+MzeSzLSdYsz/Z1YNgYMuaPNG/MY3Czj/doMjmYMaP/7BmfzJdG4UwqFUE3aNDMeklgyGEcLOts9UVBgBvAA0UaH0wNx+oNiX0MIvDwoa4DXy671MOZR5Cg4YJrSYwse1E9Fr3/+rnLCggd/2vKFYr/gP6o/W5cO8LR1YWloPqyhg1Jk8m9Z13SHtvDn59+mBqeP4MvqIo2E6dwpaSgiMrC0dmFjlrfgTA/6YB6MopIw24bTAps2ahFBQQOGwY+pCQC45Ro9dT6+23OHHncDRGI/43XX19BoQQ1VOVCtK7devGsmXLyMvLw9dXLX08fPgwWq2W2rVre2RMJZl0CdKrt0Zhfrx1Z+nSXodTocBqx9ekv+Bcc6NeS5uoQNpEBTLmHKs/lXA6FeIzCzmSksuJtHzX42RaPkV2J80j/GkR6U+LWgG0iPQn3N8Lb4MO3VkBc06Rjdj0Ak6k5RObUYBJr6VRmFpKH+HvRZHdQcyWE3y88TgHknIYO38HDUJ9CPf3ItjHSJCPAS+9Om9/X0I2Z3eu2HQ4lbd+OkSor5FujUIJ8DaQU2gjp8hOdqGNfIsdp6LgVNT34lQU0vOs5FrKNgnTaCA6zJc2tQM5kZZf6oZCswh//knO4cf9yfz0dzJD29fm8b7RRAWXbeS3PyGbSUv2cDRFbZJWMkXBz6SnX/NwbmpZkxsa18DL4N6AvcBqR6/VyrQBIa41aUfUPzuOY05qWxYcMjD2xo48euPlzam+HIqi8Hf636w8upIfT/xIjlVdRjPYK5jp3afTtZZ7bx4oikLh7t1krVhB7o9rcOarK1+cnjGDoJEjCRr9n1Jrgv9bwc6doCgYGzQg5P77KNi1k/xNm0l87jnqLVp03hLxzK8Wcvq118p9LfCOO8rdrvPzI+zJJ8hd8xMhDz5Q4fdZso46Gq10MhdCXDEebRyXl5fH0aNHAWjXrh2zZs2id+/eBAcHU6dOHaZOnUpCQgJffPGFa/9mzZrRuXNn/ve//5GWlsaECRPo2bMnn35asaVO3N1I5nj2cW5beRt+Rj+2jdp22ecT4krKzLcyd8NRPt9+Cqu9/LJ5UAPmXk1qEOZnYuvRdLYfS3NVElyMUF8TtYO8CTIbOHw6j4SswlKvG3Qabmtbi3Hd6tM80p/Dp3N5+6dDrmkGGg20rxNE32Zh9GsWToMavny6+Tgzfz6EzaFQw8/EU/2bcCAphx/3J3E650zTSG+Djhsah9K/eU16NqlBvsVObEYBcRmFxGYUoNdq6NoohA51g86bfVcUhW3H0vly+ynWHjyNj1HHoNYR3Na2FtfXCz5vdYHN4WTLkTR+O5FOdJgf3RuFUjPgEhugiWpFGse5X6V+pu+1V1d3GPsDg7+HffHZfDS6PTe1jHDvdSqowFbAY+sf4/fk313bws3hDG44mFFNR1HDXMOt1ys6cICEKU9gPXnStc0QFYVGq8V6qrjxo8FAwK23EjZ5EvoaZa+f/NrrZH75JYGjRhLx0kvYkpI4futgnHl5hD31FCHjx5V7bXt6OscG3IQzLw9DVBT60FB0gYHogoLwatmCoFGjpEmrEOKqVGW6u2/YsIHevXuX2X7PPfewYMECxo4dy8mTJ9mwYYPrtX/++YdHH32UrVu3EhISwvDhw3n11VfP2d3939z9pR2fG8/AbwfipfNix+gdl30+ITwhJbeIQ8m5ZORbyci3kplvJafITvNIf3o2rkG4f+lA0mp3sjs2k+3H0nEqCgHeatm+v7ceH5MenVaDVlPyUOf91w4y423Ulbnuvrhs9sZn4WXQcWeH2oT5lw1a98Rl8fZPh9hyNK3Udn8vPTlFaoZ+QItwpg9tTbCPWt3idCrsjs3k+31JrD1wuswNgXPxNujo3CCY7tE1iAjwwqDTYtBpMOq0/JOcy1e/n+J4ajlrpQMRAV7c3CqCxuG+RAZ6ExnoTUSAFweTclj5ZyI//JVERn7pqpvoMF+6R4dyXb1g6gSrqw8EeBtcv2QWWh3EZxYQl1lAdqGNAG8DgWYjQWYjwWYj/t7nr+TILrCh1YKfV9nOzDaHk9+PZ/DzgWScisKE7g2oF1q2TFVRFDYdSWN/QjbdGoXSpnaAx38JdjgVDiblEOZnKvdn5nIUWh38djydqGAzjcLc1LDuAiRId79K+0ydDng1HJw2nI//RYtZByi0OVj3RE8a1rgyPy9nszltPLr+UbYmbMWkM9G3Tl9ua3QbnWp2QqetnOk+ic88S/Z336Exm/G/6SYCbx+Cd8eOoCjk/for6THzKdy1CwCfrl2oU05z3+ODb8Ny+DC1Zs92lZFnLltG8gsvojGZqP/t8nLL3pNeeJGsZcvwatGCesuWonH3qhVCCFFJqkyQ7gnu/tJOKUih77K+aDVa9o7Z64YRCiHOJTGrkHX/pLDu4Gm2HUvHandiNup4+dYW3Nmx9jkDR0VR+DsxR23+93cy/yTnYtJrqRNspk6wmahgMzmFNjYfTSM111LuOc7mY9QxtH1t/tO5Dhn5Vr77M5HV+5PILTrHut9nCfU10qtJGEdS8tgXn0V5/wf2M+mpGeBFVqHtguPxMmipFehNreLlBYPMBhIyCzmZXsDJ9HyyCtSmllHB3jSr6U/zSH9qBXqz/Xg66w6mkF14pumlQafh3m71eaRPI/yLg/pdpzJ5Y80//HEiw7VfnWAzt7aJYHCbWjSpWfGlCRVFncqxLz6bAqsdu1PB7nBicyj4eenp3CCk3OkMJVJzLWw6nMqGw6lsPpJKVoENg07DqOvr8EjvRmWC9b/is/l6RyxpeRbqhfrQINSHBjV8aRDqQ7CPsdTPi6Io7I3PZunOOP5vT6JresaNzcJ5qFcDOtQNrvD7vBQSpLtfpX2mWXEwuyVoDcRNPEGPtzdh0Gk4OO2mCq3K4U5OxcnzW57n++Pf46Xz4rMBn9GmxmWsulABiqJwtHcf7MnJRM37DN9u3crdr2DHDk6NuQcUhYY/rcFYt67rNXtGBke6qsdFb9+GPijIde64CfeRv3Urxrp1qbd0Cbqz+hAVHTzIiaHDQFGou/ArzB06VOI7FUII95Ig/Tzc/aWdVZRFjyU9APjz7j8rpSGLEKKsfIudvXFZNArzvehMap7Fjo9RVyaoVxSFf5Jz2Xwkld+PZ5BbZMfqcGIrfviY9AxtX5vb29Uqs8Rdkc3BhkMpbD6SRkJWIYlZhSRkFpJvdeBj1DGgZU2GtK1F14Yhrl/kswqsbDuWzuYjaRxMyiE+s5C0vLJBuZ9JT1SwmWAfI9mFNjLyrWQVWC9pysG/hfgY6dc8nMTsIjYVrzwQ7GPkoZ4N+eNkBmuLpxoY9Vq6NAjhjxMZFNrOXNeo06LXaVwVBya9jlqB3tQJUW+A1A0xk11oY8fJTHacyCA5p+i844kK9qZbw1C6NAzBYndyNCWPoylqQ8W4jNLVEN4GnWssXgYt93arz5guddl0OJWFv8eyLz77nNcx6rSE+hqp4Wci1NdEfGYhh06fadQY5mciNc/iuolyXb0gRl1fh0Kbg6SsIhKz1b9jpwJLH7hAo4kKkCDd/SrtMz2xGT6/BYIb8mv/Ndy7YAdNwv34afIN7rtGBc3cOZMFfy9Ap9HxXp/3uKF25Y/BcuIExwfejMZgoPEfv593nnbsffeTv3kzIfffT9iUya7tOWvWkDBpMqbGjWmw6rtSx9jT0jgxfDj2xCR8unYh6pNP0OjVpqixd4+hYOdO/G++mVqzZlbaexRCiMpQbddJvxqVNI4DtXmcBOlCXBk+Jj1dG4Ve0rHlrSEP6pI5zSL8aRbhz/03XNz6wF4GHTe1jCg1J1VRFHKK7HgZtOXOcw80G7m5VQQ3tzpzTKHVQUJWIUnZhQR6G4kKLl3+frYim4PTOUUkZBYSn1lIfGYBGQVWIgO9qR+iLt1XN8SMxebkYHIOB5NyOZCYQ2xGPq1rBzKgRU061A1yNR389VAKr/1wkKMpeby2+iAAWg3c2SGKx2+MJjLQmwKrnXUHU/i/vYlsOJSK1eFEvVdwJnBPyCrkj5MZZcYLoNdqaFErgGCzAZ1WDez1Oi1JWYXsicsiLqOQrzPi+HpHXLnHt6zlT6/GYfRsUoN2UYH8cTKDN9ccYk9cFh9uOMaHG4659jXqtAxsVZM2tQM5lZ7P8bR8jqfmk5BViNXhJDG7iMTsMzcNTHotA1vWZPh1UXSuH8KJ9Hw+2Xicb/+MV28ynMwsMx6dVoPDqZRq3CiqucyT6p9B9ThcfGMnOvzKl7kv2L+ABX8vAGBat2lXJEAHKPhdnffu3a7dBRupBQ4bRv7mzWSvWEGNxx51rYGeX3wOc6dOZY7Rh4YSNXcuJ0fdRf627Zye8QY1//s8uT/9TMHOnWi8vAh78gk3vyshhLi6SER5mQy6M/M8Za10IcTZNBoNAd5l54Kfj7dRR6Mw3wrNhfYy6Kgb4kPdkPMvd2Q2QteGoXRteP6bGr2bhNG9USiLfo/lk03HaV07gCf6Nyk1FrNRz61tIrm1TSQFVrWjv82uYHOq1Qb5FnUOfWx6Aacy1D9NBi0d6wZzXf0g2kUFlelNUCLPYmfHiQy2Hk1jx6lM/Ex612fRKMyXxuF+rp4DJbo2DGXFxBB+OZjC2z8d4tDpXOqGmPlPpzrc0SGqzP6g3txQlyi0kpprITXXgkGnoX+LmqX+vhrW8OWNO1ozpX9jYrac4LcTGdTwNRIR4E1EoBeRAWrPAXGNOStIP1K8okT0BZaIdLd1seuYuUvNJE/pMIXBDQdfsWvnb1fXNjd3Lhtg/5tfn97ogoKwp6aSt2kzfn3UPkQFv/8BgE+n68s9zqtpUyLffIOERx8j86uvMNaJImPB5wCEjB+PITLSHW9FCCGuWhKkXya9Ro8GDQqKLMMmhKjyDDot93Stxz1d611wX7NRj9lY9mukQ92gS7q2r0lP76Zh9G567mWbyqPRaOjXPJw+TcNIzbUQ5mc6b5d9L4OO2kFmagede/772cL9vZh6c7OLGpOoxs4O0k8VB+lXMJNud9qZtXMWAHc1vYuxLcZesWsrTqcrk+7T+cLTPDRGIwFDhpAxfz5Z33yDX5/e2E6nYD1+HDQazNddd85j/fv1wzrpcVJnv8vp16cDoK9Zk5AJ493zZoQQ4iomLTEvk0ajObNWulOCdCGE8BSdVkPNAK/zBuhCXLbiIF0JqsvR4nL3xlcwSP/++PfE5sYSZAri8faPX9GVFiyHDuHIykJrNuPdqmWFjgm8YxgAeRs3YktJoeAPNYvu1axZqaZw5Ql54AH8Bw1yPQ978klZq1wIcU2QTLobGLVGLA6LZNKFEEKI6q44SE/VR5JvTUGv1Vxwyom72Jw2Pt77MQD3trwXs6Fi1SDukv9b8Xz06zqiMVRsKo+pYUO827encPdusld+hy0uFih/Pvq/aTQaIl57FRQnWh9f/AfdfOmDF0KIKkSCdDcw6AxgQ4J0IYQQojqz5EFBGgCHLMFACvVDfTBcoaXXvj/2PfF58QR7BTOiyYjLOpficJC79heyli3DWVSELjAQXVAg+sBA9BERBN15Jxpj6Z4O+b9tB8CnU+eLulbgsGEU7t5N1jffULJkgvkc89H/TevlRa1Zsy7qekIIUdVJkO4GJeXu0jhOCCGEqMayTql/egfzT6ZaZt443L1N49aeWsvX/3zNo+0epW1YW9d2m9PGx/vULPq4luMumEW3p6aSPG0ausBAvNu2w7tdO4z166EUFpL17QoyFizAFh9/7uPT0gh7/HHXc8Vmo3DHTgB8ulxckO5/0wBOv/46tlg1i45Oh7ljx4s6hxBCXEskSHcDo7Z4Trpk0oUQQojqq1Rnd3U+ekVWYqioPGseL297mRxrDhN+nsCMHjO4se6NAKw6uoqEvARCvEIY3mT4hYe6eDG5a38BIGvZNwDoAgJQFAVnTo7reeBdo/Bq0gRHVhaOrCysp2LJXrGCjPkLCBo5EkN4OACFf+3HWVCALjAQU5MmF/W+tD4++A8aRNbSpQB4tWiBzvfKL1snhBBVhQTpbiCN44QQQohrQHnLr7mxadzXh74mx5qDTqPD4rAwZcMUnr7uaUY0GcEn+z4B1Cy6t/7CzdPyNm0GwLdXLxx5uRT9tR9HdjYAhrp1CBk7loAhQ8o0YlMUBeupUxTu3k3qu+8R+fprABT8Xrz0WqdOaLQXX94feOcdriDdpwLz0YUQ4lomQbobGLRq8xTJpAshhBDVmKuzez2O7leDdHeVuxfYCvji7y8AeLnry+xP28+SQ0t4Y8cbrDm5hsT8REK9QyuURbenp1O0fz8ANaf9D0NYGIrVStE//+AsKsLcoQMana7cYzUaDeFPP8XJkaPIXrGC4HvG4NWkiWt9dJ8KrI9eHq+WLfFq2ZKi/fvx7dXzks4hhBDXClmCzQ1cc9IdMiddCCGEqLaKg/Rsr0hyLXZ0Wg313NTZfdnhZWRaMonyi+KWBrfwfKfnmdR+EgB7U/cCML7leLz0Xhc8V/62bQCYmjXDEBYGqGuWe7dujc/1158zQC/h3bYtfjfdBIpCyltv4ywqovDPPwEwd764+eglNBoNUR/Ope5XX2Lu0OGSziGEENcKCdLdQMrdhRBCiGtAcZAe61QD33ohZoz6y/9VqtBeyPz98wG4v+m9OBNPo9FoGN9qPDN6zMCgNVDXvy53NL6jQufL21xc6t69+yWPKWzKZDAYyN+yhdQ5c1BsNvTh4Rjr1bvkc+pr1JCGcUIIUQESpLuBNI4TQgghqjmnEzLV7u4HioIB95W6Lz+8nPSidGr51uL6b//h2I03ugLtQQ0G8evwX1k8aHGFsuiK00n+lq0A+PS49CDdWKcOwXeNAiBjXox6vs6d0Wg0l3xOIYQQFSNBuhsYdMVz0iWTLoQQQlRPecngsIBGxz8FanBeP/TyS90tDosriz6+1XgKt/8O4GqyBhBgCsDPWLEbAkUHDuLIyEDr44O5bdvLGlvIgw+i9Ttz3UstdRdCCHFxJEh3A8mkCyGEENVcSWf3wChyir/u/b0Nl33aFUdWkFKYQrg5nMFRN2M9pWbr8zZuwpGXd9Hny9+8CQBzl85ojMbLGps+KIjQBx9wPb/UpnFCCCEujnR3dwNpHCeEEEJUc2ctv1ZodQDgYzx/A7YLsTlszNs/D1Cz6MQngt0OgGK1krd+PQGDB1/UOfM2bwHAt8cNlzW2EkGjR1Owazf6sBoYIiLcck4hhBDnJ0G6G0jjOCGEEKKaOytIz09Vg3Rv4+X9GvXmjjdJzk+mhncNhkYPxbL211Kv56z+8aKCdEd2NoV79gDg273bZY2thNZkImruB245lxBCiIqRcnc3kHXShRBCiGquuGmcmklXs92Xk0lfemgpXx/6Gg0aXuj8AiadCcvRYwB4tWkNQN7WrTiysyt8zvztv4HTibFhQwy1al3y2IQQQniWBOluIJl0IYQQopo7O5NuKcmkX1qQviN5B9N/nw7Ao+0epXed3gBYjh4FwL//AEzR0WCzkfvLugqfN2/L5S+9JoQQwvMkSHeDksZxMiddCCGEqKbOnpNuK56Tbrr4cvf43HimbJiCXbFzU72bmNBqgus16zE1SDdFN8L/5oEA5Pz4Y4XOqygK+cXz0X169LjocQkhhLh6SJDuBq5MupS7CyGEENWPtUBdgg2KM+lqubu34eIy6QW2Ah779TGyLFk0D2nOtG7TXOuOKzYblpNqSb2pYUP8B6pBev727dgzMy94bsuRI9hPn0bj5YX5uo4XNS4hhBBXFwnS3UDK3YUQQohqLCtW/dMrALyDznR3v8hM+qu/vcqRzCOEeIXwbu938dZ7u16zxsaCzYbWbEYfGYmxXj1MzZuBw0Huz2sveO78zWqpu7nT9WhNposalxBCiKuLdHd3A2kcJ4QQQlRjZ5W6K4pCfnHjOPNFzEnPsebw4wm1dH1mr5nU9KlZ6vWSpnHGhg1d2XX/gQNJPXCQnB9/JGjE8DP7njhB9qpVKJYzv3fkrV8PgG93KXUXQoiqToJ0N3Ctk+6UOelCCCFEtVMSpAfWxWJ34lTUpxcTpG+O34xdsdMgoAEdwjuUed1y9AiglrqX8B84kNSZsyj44w/saWlozWbSPvqY9PnzwVb+7xy+N0iQLoQQVZ0E6W5Q0jhOMulCCCFENXRWJr2guNQdwHwR66Svi12HRlG4MbJXua9bj6mZdFN0I9c2Y+3aeLVuTdG+fZx+400Kdu3EnpgEgE/XrpiaNS11Dq9mzTHWrVvhMQkhhLg6SZDuBtI4TgghhKjGSgXpaqm7Sa9Fp9VU6HCLw8KWhC1MWumk8wcLsX03AkNkZOl9jqid3Y1nZdJBzaYX7dtHzv/9HwCGyEjCn38O3z59XGXxQgghqhdpHOcGBl3xnHRpHCeEEEKUa+7cudSvXx8vLy86dOjA5uJGZ+UZO3YsGo2mzKNFixZXcMRnKSeTfjGl7r8n/Y4up4BOhxQ0uflk//BDqdcVux3LSfUapkbRpV7zv3kgWrMZDAZCHniABj98j1/fvhKgCyFENSZBuhvIOulCCCHEuS1ZsoRJkybx/PPP8+eff9KjRw8GDhxIbGxsufu/++67JCUluR5xcXEEBwdz5513XuGRA4oCWerSaKWD9IsrdW91UkFbPJc995dfSr1ujY0Dmw2NtzeGyIhSrxnCw6m/ahWNfvmFsMmT0Hp7I4QQonqTIN0NZAk2IYQQ4txmzZrF+PHjmTBhAs2aNWP27NlERUXx4Ycflrt/QEAANWvWdD127txJZmYm99577xUeOZCfCrYC0GghIIoCy8V1dnc4HWyI20Db44prW9HefdhOn3Y9dzWNa9AAjbbsr2bG2rUwhIddxpsQQghRlUiQ7gbSOE4IIYQon9VqZdeuXfTv37/U9v79+7Nt27YKnWPevHnceOON1D1PUzSLxUJOTk6ph1tkxwEa8K8NeuOZTHoF10jfk7qHzMJ02p1Qn2t9fQHIXbfOtY+raVyjRmWOF0IIce2RIN0NpHGcEEIIUb60tDQcDgfh4eGltoeHh5OcnHzB45OSkvjxxx+ZMGHCefebPn06AQEBrkdUVNRljdulVgd4PhnGqWucF9iKg3RDxTLp62PXUycFAvMUNGYzIePHAZD3y5kg3dU0rlHDcs8hhBDi2iJBuhtI4zghhBDi/P7d6ExRlAo1P1uwYAGBgYEMGTLkvPtNnTqV7Oxs1yMuLu5yhluawQsCagO4yt19TBcO0hVFYV3sOlepu0+nTvgPHAhA/h9/4MjOBsAimXQhhBBnkSDdDaRxnBBCCFG+0NBQdDpdmax5SkpKmez6vymKQkxMDHfffTdGo/G8+5pMJvz9/Us9KkNJubt3BRrHHc48TEJeAu2Pq899enTHWK+euha63U7exo0odjvWE2otvATpQgghQIJ0t5DGcUIIIUT5jEYjHTp0YO3ataW2r127lq5du5732I0bN3L06FHGjx9fmUO8KCXrpPtUoHHc+rj1eFsUGsermXTfHj3UP2+8EYDctb9gjYtDsVrReHlhqFWrkkYthBCiKvFokL5p0yZuvfVWIiMj0Wg0rFy5ssLHbt26Fb1eT9u2bSttfBUljeOEEEKIc5syZQqfffYZMTExHDx4kMmTJxMbG8uDDz4IqKXqY8aMKXPcvHnz6NSpEy1btrzSQz6nM5n0CgTpsetpeUpB51Qw1q2LsXievF9xkJ63ZQtFfx8Azt3ZXQghxLXHo98G+fn5tGnThvfff/+ijsvOzmbMmDH07du3kkZ2cUrmpNucNhRFucDeQgghxLVlxIgRzJ49m2nTptG2bVs2bdrE6tWrXd3ak5KSyqyZnp2dzfLly6+qLDqcCdJ9LlDunpCXwD8Z/9C2pNT9hhtcr3k1b44+MgKlsJCML74ApGmcEEKIMyq2fkglGThwIAOLG6hcjAceeIC77roLnU53Udn3ylJS7g5qoH72cyGEEELAxIkTmThxYrmvLViwoMy2gIAACgoKKnlUF6+k3P1CmfSNcRtBUbj+pB6w4tuju+s1jUaDX98byfzyS4r27QPA1Ci60sYshBCiaqlydVXz58/n2LFjvPTSSxXav9LWTT1LSbk7SMm7EEIIUZ3luzLp5w/SdyTvoFY6BGRa0RiNmK+7rtTrJSXvJUySSRdCCFGsSgXpR44c4dlnn2XhwoXo9RUrAqi0dVPPcnbmXJrHCSGEqOrq1avHtGnTypSgCygsDtLN5yl3dypOdp3e5Vp6zXzddWi9vUvtY+7QHl1goOu5dHYXQghRosoE6Q6Hg7vuuov//e9/NG7cuMLHVeq6qcW0Gi16jfplLZl0IYQQVd0TTzzBd999R4MGDejXrx9ff/01FovF08O6KuQXr5NuPs866ceyjpFpyaT9CXUdeJ+zSt1LaPR6fHv3Vv/bZJLO7kIIIVyqTJCem5vLzp07eeSRR9Dr9ej1eqZNm8bevXvR6/WsX7++3OOu1LqpruZxsla6EEKIKu7RRx9l165d7Nq1i+bNm/PYY48RERHBI488wu7duz09PI8qtJVk0s8dpO9I3oHRptAs1gmcWXrt3/xvVvvyeLVogUZ34W7xQgghrg1VJkj39/fnr7/+Ys+ePa7Hgw8+SJMmTdizZw+dOnXy6PhkrXQhhBDVTZs2bXj33XdJSEjgpZde4rPPPuO6666jTZs2xMTEXJMrmrgy6ecpd995eifNYxX0dgVDZCTGBg3K3c+3Rw9qfziXyDffqJSxCiGEqJo82t09Ly+Po0ePup6fOHGCPXv2EBwcTJ06dZg6dSoJCQl88cUXaLXaMuukhoWF4eXldVWsnyprpQshhKhubDYbK1asYP78+axdu5bOnTszfvx4EhMTef755/nll19YtGiRp4d5RZ2Zk15+5ltRFHad3sXg4vnoPt27o9Foznk+v+KSdyGEEKKER4P0nTt30vusL6cpU6YAcM8997BgwYJy1029WkkmXQghRHWxe/du5s+fz+LFi9HpdNx999288847NG3a1LVP//79ueGstb+vFfkXaBx3PPs4GUUZtD2hPvfp3u1KDU0IIUQ14dEgvVevXuctlStv3dSzvfzyy7z88svuHdQlMmjVOemSSRdCCFHVXXfddfTr148PP/yQIUOGYDAYyuzTvHlzRo4c6YHRedaFMuk7kncQkq0Qma6ATodP585XcnhCCCGqAY8G6dVJSSZdGscJIYSo6o4fP07dunXPu4+Pjw/z58+/QiO6OtgcTqwOtRmczzky6TtP76TtCTUB4d26NbpKalgrhBCi+qoyjeOudq456VLuLoQQoopLSUnh999/L7P9999/Z+fOnR4Y0dWhoDiLDuBdTiZdURR2JO+g9YmS+ehS6i6EEOLiSZDuJq456VLuLoQQoop7+OGHiYuLK7M9ISGBhx9+2AMjujoUWNXO7gadBqO+7K9QJ7JPkFWQTuuTapDu273s+uhCCCHEhUiQ7iYl66RLJl0IIURVd+DAAdq3b19me7t27Thw4IAHRnR1KMmkexvKn4++8/ROGiaBTxFoAwLwugpWnxFCCFH1SJDuJiXl7jInXQghRFVnMpk4ffp0me1JSUno9dduO5sCixqk+5jK/wx2JO+gTcnSa126oNGVH8wLIYQQ5yNBuptIubsQQojqol+/fkydOpXs7GzXtqysLJ577jn69evnwZF5Vkm5+7nmo+88vZM2J9TGcr4yH10IIcQlunZvh7uZNI4TQghRXcycOZMbbriBunXr0q5dOwD27NlDeHg4X375pYdH5zkl5e7ldXY/mXOSgsxUohPV5z7dJEgXQghxaSRIvwyZX39Nyjuz8buxL4aBsk66EEKI6qFWrVrs27ePhQsXsnfvXry9vbn33nsZNWpUuWumXytcc9LLyaTvSN5Bq5MKWgWMDRtiiIi40sMTQghRTUiQfpmc2dk4c3Iw6kIByaQLIYSoHnx8fLj//vs9PYyrSn5xubtPOUG6Wupe0tVdsuhCCCEunQTpl0Hr4wuAIy8PozYSkMZxQgghqo8DBw4QGxuL1Vr6BvTgwYM9NCLPKizOpJv/Ve6uKAo7k3YwqGR9dCl1F0IIcRkuKUiPi4tDo9FQu3ZtAP744w8WLVpE8+bNr6m77lpfHwCcefnSOE4IIUS1cfz4cW6//Xb++usvNBoNiqIGnxqNBgCHw+HJ4XlMSSbd/K9MeqYlE31CKmHZgMGA+brrPDA6IYQQ1cUldXe/6667+PXXXwFITk6mX79+/PHHHzz33HNMmzbNrQO8mml9ioP0/HwMWlknXQghRPXw+OOPU79+fU6fPo3ZbObvv/9m06ZNdOzYkQ0bNnh6eB5zJpNeOkiPzYmlbfHSa+aOHdB6e1/xsQkhhKg+LilI379/P9dffz0AS5cupWXLlmzbto1FixaxYMECd47vqqbzVcvdnXl5kkkXQghRbWzfvp1p06ZRo0YNtFotWq2W7t27M336dB577DFPD89j8ovXSTf/a530kzknae2aj979io9LCCFE9XJJQbrNZsNkMgHwyy+/uOamNW3alKSkJPeN7iqnLSdItzllTroQQoiqzeFw4Fv8HRcaGkpiorquWN26dTl06JAnh+ZRhbbicndD6Ux6fNpxWsQWz0eXIF0IIcRluqQgvUWLFnz00Uds3ryZtWvXctNNNwGQmJhISEiIWwd4NXMF6QUFGJEl2IQQQlQPLVu2ZN++fQB06tSJN998k61btzJt2jQaNGjg4dF5zrky6Y4/duNlA0uoP6bGjT0xNCGEENXIJQXpb7zxBh9//DG9evVi1KhRtGnTBoBVq1a5yuCvBSVz0gG8bOoddAnShRBCVHX//e9/cTqdALz66qucOnWKHj16sHr1at577z0Pj85zCs4xJz1kxzEAHN3bu5rrCSGEEJfqkrq79+rVi7S0NHJycggKCnJtv//++zGbzW4b3NVOYzSCwQA2G6Yi9ZcZaRwnhBCiqhswYIDrvxs0aMCBAwfIyMggKCjomg5CC8rp7u602Yj+OxuAgH79PTIuIYQQ1cslZdILCwuxWCyuAP3UqVPMnj2bQ4cOERYW5tYBXs00Gg264my6qbgETtZJF0IIUZXZ7Xb0ej379+8vtT04OPiaDtDh7Ez6mRxH8m8b8C9QyPWGOt1v8tTQhBBCVCOXFKTfdtttfPHFFwBkZWXRqVMnZs6cyZAhQ/jwww/dOsCrXcm8dJNFMulCCCGqPr1eT926da/ZtdDPpyST7nNWJj11zfcAHGzqg9EkS68JIYS4fJcUpO/evZsePXoA8M033xAeHs6pU6f44osvrrm5aiXz0g2F6he3zEkXQghR1f33v/9l6tSpZGRkeHooV5WSTLp3cZCuKAps+h2A0x3reWpYQgghqplLmpNeUFCAn58fAD///DNDhw5Fq9XSuXNnTp065dYBXu1KMumGIrXMXTLpQgghqrr33nuPo0ePEhkZSd26dfE5q1EqqDfrr0UlQbpPcXd3y8GDGFOzKTKA8/rWnhyaEEKIauSSgvRGjRqxcuVKbr/9dn766ScmT54MQEpKCv7+/m4d4NVO63tWJl0rc9KFEEJUfUOGDPH0EK5KJeXu3sXrpOf+8gsAextoiApt6LFxCSGEqF4uKUh/8cUXueuuu5g8eTJ9+vShS5cugJpVb9eunVsHeLUraRynL7KBWcrdhRBCVH0vvfSSp4dw1XE4FYpsav+Zkkx67lo1SP8jWsMovzoeG5sQQojq5ZKC9DvuuIPu3buTlJTkWiMdoG/fvtx+++1uG1xVoPVRy911BVY1SJdydyGEEKLaKbSdaaRnNuqwnjqF5cgR7FrY3UjDVP96nhucEEKIauWSgnSAmjVrUrNmTeLj49FoNNSqVYvrr7/enWOrEkrmpOsKLIBk0oUQQlR9Wq32vMutXYud3wssaqm7VgMmvZaMX9YBcKCOBovZQIRvhCeHJ4QQohq5pCDd6XTy6quvMnPmTPLy8gDw8/PjiSee4Pnnn0ervaSm8VVSyZx0TWERADanzEkXQghRta1YsaLUc5vNxp9//snnn3/O//73Pw+NyrPOXiNdo9G45qP/0VhDbb/a6LWXnPcQQgghSrmkb5Tnn3+eefPmMWPGDLp164aiKGzdupWXX36ZoqIiXnvtNXeP86pVsgSbVjLpQgghqonbbrutzLY77riDFi1asGTJEsaPH++BUXlWfnHTOLNRhz01lcI9ewDY0VhDK3+Zjy6EEMJ9LilI//zzz/nss88YPHiwa1ubNm2oVasWEydOvKaCdF1xubumoBAAh+LA4XSg0+o8OSwhhBDC7Tp16sR9993n6WF4RKErk64jd916UBQyG4SS6ZdFHWkaJ4QQwo0uqS49IyODpk2bltnetGlTMjIyLntQVUnJnHTyC13bpHmcEEKI6qawsJA5c+ZQu3ZtTw/FI/LPKncvyaIfaaouO1vXv66nhiWEEKIauqRMeps2bXj//fd57733Sm1///33ad26tVsGVlWUlLsr+QWubVaHFW+9t6eGJIQQQlyWoKCgUo3jFEUhNzcXs9nMV1995cGReU7hWeXu1thYAI4EqN/9EqQLIYRwp0sK0t98800GDRrEL7/8QpcuXdBoNGzbto24uDhWr17t7jFe1UqWYFMK8l3bpHmcEEKIquydd94pFaRrtVpq1KhBp06dCAoK8uDIPCffUpxJN+mxxqlB+gEvtXpQgnQhhBDudElBes+ePTl8+DAffPAB//zzD4qiMHToUO6//35efvllevTo4e5xXrVKurs78/Ixao1YnVZpHieEEKJKGzt2rKeHcNUpKF4nPUCx4UhNAyAxwIFRa6KmT01PDk0IIUQ1c8nrhURGRpZpELd3714+//xzYmJiLntgVUVJ4zhnXh5GnVmCdCGEEFXe/Pnz8fX15c477yy1fdmyZRQUFHDPPfd4aGSeU7JOeni+GqA7/XzI97bQ0C8KrebaWXpWCCFE5ZNvlcvkmpNuteKtGABpHCeEEKJqmzFjBqGhoWW2h4WF8frrr3tgRJ5Xsk56aI4apBeGBwBQR5ZfE0II4WYSpF+mkiAdwNeuLrtmc8icdCGEEFXXqVOnqF+/fpntdevWJba4adq1pqC4cVxIVgoAmaEmQOajCyGEcD8J0i+TRq9H4612cve3qbMHJJMuhBCiKgsLC2Pfvn1ltu/du5eQkBAPjMjzSjLpAZmnAUgMdAKSSRdCCOF+FzUnfejQoed9PSsr66IuvmnTJt566y127dpFUlISK1asYMiQIefc/9tvv+XDDz9kz549WCwWWrRowcsvv8yAAQMu6rrupvX1wVFYiK9VBwZkTroQQogqbeTIkTz22GP4+flxww03ALBx40Yef/xxRo4c6eHReUZJkO6boQbpx33U5dfq+dfz1JCEEEJUUxcVpAcEBFzw9TFjxlT4fPn5+bRp04Z7772XYcOGXXD/TZs20a9fP15//XUCAwOZP38+t956K7///jvt2rWr8HXdTWf2wUGaq9xdgnQhhBBV2auvvsqpU6fo27cver36q4LT6WTMmDHX8Jz04nXSU5MA+MecBUAdP8mkCyGEcK+LCtLnz5/v1osPHDiQgQMHVnj/2bNnl3r++uuv89133/F///d/Hg3StcUd3s1WdU1ZKXcXQghRlRmNRpYsWcKrr77Knj178Pb2plWrVtSte+3Ovy6wOtA77RjTUwG13N1bbybMHObhkQkhhKhuLnkJtquB0+kkNzeX4ODgc+5jsViwWCyu5zk5OW4fR0mQ7lMcpEvjOCGEENVBdHQ00dHRnh7GVaHA6iCsIBON4kTxMpLl46CxXxQajcbTQxNCCFHNVOnGcTNnziQ/P5/hw4efc5/p06cTEBDgekRFRbl9HCUd3iWTLoQQojq44447mDFjRpntb731Vpm1068V+RY7kflnLb+m0UhndyGEEJWiygbpixcv5uWXX2bJkiWEhZ271Gzq1KlkZ2e7HnFxcW4fi9ZXDdK9ixP2MiddCCFEVbZx40YGDRpUZvtNN93Epk2bPDAizyu0OYjITwcgO1Rd1aW2X21PDkkIIUQ1VSXL3ZcsWcL48eNZtmwZN95443n3NZlMmEymSh2Prrjc3cuqABKkCyGEqNry8vIwGo1lthsMhkqZNlYVFFjLBul+Bj9PDkkIIUQ1VeUy6YsXL2bs2LEsWrSo3Lv8nqD1UYN0b4u6ZqrNKXPShRBCVF0tW7ZkyZIlZbZ//fXXNG/e3AMj8rwCi53IPLXcPSvUCwCjruyNDCGEEOJyeTSTnpeXx9GjR13PT5w4wZ49ewgODqZOnTpMnTqVhIQEvvjiC0AN0MeMGcO7775L586dSU5OBsDb2/uCy8NVppI56aYiNUiXTLoQQoiq7IUXXmDYsGEcO3aMPn36ALBu3ToWLVrEN9984+HRXXmKolBgcxBRoGbSM0PU4FyCdCGEEJXBo5n0nTt30q5dO9fyaVOmTKFdu3a8+OKLACQlJREbG+va/+OPP8Zut/Pwww8TERHhejz++OMeGX+Jku7upuJMujSOE0IIUZUNHjyYlStXcvToUSZOnMgTTzxBQkIC69evp169ep4e3hVXZHOC00nN/AwA0kMMABi1EqQLIYRwP49m0nv16oWiKOd8fcGCBaWeb9iwoXIHdIlKGscZi+yAZNKFEEJUfYMGDXJNK8vKymLhwoVMmjSJvXv34nA4PDy6K6vAaiekMAej0w56PekBOiiSTLoQQojKUeXmpF+NShrHGYrUX1okSBdCCFEdrF+/ntGjRxMZGcn777/PzTffzM6dOz09rCtObRqnzkc31IqkCPV73qSr3Ma0Qgghrk1Vsrv71aZkTrqhUG0YJ43jhBBCVFXx8fEsWLCAmJgY8vPzGT58ODabjeXLl1+7TePO6uxujKqDzVGg/rdk0oUQQlQCyaS7QcmcdH2RGpxLJl0IIURVdPPNN9O8eXMOHDjAnDlzSExMZM6cOW4599y5c6lfvz5eXl506NCBzZs3n3d/i8XC888/T926dTGZTDRs2JCYmBi3jOVi5VvtRJYE6XWisDgs6n9LkC6EEKISSCbdDUqWYNMXqsG5NI4TQghRFf3888889thjPPTQQ0RHR7vtvEuWLGHSpEnMnTuXbt268fHHHzNw4EAOHDhAnTp1yj1m+PDhnD59mnnz5tGoUSNSUlKw2+1uG9PFKDy73L1OHSyOHYA0jhNCCFE5JJPuBiXl7rpCKyiKZNKFEEJUSZs3byY3N5eOHTvSqVMn3n//fVJTUy/7vLNmzWL8+PFMmDCBZs2aMXv2bKKiovjwww/L3X/NmjVs3LiR1atXc+ONN1KvXj2uv/56unbtetljuRT5FvuZcvc6dVzT2mROuhBCiMogQbob6Iq7u2ucCkY72BwyJ10IIUTV06VLFz799FOSkpJ44IEH+Prrr6lVqxZOp5O1a9eSm5t70ee0Wq3s2rWL/v37l9rev39/tm3bVu4xq1atomPHjrz55pvUqlWLxo0b8+STT1JYWHjO61gsFnJycko93KXw7HL3KCl3F0IIUbkkSHcDjdkMGg0A3hYpdxdCCFG1mc1mxo0bx5YtW/jrr7944oknmDFjBmFhYQwePPiizpWWlobD4SA8PLzU9vDwcJKTk8s95vjx42zZsoX9+/ezYsUKZs+ezTfffMPDDz98zutMnz6dgIAA1yMqKuqixnk+RemZ+NiLADBERbkq5iRIF0IIURkkSHcDjUbjah5ntkjjOCGEENVHkyZNePPNN4mPj2fx4sWXfB5N8c3sEoqilNlWwul0otFoWLhwIddffz0333wzs2bNYsGCBefMpk+dOpXs7GzXIy4u7pLH+m9KvHquPL9gtF5eru95KXcXQghRGaRxnJtofXxw5ubibZVMuhBCiOpHp9MxZMgQhgwZclHHhYaGotPpymTNU1JSymTXS0RERFCrVi0CAgJc25o1a4aiKMTHx5fb1M5kMmEyVU7QrE1KACAvJBxFUVzf85JJF0IIURkkk+4m2uJ56d5WReakCyGEEMWMRiMdOnRg7dq1pbavXbv2nI3gunXrRmJiInl5ea5thw8fRqvVUrt27Uodb3n0pxMBKAyLwK7YcSpOQIJ0IYQQlUOCdDfRFS/D5i3l7kIIIUQpU6ZM4bPPPiMmJoaDBw8yefJkYmNjefDBBwG1VH3MmDGu/e+66y5CQkK49957OXDgAJs2beKpp55i3LhxeHt7X/Hxe6WoQbo1LKLUd7wswSaEEKIySLm7m5TMSfe2wGkpdxdCCCFcRowYQXp6OtOmTSMpKYmWLVuyevVq6tatC0BSUhKxsbGu/X19fVm7di2PPvooHTt2JCQkhOHDh/Pqq696ZPzmNLVU31GzVukgXTLpQgghKoEE6W5Ssla62SqZdCGEEOLfJk6cyMSJE8t9bcGCBWW2NW3atEyJvKf4pp8GQKlV27X8ml6rR6uRgkQhhBDuJ98ublKSSfeygs0pc9KFEEKI6sCRl49PfjYA2lpRrr4z0tldCCFEZZEg3U1KGseZLYpk0oUQQohqwla8/FqOwYxXUIArky5BuhBCiMoi5e5uUlLuLo3jhBBCiOpDHx7O133HkpqWw3CjzrX8mkFr8PDIhBBCVFcSpLuJrqRxXPE66YqioNFoPDwqIYQQQlwOfVAQm+t34rhfPmONeteNeMmkCyGEqCxS7u4mWp8zQTqA3Wn34GiEEEII4S4FVgcAZqPOVe4und2FEEJUFgnS3eTsJdgAVzmcEEIIIaq2fKt6491s1Lky6RKkCyGEqCwSpLuJ1scMgLdFAWReuhBCCFEdKIpCYXEm3cd0ptzdqJUgXQghROWQIN1NSuakm63qPHQJ0oUQQoiqz+pwYneqN+C9z2ocJ3PShRBCVBYJ0t1E61t6TrqUuwshhBBVX0kWHcBsODMn3aCT7u5CCCEqhwTpbnJmTrp6t93msHlyOEIIIYRwg/ziIN2o16LXaaW7uxBCiEonQbqblKyT7mUDjVORTLoQQghRDRQWN43zMeoApHGcEEKISifrpLtJSSYdwMsqc9KFEEKI6qCGrxfvjGjjeu5agk0axwkhhKgkEqS7idZoRGMwoNhsmK2QZ8vz9JCEEEIIcZkCzAZub1fb9VwaxwkhhKhsUu7uRiUl794WSMxL9PBohBBCCOFuUu4uhBCiskmQ7kZnmsdBfG68h0cjhBBCCHeTIF0IIURlkyDdjc4sw6YQnydBuhBCCFHduOakS5AuhBCikkiQ7kZa3+Jydysk5CZ4eDRCCCGEcDdZgk0IIURlkyDdjUrmpJstSCZdCCGEqIZKGsdJd3chhBCVRYJ0N9L5nJmTnmXJIs8qHd6FEEKI6kTmpAshhKhsEqS7Ucmc9CCnFwAJeVLyLoQQQlQnUu4uhBCiskmQ7kYlQXoNRf1TOrwLIYQQ1UtJ4ziDzuDhkQghhKiuJEh3I62PGYBgh5pJl3npQgghRPUimXQhhBCVzaNB+qZNm7j11luJjIxEo9GwcuXKCx6zceNGOnTogJeXFw0aNOCjjz6q/IFWkK44kx5gV+epSSZdCCGEqF5cc9KlcZwQQohK4tEgPT8/nzZt2vD+++9XaP8TJ05w880306NHD/7880+ee+45HnvsMZYvX17JI62YknJ3X5sOkEy6EEIIUd1YnLJOuhBCiMql9+TFBw4cyMCBAyu8/0cffUSdOnWYPXs2AM2aNWPnzp28/fbbDBs2rJJGWXElS7B5WxVAMulCCCFEdWNz2AApdxdCCFF5qtSc9O3bt9O/f/9S2wYMGMDOnTux2WzlHmOxWMjJySn1qCza4iXYjIV2ABLzEnEqzkq7nhBCCCGurJLGcZJJF0IIUVmqVJCenJxMeHh4qW3h4eHY7XbS0tLKPWb69OkEBAS4HlFRUZU2Pq2vmknXFlrQaXRYnVZSC1Ir7XpCCCGEuLJknXQhhBCVrUoF6QAajabUc0VRyt1eYurUqWRnZ7secXFxlTa2ksZxSn4BET4RgMxLF0IIIaoTaRwnhBCislWpIL1mzZokJyeX2paSkoJeryckJKTcY0wmE/7+/qUelaVkTrozL49afrUAmZcuhBBCVCdWpyzBJoQQonJVqSC9S5curF27ttS2n3/+mY4dO2IwGDw0qjNKursrNht1TGomPSEvwZNDEkIIIYSbKIrimpNu0Hn+9w4hhBDVk0eD9Ly8PPbs2cOePXsAdYm1PXv2EBsbC6il6mPGjHHt/+CDD3Lq1CmmTJnCwYMHiYmJYd68eTz55JOeGH4ZJZl0gDr6GoBk0oUQQojqwuY806RWMulCCCEqi0eXYNu5cye9e/d2PZ8yZQoA99xzDwsWLCApKckVsAPUr1+f1atXM3nyZD744AMiIyN57733rorl1wA0Oh0asxmloIDammBA5qQLIYQQ1UXJfHSQIF0IIUTl8WiQ3qtXL1fjt/IsWLCgzLaePXuye/fuShzV5dH6mHEUFFBTEwBAQq6UuwshhBDVQUmpO4BBK+XuQgghKkeVmpNeFegDgwAISVNL4lIKUyiyF3lySEIIIYRwg5Jyd6PWeM5VZYQQQojLJUG6m/l06waAc9N2fAzqHPXEvERPDkkIIYQQblCSSZc10oUQQlQmCdLdzK9/PwDyft1AXa9IQOalCyGEENWBa410CdKFEEJUIgnS3cy7bVt0NUJx5uVxfYI3IB3ehRBCiOpAgnQhhBBXggTpbqbRavG78UYAWv6dB0gmXQghhKgOSsrdpbO7EEKIyiRBeiXw76eWvEfujkfjVKTDuxBCCFENWJ2SSRdCCFH5JEivBObrrkMbEIA+p4Cm8ZJJF0IIIaoDV7m7VoJ0IYQQlUeC9EqgMRjw69MHgM7/OInPjT/vevBCCCGEuPqVBOlS7i6EEKIySZBeSfyKS96vP6xQaMsn05Lp4REJIYQQ4nKUzEk36AweHokQQojqTIL0SuLTrStas5mQXGiQhMxLF0IIIao4yaQLIYS4EiRIryRakwnfXj0B6HTIKfPShRBCiCpO5qQLIYS4EiRIr0QlJe+dDinE58R5eDRCCCGEuBwl5e7S3V0IIURlkiC9Evn0uAGnQUdEJuQdOuDp4QghhBDiMpQswSbl7kIIISqTBOmVSOfrQ0H7JgD4bZMgXQghhKjKXOXukkkXQghRiSRIr2R+/W4EoMEf8eRZcj08GiGEEEJcKgnShRBCXAkSpFeyRkPuptBLS0SGwh/LPvD0cIQQQghxiVxz0qVxnBBCiEokQXol0/v6kjKgHQDKohUeHo0QQgghLpXNaQNkTroQQojKJUH6FdDk/inYdBB5PIfTv23y9HCEEEIIcQlKMukGncHDIxFCCFGdSZB+BdRv2J6/2gcBcOLDdzw8GiGEEEJcipIgXTLpQgghKpME6VeI7j9DAfD/4x8sx094eDRCCCGEuFg2h5S7CyGEqHwSpF8hvXqMZmcjDRoF4j+VBnJCCCFEVeMqd9dKubsQQojKI0H6FVLTpyb/3NQUgKLv12BPS/PwiIQQQghxMaxOdQk2yaQLIYSoTBKkX0Gt+w7ncCRobQ4yvvrK08MRQgghxEWQddKFEEJcCRKkX0H96w/gh85qiVz6ooU48/M9PCIhhBBCVJQE6UIIIa4ECdKvoCCvIPS9upIUBOTkkfHll54ekhBCCCEqqGROugTpQgghKpPe0wO41gxsOIilPTbz+ConaR99TMDgwRgiIz09LCGEEKJSzZ07l7feeoukpCRatGjB7Nmz6dGjR7n7btiwgd69e5fZfvDgQZo2bVrZQz2nkky6zEkX1xKHw4HNZvP0MISoEoxGI1rt5efBJUi/wvrU6cP/Wnlx4M8CmscVcfrNt6g9W9ZOF0IIUX0tWbKESZMmMXfuXLp168bHH3/MwIEDOXDgAHXq1DnncYcOHcLf39/1vEaNGldiuOdU0jjOqJVMuqj+FEUhOTmZrKwsTw9FiCpDq9VSv359jMbL+56QIP0K8zH40DOqFzH9f+KtGCe5a9aQv304Pl26eHpoQgghRKWYNWsW48ePZ8KECQDMnj2bn376iQ8//JDp06ef87iwsDACAwOv0CgvTOaki2tJSYAeFhaG2WxGo9F4ekhCXNWcTieJiYkkJSVRp06dy/o3I0G6B9zW6DZ+PvUzv15nos8fRSS/+hoNVq5AY5B1V4UQQlQvVquVXbt28eyzz5ba3r9/f7Zt23beY9u1a0dRURHNmzfnv//9b7kl8CUsFgsWi8X1PCcn5/IGXg4pdxfXCofD4QrQQ0JCPD0cIaqMGjVqkJiYiN1ux3AZsZ00jvOArpFdCfMO48uuNhwBPliPHSNj4UJPD0sIIYRwu7S0NBwOB+Hh4aW2h4eHk5ycXO4xERERfPLJJyxfvpxvv/2WJk2a0LdvXzZt2nTO60yfPp2AgADXIyoqyq3vA6RxnLh2lMxBN5vNHh6JEFVLSZm7w+G4rPNIkO4Beq2ewY0Gk++tYcOg2gCkzXkfe2qqh0cmhBBCVI5/l/0pinLOUsAmTZpw33330b59e7p06cLcuXMZNGgQb7/99jnPP3XqVLKzs12PuLg4t45fURRsTjVwkSBdXCukxF2Ii+OufzMSpHvIbQ1vA+CzusfRtWiKMz+fuIcfIWXWO2R+vYS8zZuxxsZ6eJRCCCHE5QkNDUWn05XJmqekpJTJrp9P586dOXLkyDlfN5lM+Pv7l3q4U0nTOJDGcUIIISqXBOkeUi+gHu3D2uNAYfd/2oNWS9G+faR/8gnJL79M3H33c6z/ADKXLfP0UIUQQohLZjQa6dChA2vXri21fe3atXTt2rXC5/nzzz+JiIhw9/AqrKTUHWROuhDCcxYsWHBVNdQUlUOCdA8a0mgIAAs1f1Bv2VLCnnyCoLvuwrdXLwzFc+lyVq/24AiFEEKIyzdlyhQ+++wzYmJiOHjwIJMnTyY2NpYHH3wQUEvVx4wZ49p/9uzZrFy5kiNHjvD3338zdepUli9fziOPPOKpt+BqGgfqtDUhxNXn1ltv5cYbbyz3te3bt6PRaNi9e7dr2/Lly+nTpw9BQUGYzWaaNGnCuHHj+PPPP0sda7Vaeeutt2jfvj0+Pj4EBATQpk0b/vvf/5KYmFju9caOHYtGoznv41KMGDGCw4cPX9Kx5WnSpAlGo5GEhAS3nVNcPgnSPah/vf546705mXOSf2pYCZkwgZovvkDURx8S9fHHABTu3IWzsNDDIxVCCCEu3YgRI5g9ezbTpk2jbdu2bNq0idWrV1O3bl0AkpKSiD1ripfVauXJJ5+kdevW9OjRgy1btvDDDz8wdOhQT72FUp3dZZ6uEFen8ePHs379ek6dOlXmtZiYGNq2bUv79u0BeOaZZxgxYgRt27Zl1apV/P3333zyySc0bNiQ5557znWcxWKhX79+vP7664wdO5ZNmzaxa9cu3nzzTdLT05kzZ065Y3n33XdJSkpyPQDmz59fZlsJq9Va3mnK8Pb2JiwsrEL7XsiWLVsoKirizjvvZMGCBW455+UoaVgoAOUak52drQBKdna2p4eiKIqi/HfLf5WWC1oqL2x5odR2p9OpHO7dWznQpKmSu2mzh0YnhBDiSrjavpuqA3d/pieyTigtF7RUuizq4pbzCXE1KywsVA4cOKAUFhZ6eigXxWazKeHh4crLL79cant+fr7i5+enzJkzR1EURdm+fbsCKO+++26553E6na7/nj59uqLVapXdu3dfcN/zAZQVK1a4nvfs2VN5+OGHlcmTJyshISHKDTfcoCiKosycOVNp2bKlYjabldq1aysPPfSQkpub6zpu/vz5SkBAgOv5Sy+9pLRp00b54osvlLp16yr+/v7KiBEjlJycnAuOaezYscqzzz6r/Pjjj0qDBg3KvJe4uDhlxIgRSlBQkGI2m5UOHToov/32m+v17777TunQoYNiMpmUkJAQ5fbbbz/n+1UURQkICFDmz5+vKIqinDhxQgGUJUuWKD179lRMJpMSExOjpKWlKSNHjlRq1aqleHt7Ky1btlQWLVpU6jwOh0OZMWOG0rBhQ8VoNCpRUVHKq6++qiiKovTu3Vt5+OGHS+2flpamGI1GZd26dRf8TC7X+f7tXMz3kscz6XPnzqV+/fp4eXnRoUMHNm/efN79Fy5cSJs2bTCbzURERHDvvfeSnp5+hUbrfiUl7z+d/IkCW4Fru0ajwad4rl7+1q2eGJoQQgghirmWX5OmceIapSgKBVa7Rx6KolRojHq9njFjxrBgwYJSxyxbtgyr1cp//vMfABYvXoyvry8TJ04s9zxnV8ssXryYfv360a5duwvue7E+//xz9Ho9W7du5ePiKlqtVst7773H/v37+fzzz1m/fj1PP/30ec9z7NgxVq5cyffff8/333/Pxo0bmTFjxnmPyc3NZdmyZYwePZp+/fqRn5/Phg0bXK/n5eXRs2dPEhMTWbVqFXv37uXpp5/G6XQCuKqbBg0axJ9//sm6devo2LHjRX8GzzzzDI899hgHDx5kwIABFBUV0aFDB77//nv279/P/fffz913383vv//uOmbq1Km88cYbvPDCCxw4cIBFixa5GpFOmDCBRYsWYbGc6SOycOFCIiMj6d2790WPz1M8OqlqyZIlTJo0iblz59KtWzc+/vhjBg4cyIEDB6hTp06Z/bds2cKYMWN45513uPXWW0lISODBBx9kwoQJrFixwgPv4PK1D2tPHb86xObG8vOpn11BO4Bvt25kf7Oc/G3bPDdAIYQQQriWX5OmceJaVWhz0PzFnzxy7QPTBmA2VixsGTduHG+99RYbNmxwBWUxMTEMHTqUoKAgAA4fPkyDBg3Q68+cc9asWbz44ouu5wkJCQQEBHD48GF69epV6hq33367qxlm69at2XaJv6s3atSIN998s9S2SZMmuf67fv36vPLKKzz00EPMnTv3nOdxOp0sWLAAPz8/AO6++27WrVvHa6+9ds5jvv76a6Kjo2nRogUAI0eOZN68ea7PbNGiRaSmprJjxw6Cg4Nd4y3x2muvMXLkSP73v/+5trVp06aC77z0+/33VKYnn3zS9d+PPvooa9asYdmyZXTq1Inc3Fzeffdd3n//fe655x4AGjZsSPfu3QEYNmwYjz76KN999x3Dhw8H1GkGJT0CqgqPZtJnzZrF+PHjmTBhAs2aNWP27NlERUXx4Ycflrv/b7/9Rr169XjssceoX78+3bt354EHHmDnzp1XeOTuo9FouD36dgC+OfxNqbt+5s6dQaPBcvgwtpQUTw1RCCGEuOa5MumyRroQV7WmTZvStWtXYmJiADXLvHnzZsaNG1dqv38HbOPGjWPPnj18/PHH5Ofnl/qd/N/7zp07lz179jBu3DgKCgq4VOVlnn/99Vf69etHrVq18PPzY8yYMaSnp5Ofn3/O89SrV88VoANERESQcoHYYd68eYwePdr1fPTo0Xz77bdkZWUBsGfPHtq1a+cK0P9tz5499O3b97zXqIh/fwYOh4PXXnuN1q1bExISgq+vLz///LOrb8nBgwexWCznvLbJZGL06NGuv/89e/awd+9exo4de9ljvZI8lkm3Wq3s2rWLZ599ttT2/v37n/NuVNeuXXn++edZvXo1AwcOJCUlhW+++YZBgwad8zoWi6VUuUNOTo573oAb3drgVubumcve1L3E7I9hfKvxAOiDgvBq0YKi/fvJ37aNwCFDPDtQIYQQ4holQbq41nkbdByYNsBj174Y48eP55FHHuGDDz5g/vz51K1bt1RQFx0dzZYtW7DZbBgMBgACAwMJDAwkPj6+1Lmio6P5559/Sm0rWQ7yXAFsRfn4+JR6furUKW6++WYefPBBXnnlFYKDg9myZQvjx48/b1O1kvdQQqPRuMrSy3PgwAF+//13duzYwTPPPOPa7nA4WLx4MQ899BDe3t7nHfuFXtdoNGWmKZT3Hv79GcycOZN33nmH2bNn06pVK3x8fJg0aZKrsd6FrgtqyXvbtm2Jj48nJiaGvn37uhqVVhUey6SnpaXhcDhc8wdKhIeHk5ycXO4xXbt2ZeHChYwYMQKj0UjNmjUJDAw8Z1dFgOnTpxMQEOB6RBUvbXY1CfcJ59nr1ZsV7/35HtsSz9ykcM1Ll5J3IYQQwmNsDvWXS5mTLq5VGo0Gs1HvkcfFlikPHz4cnU7HokWL+Pzzz7n33ntLnWPUqFHk5eWdt4T87H3Xrl1bZlm2yrBz507sdjszZ86kc+fONG7c+JxLvF2OefPmccMNN7B371727Nnjejz99NPMmzcPUMv49+zZQ0ZGRrnnaN26NevWrTvnNWrUqFGqg/2RI0cqVHWwefNmbrvtNkaPHk2bNm1o0KABR44ccb0eHR2Nt7f3ea/dqlUrOnbsyKeffsqiRYvKVFFUBR5vHPfvf3SKopzzH+KBAwd47LHHePHFF9m1axdr1qzhxIkTrnVWyzN16lSys7Ndj7i4OLeO313ubHwnQ6OH4lScPL3paRLy1LUKfbp1AyB/2/YKN80QQgghhHtJJl2IqsPX15cRI0bw3HPPkZiYWKbUuUuXLjzxxBM88cQTTJkyhS1btnDq1Cl+++035s2bh0ajQatVw6TJkyfTpUsX+vTpw7vvvsvu3bs5ceIEP/30Ez/++CM63cVl+c+nYcOG2O125syZw/Hjx/nyyy/56KOP3HZ+ULPZX375JaNGjaJly5alHhMmTGDXrl3s3buXUaNGUbNmTYYMGcLWrVs5fvw4y5cvZ/v27QC89NJLLF68mJdeeomDBw/y119/lZpf36dPH95//312797Nzp07efDBB8tk/MvTqFEj1q5dy7Zt2zh48CAPPPBAqQSul5cXzzzzDE8//TRffPEFx44dc/29nW3ChAnMmDEDh8PB7bff7qZP78rxWJAeGhqKTqcrkzVPSUkpk10vMX36dLp168ZTTz1F69atGTBgAHPnziUmJqbMWoMlTCYT/v7+pR5XI41Gw3OdnqNlSEuyLdlM/nUyRfYivNu1RePtjSMtDcvhw54ephBCCHFNsjrPrJMuhLj6jR8/nszMTG688cZyG1K//fbbLFq0iD///JNbbrmF6Oho7rzzTpxOJ9u3b3fFDF5eXqxbt45nn32W+fPn0717d5o1a8akSZPo1q0bK1eudNuY27Zty6xZs3jjjTdo2bIlCxcuZPr06W47P8CqVatIT08vN3CNjo6mVatWzJs3D6PRyM8//0xYWBg333wzrVq1YsaMGa6bEr169WLZsmWsWrWKtm3b0qdPn1Id2GfOnElUVBQ33HADd911F08++SRms/mC43vhhRdo3749AwYMoFevXq4bBf/e54knnuDFF1+kWbNmjBgxoswc/FGjRqHX67nrrrvw8vK6hE/KszSKB9OznTp1okOHDqVKTZo3b85tt91W7g/ksGHD0Ov1LFmyxLVt+/btdO3alYSEBCIjIy94zZycHAICAsjOzr4qA/bk/GRGfD+CjKIMBjcczKvdXiXuwQfJ37iJsKeeImR81SvXEEIIcX5X+3dTVeTuz/TbI9/y0raX6Fm7J+/3fd8NIxTi6lVUVMSJEydcyyQLUdXExcVRr149duzYQfv27a/Ydc/3b+divpc8Wu4+ZcoUPvvsM2JiYjh48CCTJ08mNjbWVb4+depUxowZ49r/1ltv5dtvv+XDDz/k+PHjbN26lccee4zrr7++QgF6VVDTpyZv3fAWWo2WVcdWseLoCnxLSt5lvXQhhBDCI6wONZMu5e5CCHH1stlsxMbG8swzz9C5c+crGqC7k0eD9BEjRjB79mymTZtG27Zt2bRpE6tXr3Z130tKSnK12wcYO3Yss2bN4v3336dly5bceeedNGnShG+//dZTb6FSXB9xPY+1ewyA2btmo1zXGoCCXbtwFhW59rNnZpIeMx/LWc0UhBBCCOF+MiddCCGuflu3bqVu3brs2rXL7fP5rySPLcFWYuLEiUycOLHc1xYsWFBm26OPPsqjjz5ayaPyvDEtxrDq2CqOZx9nXu5P3B4ejv30aQp27cK3Wzfytm4laepz2FNSSJ0zh8jp0/G/yTPLYgghhBDVXUkmXeakCyHE1atXr17Votm2x7u7i/IZtAaevu5pABb/8zXOjq0AyPt1A6enTydu/ATsKSlovL1RCgtJmDSJ1PfmoJy1JqJit5Pz44/ETXyYzK+XlHsdIYQQQlxYSeM4g/bC3YmFEEKIyyFB+lWsW61u9KzdE7tiZ3UNdUm2zK++IuPzLwAIumsU0Zs3E3zPPQCkzZ1LwuOTsCUnkz4vhqP9+5MweQp569dz+rXXsJ1OOee1hBBCCHFuJeXukkkXQghR2SRIv8o92fFJ9Fo93/ifWX5NFxJC7Y8+pOaLL6Lz9SF86rNEvPYaGoOB3LVrOdqrNylvvYU9MQldcDCG2rVRbDYyypk+IIQQQogLszlsgATpQgghKp8E6Ve5egH1GN1sNDk+Gn7sH4z/sNtpsOo7/Hr1KrVf4LCh1Pnic3ShoQCYoqOJeO1VGv26npov/BeAzCVLcGRlXeF3IIQQQlR9JZl0g07K3YUQQlQuCdKrgPtb30+wVzDzO+Sw7q6m6ENCyt3P3K4dDb//P+otW0r9Vd8ROGwYWpMJnxtuwNS0KUpBARlfLbzCoxdCCCGqPmkcJ4QQ4kqRIL0K8DP6uZZke3vn2wxeOZipm6fy1YGv2JOyB6dyplmcLjAQ71at0Gg0rm0ajYbQ++8DIPPLL3Hm51/ZNyCEEEJUca510rWyBJsQQojKJUF6FTGk0RD61umLgsKJ7BN8f/x73tjxBnf/eDdTNkzB4XSc93i/AQMw1K2DIzubzKXLrtCohRBCiOpB1kkXQnjCggULCAwMdD1/+eWXadu27XmPGTt2LEOGDKnUcYnKJUF6FaHT6pjdezYbhm9gbt+5PNz2YXpF9cKoNbIudh3v7HrnvMdrdDpCJkwAIGP+fJxWq+s1xWol+//+j/zffqvU9yCEEEJUVSVLsEmQLsTV69Zbb+XGG28s97Xt27ej0WjYvXu3a9vy5cvp06cPQUFBmM1mmjRpwrhx4/jzzz9LHWu1Wnnrrbdo3749Pj4+BAQE0KZNG/773/+SmJhY7vWWL1+OTqcjNja23NebNm3KY489dtHv8cknn2TdunUXfdy5NGnSBKPRSEJCgtvOKS6fBOlVTIh3CD1q9+DBNg8yp88cXuv+GgCfH/icbw5/c95jA267DX14OPaUFLK/+w7FZiNz2TKO3nQTiU89Tey48eT//seVeBtCCCFElSJz0oW4+o0fP57169dz6tSpMq/FxMTQtm1b2rdvD8AzzzzDiBEjaNu2LatWreLvv//mk08+oWHDhjz33HOu4ywWC/369eP1119n7NixbNq0iV27dvHmm2+Snp7OnDlzyh3L4MGDCQkJ4fPPPy/z2tatWzl06BDjx4+/6Pfo6+tLyDn6U12sLVu2UFRUxJ133smCq2AVKJvN5ukhXDUkSK/ibqp/ExPbTgTgtd9e47ekc2fDtUYjwfeOBSDtg7kcu3kQyS+8iD0xCY3BAE4nCVOmYDt9+koMXQghhKgyXHPSJZMurlWKAtZ8zzwUpUJDvOWWWwgLCysTcBYUFLBkyRJXUPzbb7/x5ptvMmvWLGbNmkWPHj2oX78+PXv25Pnnn2f16tWuY9955x22bNnC+vXreeyxx+jQoQONGjViwIABfPjhh7z++uvljsVgMHD33XezYMEClH+NPyYmhg4dOtCmTRtmzZpFq1at8PHxISoqiokTJ5KXl3fO9/jvcneHw8GUKVMIDAwkJCSEp59+usz1zmXevHncdddd3H333cTExJQ5Lj4+npEjRxIcHIyPjw8dO3bk999/d72+atUqOnbsiJeXF6GhoQwdOtT1mkajYeXKlaXOFxgY6Pq7OXnyJBqNhqVLl9KrVy+8vLz46quvSE9PZ9SoUdSuXRuz2UyrVq1YvHhxqfM4nU7eeOMNGjVqhMlkok6dOrz2mpq47NOnD4888kip/dPT0zGZTKxfv75Cn8vVQO/pAYjL92DrBzmZfZLVJ1YzZcMUvrr5KxoENCh336A77yT9o4+xJycD6prrIfdNIOC224gdey+WQ4dImDSZup8vQGOUX0SEEEIIOGtOujSOE9cqWwG8HumZaz+XCEafC+6m1+sZM2YMCxYs4MUXX3Q1Ul62bBlWq5X//Oc/ACxevBhfX18mTpxY7sP2504AAEtKSURBVHnObsC8ePFi+vXrR7t27S6477+NHz+eWbNmsXHjRnoVL5+cn5/P0qVLefPNNwHQarW899571KtXjxMnTjBx4kSefvpp5s6de8H3CzBz5kxiYmKYN28ezZs3Z+bMmaxYsYI+ffqc97jc3FyWLVvG77//TtOmTcnPz2fDhg307t0bgLy8PHr27EmtWrVYtWoVNWvWZPfu3TidasPqH374gaFDh/L888/z5ZdfYrVa+eGHHyo05rM988wzzJw5k/nz52MymSgqKqJDhw4888wz+Pv788MPP3D33XfToEEDOnXqBMDUqVP59NNPeeedd+jevTtJSUn8888/AEyYMIFHHnmEmTNnYjKplU8LFy4kMjLS9d6qAsmkVwMajYZp3abRtkZbcq25PLLuEZLzk8vdV+vjQ/jzz2Fq3JgaT0yh0dqfCRk7Fn1QELXfexetnx+Ff/7J6bfevsLvQgghhLh62ZxqGaaUuwtxdRs3bhwnT55kw4YNrm0xMTEMHTqUoKAgAA4fPkyDBg3Q68/kK2fNmoWvr6/rkZ2d7dq3SZMmpa5x++23u/br2rXrOcfSvHlzOnXqxPz5813bli5disPhYNSoUQBMmjSJ3r17U79+ffr06cMrr7zC0qVLK/x+Z8+ezdSpUxk2bBjNmjXjo48+IiAg4ILHff3110RHR9OiRQt0Oh0jR45k3rx5rtcXLVpEamoqK1eupHv37jRq1Ijhw4fTpUsXAF577TVGjhzJ//73P5o1a0abNm1KTROoqEmTJjF06FDq169PZGQktWrV4sknn6Rt27Y0aNCARx99lAEDBrBsmdr4Ojc3l3fffZc333yTe+65h4YNG9K9e3cmFPfeGjZsGBqNhu+++851jfnz5zN27Njz3lC52kgmvZow6UzM7j2b/6z+D3G5cfxn9X+Y23cuTYKblNk34NZbCbj11jLbjXXrEvnGDOInPkzml1/i3aYNAbcMOu917RkZ6AID0Wjlfo8QQojqS7q7i2uewaxmtD117Qpq2rQpXbt2JSYmht69e3Ps2DE2b97Mzz//XGq/fwds48aNY/Dgwfz++++MHj26VOn3v/edO3cu+fn5vPfee2zatOm84xk/fjyTJk3i/fffx8/Pz3XDoKRj+6+//srrr7/OgQMHyMnJwW63U1RURH5+Pj4+568eyM7OJikpyRU4g1pN0LFjxwuWvM+bN4/Ro0e7no8ePZobbriBrKwsAgMD2bNnD+3atSM4OLjc4/fs2cN999133mtURMeOHUs9dzgczJgxgyVLlpCQkIDFYsFisbg+i4MHD2KxWOjbt2+55zOZTIwePZqYmBiGDx/Onj172Lt3b5nS+6udRFbVSIh3CDEDYmgY0JCUghTuWXMP2xK2XdQ5/Pr0IeT++wFIeuEFCvf/Xe5+itNJyttvc6RrN5Jfevlyhy6EEEJc1SRIF9c8jUYtOffE4yIzoOPHj2f58uXk5OQwf/586tatWyqoi46O5tixY6UalQUGBtKoUSNq1apV6lzR0dGuUuoSERERNGrU6JwB7NlGjhyJRqNhyZIlHD16lC1btrjmxp86dYqbb76Zli1bsnz5cnbt2sUHH3wAVG4TtQMHDvD777/z9NNPo9fr0ev1dO7cmcLCQtf8b29v7/Oe40KvazSaMjcKyntP/74RMXPmTN555x2efvpp1q9fz549exgwYADW4pWpLnRdUEve165dS3x8PDExMfTt25e6dete8LiriQTp1UykbyRf3PwF19e8nnxbPhPXTeTbI99e1DlqPP4Y5i6dUQoLOXXXXWR8tbDUPzJnYSEJj08i/TO1JCbrm2+wHDvm1vchhBBCXE1sDvWXS5mTLsTVb/jw4eh0OhYtWsTnn3/OvffeWyobPmrUKPLy8io073vUqFGsXbu2zLJsFeXn58edd97J/PnziYmJoUGDBq756Tt37sRutzNz5kw6d+5M48aNz7mkW3kCAgKIiIjgt7OWUbbb7ezateu8x82bN48bbriBvXv3smfPHtfj6aefdpW8t27dmj179pCRkVHuOVq3bn3epeBq1KhBUlKS6/mRI0coKCi44HvavHkzt912G6NHj6ZNmzY0aNCAI0eOuF6Pjo7G29v7vNdu1aoVHTt25NNPP2XRokWMGzfugte92kiQXg35G/356MaPuLXBrTgUBy9te4l3dr2Dw+mo0PEanY7as2fj26sXitXK6VdfJX7iw9gzM7GnpnJqzD3krl2LxmDAFN0IFIW0jz+u5Hd1hj0jg7yNG0md8z7xkyaTd4EyIyGEEOJylWTSZU66EFc/X19fRowYwXPPPUdiYiJjx44t9XqXLl144okneOKJJ5gyZQpbtmzh1KlT/Pbbb8ybNw+NRoO2eCrn5MmT6dKlC3369OHdd99l9+7dnDhxgp9++okff/wRnU53wfGMHz+ebdu28eGHHzJu3DjXDYOGDRtit9uZM2cOx48f58svv+Sjjz66qPf6+OOPM2PGDFasWME///zDxIkTycrKOuf+NpuNL7/8klGjRtGyZctSjwkTJrBr1y727t3LqFGjqFmzJkOGDGHr1q0cP36c5cuXs337dgBeeuklFi9ezEsvvcTBgwf566+/XM3wQO2y/v7777N792527tzJgw8+iMFguOD7adSoEWvXrmXbtm0cPHiQBx54gOTkM722vLy8eOaZZ3j66af54osvOHbsmOvv7WwTJkxgxowZOBwObr/99ov6TK8GEqRXUwadgde6v8YDrR8AIGZ/DA/88gDphekVOl4XEEDtD+cS/vzzaAwG8n79lRO3DeHEiBEU/fUXusBA6syPIWL6DAByvv8BazlrUlaEIyeH+Mcnkfj88yjFHSPLk/H55xzt05cjXbsR98CDpH3wAblr1pDw1NPYMzMv6dpCCCFERVidsgSbEFXJ+PHjyczM5MYbb6ROnTplXn/77bdZtGgRf/75J7fccgvR0dHceeedOJ1Otm/fjr+/P6AGhevWrePZZ59l/vz5dO/enWbNmjFp0iS6detWobnO3bt3p0mTJuTk5HDPPfe4trdt25ZZs2bxxhtv0LJlSxYuXMj06dMv6n0+8cQTjBkzhrFjx9KlSxf8/PzOG5SuWrWK9PT0cveJjo6mVatWzJs3D6PRyM8//0xYWBg333wzrVq1YsaMGa6bEr169WLZsmWsWrWKtm3b0qdPn1LLs82cOZOoqChuuOEG7rrrLp588knM5gv3FnjhhRdo3749AwYMoFevXq4bBf/e54knnuDFF1+kWbNmjBgxgpSUlFL7jBo1Cr1ez1133YWXl9cFr3u10SgVXUivmsjJySEgIIDs7GzXP77q7scTP/LStpcotBcSZg5jZs+ZtA1rW+Hjiw4eJGHKE1hPnADAWK8eUR9/hLF4bkfs/feTv2kzAcOGElm8RmFFOXJziR0/gaJ9+wCo+co0gu68s8x++X/8QeyYM/9TMzZogHerlhTu+wvriRMEjhpJxEsvXdS1hRDianEtfjdVNnd+pk7FSZsv2gCwacQmgryC3DFEIa5aRUVFnDhxgvr161fJAEeIuLg46tWrx44dO2jfvv0Vu+75/u1czPeSZNKvAQPrD2TxoMXUD6hPSkEK9665ly8PfEmB7cLzQgC8mjWj/vJvCB47loDbBlPv68WuAB0g9KGHAMj+bhXW+IQKj8uRl0/c/Q+oAXrxEhipM2fh+FeJjmKzcfqVVwAIuP12Gu/4g4arfyDyjTeo+b+XAchaspSifzX1EEIIIdzB6rC6/lvK3YUQ4upls9mIjY3lmWeeoXPnzlc0QHcnCdKvEQ0DG7J40GIG1BuAXbHz5o436ba4G3f9cBdv7XiLdafWkW3JPufxWrOZ8GefIfKNN9AVLxlRwtyuHeYuncFuJ/2zTys0HmdBAfEPPkjhn3+iDQig3uJF/H97dx5e07U+cPx7hpzMkyQyNCQxRMzaBDUURWNqjUXNs4uaqsbSxlhUDVUNWhHX0BaXll5UUNwaSqmomucEiYghg0Sms35/qPPraUKiIgnez/OcR7L32nuvvZ4467xnrf0uy7JlyLpzh7jPPjMre2v5CtLOnkPn7Iz7mNHo7O1N+2xr1MC+WVMwGrk+dVquy00IIYQQj+vB8+hw/3EyIYQQRdPevXvx8fHh8OHDj/18f1Ei66S/QGwtbJlVbxYvF3+Z8D/CuZ5ynWPxxzgWf4zlJ5Zjo7ehZ8WedK/YHVuLR6/L+HeuAwcStf8XEtatx3XgQCzc3clKSuLO6tXcXr0GlMJQuhSWpctgWboUCT/8l5RDh9Da2VFyyRKsK1fCfcKHRPXowZ1vV+P09ttYV6xIRmwsN/5ciqL4yJHZviAAcB81iuSdu0g5dIikH3/EoVmz/GguIYQQAvj/kXStRoteIx+dhBCiqGrQoMFzMWgnPc0LRqPR0KV8FzoHdCbmbgxH4o5wJO4IB2IOcCnxEqFHQ/nm1Df0r9KfDuU65DlBjm2NGlgHBZJ66DA35s5D7+bK7W++xZicbCqTceUKd3f/fyZ2ra0tJZd8hXXlSvfPUbMGDs2bk7h5M9enTMXn61Vcnz4DlZKC9csv49imdY7XtvDywqVfX+I/X8D1T2Zh16AB2jysoSiEEELkhSlpnNZgtoyTEEII8TRIkP6C0mg0eNl54WXnRYtSLVBKse3yNj4/8jmXEi8x89eZLD+xnD6V+tCqTCus9LknDXEdOJDoPn1J+EuWS0OZ0rj07oOhhDdp5y+Qdv486efPkXUnAfcJ47GuVs3sHMXHjCZ51y5SIyOJGT+BpK1bQavFI+QjNNqHP53h0qcPCevWk3HtGje/WoLb0CH/tGmEEEIIMw+mu0tmdyGEEAVBgnQB3A/ag32DaViyIRvObSD0aCgxd2OYemAqoUdD6RTQiXfKvYOTldNDz2FbuzY2NWuScuAA1q+8gkvfvtg1qG8Krm2qV8+1Hhbu7ri+O4i4WZ+S8N13ADh37YJVQMAjj9NaWVF8zBiuDhvGzSVLsK31ap6uJ4QQQuQmIysDkKRxQgghCoYE6cKMXqunnX87WpRqwbqz61hxYgVXk6/yReQXLP1jKa+99BrWemv0Wj16rR4bvQ2ty7amlGMpNBoNJRaGkhkfjyGH9Sjzqli3btxZt570CxfQu7nhNnRono6zD34D2zp1uLt3L5d79MRt6FBc+vd75Aj8k1JKydRHIYR4zslIuhBCiIIk2d1Fjqz0VnQp34X/tvkvM1+bSUCxAFIzU4m4HMGG8xtYd3Ydq0+vJvx4OD229OBy4mXgfhb4JwnQATQGA17TP8aqahU8p09HZ2eXt+M0Grznf4Zjq1ZgNHJj3jyi+/Un8+bNRx6XlZhI7McfEzdv3mMlmri7bx9n69UjVrLKCyHEc+1B4jgJ0oUQQhQEGUkXj6TX6mleqjnN/JpxMPYgp26dItOYef+lMtkZtZPTt08zaPsgVjZfibOVc75c17pqVfxWr37s47S2tnjNnIFNzZrETp7M3b17udi6DR5TJmNXv362Ue+Uw4e5OmoUmddigPtJ6Jw7dMj1OqnH/iB68BBUSgq3V67Espw/zu3b56mOGdevk7R1Kw5vvYXeOX/aSwghxNNjCtK1EqQLIYR4+mQkXeSJRqOhpmdNelTsQZ/KffhX1X/xbrV3WfTGIrxsvYhKimLIT0O4l3mvsKsKgFPbNvitXYOhdGkyb9zgyoCBXHjrLW5/uxpjaioqM5Mbny/gcrfuZF6LQfvn2utxM2aScfXqI8+ddvEi0f37o1JS0BcvDsD1qdO4d/pMrvXKSkoiqkdPrn88naievci6c+eJ7/VpyIiJ4er7I7l74GBhV0UIIQrdg+nu8ky6EOJ5s2vXLjQaDXeK6GfSF5UE6eKJuFq7srDxQuwN9hy9cZQP9nyAURmB+89rRydGs+nCJo7dOFbgdbMsWxa/tWso1qsXWltb0s+dJ3biRM42eJ2Lb7cn/osvwGjEsVUryuzYjnVgIMaUFK6Nn4AyGnM8Z8b1OKL79CXr9m2sKlSg1Kb/Yvvaa6i0NK4OH47x7t2H1kcZjVwbOYr0S5cASDt9mqj+/yIr+eHHFAalFDEhISRu2sSVwYNJj44u7CoJIUSherAEm4XOopBrIoTIi+joaPr06YOXlxcGgwEfHx+GDRvGzRwegTx37hy9evXC29sbS0tL/Pz86NSpE4cOHTIrt3PnTpo3b46Liws2NjZUqFCB999/n6t/Du48CHYrVapEVlaW2bFOTk4sW7bM9Luvry8ajYZffvnFrNzw4cNp0KBBjvc0ceJENBrNI1+X/vyM+Thq165NTEwMjo6Oj31sToKDg9HpdNnuTTweCdLFEyvlVIrPXv8MC60F2y5vY/T/RjN692gar21M8++aM/bnsXTe3Jkum7vw46UfyTRmFljdtDY2uI8ZTZndu3D/YBwWJUpgTEgg7dSp+1PjZ83Ca+YMdA4OeH08DY2VFSm//MKdHKbaZyUmEt2vHxnXrmHhU5ISX32Jzt4er5kz0BcvTvrFi8RMmvTQ59NvfP45ybt3ozEY8JwxHZ2jI/d+/50rgwZhvFc0ZiAAJO/cxd3//QyAMSmJK8OGYUxLK+RaCSFE4Xkw3V1G0oUo+i5cuEBQUBBnzpzhm2++4dy5cyxatIgdO3ZQq1Ytbt26ZSp76NAhAgMDOXPmDIsXL+bEiRN89913BAQE8P7775vKLV68mMaNG+Ph4cG6des4ceIEixYtIiEhgdmzZ5td//z58yxfvjzXelpZWTFmzJg839fIkSOJiYkxvby9vZk8ebLZthIlSpjKp6en5+m8BoMBDw+PfEmEHBUVxf79+xk8eDBhYWFPfL4nlZGRUdhV+MckSBf5orpHdabUmQLA1ktb2XJpC3GpcVhoLajoUhELrQW/3/idUbtH0Xx9c/59/N+kZqYWWP10dnYU696d0j9uwTv0C4r17o3f99/h+NabpjIGHx+K//mGfH3Wp6YRZGN6OnfWreNSh46knTmDzs2VkmFh6F1cANAXK8ZLc2aDVkvixh9IWL8+2/UTt0Zwc+EiADwmT8KpdWtKLFmC1taWlIMHuTJ0KCqPb6ZPkzEtjevTpwPg2KYNOmdn0k6cNG0TQogXkTyTLsT9mXYpGSmF8nqcBL3vvvsuBoOBiIgI6tevT8mSJWnWrBnbt2/n6tWrjB8/3nQ/PXv2pGzZsvz888+0aNGC0qVLU61aNUJCQtiwYQMAV65cYejQoQwdOpSlS5fSoEEDfH19qVevHkuWLOGjjz4yu/6QIUMICQnhXi4DMP/617/45Zdf2Lx5c57uy87ODg8PD9NLp9Nhb29v+n3s2LG0a9eO6dOn4+Xlhb+/PwArV64kKCjIVLZz587ExcWZzvv36e7Lli3DycmJrVu3Ur58eezs7GjatCkxMTG51jE8PJw333yTgQMHsnr1au7+bYbpnTt36N+/P+7u7lhZWVGpUiX++9//mvbv3buX+vXrY2Njg7OzM02aNOH27dvA/dkH8+bNMztftWrVmDhxoul3jUbDokWLaNWqFba2tkydOpWsrCz69OmDn58f1tbWlCtXjs8++yxb3ZcuXUrFihWxtLTE09OTwYMHA9C7d2/efPNNs7KZmZl4eHiwdOnSXNvkn5LEcSLftCjVgpTMFHZG7aSya2WCPIKo7FoZK70V8anxrDm9htWnVxNzN4ZPD33KypMrGRE4gqa+TQtsGTONTod9w4bYN2yY437nLp1Jiogg5ddfiRn3AbZ163Jr1UqybsQDoHNyouSSJRi8vc2OswkKwm3oUG7Mm0fspMkk79qNdeAr2AQGotHruTZuHADFenTHqXVrAKwrV6LE4kVE9e3H3f/9zJWhw/CYNBELd/d8u9+spCSSIiIw+Plh88oruZa/FR5ORnQ0+uLF8ZgwnpQWLYju1487367GJjDI7EsNIYR4UcgSbEJAamYqNb+uWSjXPtD5ADYWNrmWu3XrFlu3bmXatGlYW1ub7fPw8KBLly6sXr2a0NBQIiMjOX78OF9//TXaHJbrdXJyAmDt2rWkp6czevToHK/5oNwDw4cPZ+XKlSxYsICRI0c+tK6+vr4MGDCAcePG0bRp0xzr8Lh27NiBg4MD27ZtM32xkZ6ezpQpUyhXrhxxcXG899579OzZ85FfDqSkpPDpp5+yYsUKtFotXbt2ZeTIkaxateqhxyilCA8P54svviAgIAB/f3/WrFlDr169ADAajTRr1oykpCRWrlxJ6dKlOXHiBDqdDoDIyEgaNWpE7969mT9/Pnq9np07d2Z7dCA3ISEhTJ8+nblz56LT6TAajXh7e7NmzRpcXV3Zt28f/fv3x9PTkw5/JoteuHAhI0aMYMaMGTRr1oyEhAT27t0LQN++falXrx4xMTF4enoCsHnzZpKTk03HPw0SpIt81d6/Pe39s2c5d7V2ZVC1QfSp3IdNFzax+Ohirt29xuj/jebbU98ypsYYKrhUKIQam9NotXh+PI0LrVqTcugQKX8+j6R3d6dY9+44dWiP7s8kc3/n0r8fqUePkrxzJ0nbtpG0bZvZfptXX6X4qFHm24KC8F6wgCsDB5K8axfnmzSlWK+euPTpi87O9h/fR9q5c9xatYqEDRtRKSloLC0pvXkTFi+99NBjMmJiiF/8JQDFR41Ca2uLXd06uA4cQHzoQmJCQrCqUB7L0qX/cb2EEOJZlGG8P2VSprsLUbSdPXsWpRTly5fPcX/58uW5ffs2N27c4OzZswAEBATkek4HBwdTgJYbGxsbQkJC+OCDD+jXr98jn/WeMGEC4eHhrFq1im7duuXp/I9ia2vLkiVLMBj+/wvF3r17m34uVaoU8+fPp0aNGiQnJ2P3kGWOMzIyWLRoEaX//Mw3ePBgJk+e/Mhrb9++nZSUFJo0aQJA165dCQsLMwXp27dv5+DBg5w8edI0yl+qVCnT8Z988glBQUGEhoaatlWsWPFxbh+Azp07m90zwKRJk0w/+/n5sW/fPtasWWMKsqdOncr777/PsGHDTOWqV68O3H9mv1y5cqxYscL0RU14eDjt27d/aPvlBwnSRYGy1FnStmxbmvvdn/Ie9kcYv8X9xjv/fYeanjXxsPXAxcoFV2tXXK1d8bTz5CW7l3Cxcimw0XZDiRJ4jP+AmAkfYhkQgEvvXjg0bYrG4tEJgzRaLd5fLCD1t99I+e3I/X+PHMGYkICFtzcvzZ2DRp/9v5xd3Tr4fL2K6x9PJ/XIEW4uXMSdNWtxHTQQffHiZMbEkhETQ0ZsDColFZ2TI1pHR3ROTugcHEEpjPdSUffSMN5L5d7xE6T8JVmHxmBApaVx/ZNZeH8276H1v/7JJ6jUVKyDAnF4s4Vpu+u775Ly2xFSfvmFK0OHUTJsCRYeHo/fsEII8YySkXQhwFpvzYHOBwrt2vnhweiywWAw/Zzb50ul1GN/Bu3Tpw9z5sxh5syZfPzxxw8t5+bmxsiRI/noo4/o2LHjY10jJ5UrVzYL0AGOHDnCxIkTiYyM5NatWxj/TI4cFRVFhQo5D5DZ2NiYAnQAT09PsynyOQkLC6Njx47o//ys26lTJ0aNGsXp06cpV64ckZGReHt7mwL0v4uMjKR9HpczfpSgoKBs2xYtWsSSJUu4fPkyqamppKenU61aNQDi4uK4du0ajRo1eug5+/bty5dffsno0aOJi4tj06ZN7Nix44nr+igSpItCYaW34l9V/0WrMq2Ye3gumy9u5peYh2eBtNJZ4WXnRRmnMjQq2Yj6Jepja/HPR5pz49SuHfZNmqC1tX2sN2aNVotNUBA2f75BKKOR9MuX0bu6PnQEHsC6cmV8vl5F0rZtxM2eTcblKK5PmfrPb0Crxb5RQ5y7dEXn7MTFNm1J2rqVu/v3Y1urVrbid385QNKWH0GrxWPCBLN71uh0vPTpLC62aUv6+fNcaNkKj48+wvEvgbwQQjzPJEgX4n4wm5cp54WpTJkyaDQaTpw4Qes/Hy/8q1OnTuHm5oaTk5MpWDx58qQpYMuJv78/CQkJZtOdc6PX65k6dSo9e/Y0Pdv8MCNGjCA0NNRsBPmfsrU1/2x89+5dgoODCQ4OZuXKlbi5uREVFUWTJk0emVjO4m8DUxqN5pF5AW7dusX3339PRkYGCxcuNG3Pyspi6dKlzJw5M9vjB3+X236tVputDjklhvt7G6xZs4b33nuP2bNnU6tWLezt7Zk1axYHDhzI03UBunfvztixY9m/fz/79+/H19eX1157LdfjnkShB+mhoaHMmjWLmJgYKlasyLx58x5502lpaUyePJmVK1cSGxuLt7c348ePzzatQTwbPGw9mFlvJr0q9eLkzZPcvHeT+NR44lPjuZFyg6vJV4lLieNe1j0uJFzgQsIFIi5HYNAaqPNSHd7weYNA90A8bD3QavI3D6IuH6awaLRaLP388lZWo8EhOBj7Bg24vXoNd9auRWNlhYWHBxaenug9PdDa2mJMTCTrzh2y7iSQlZAAOi1aSys01lZoLa3QFSuGY4vmZlPbnTt14vaqVcROm0ap774zmxWQefMmsX9OA3J+pyNWOUz70ru6UnL5v7k2ajT3/viDayNHkvzTDjw++gjd357F+jtjWhoanS7HWQRCCPEsyMi6/0FQEscJUbS5uLjwxhtvEBoaynvvvWcWgMXGxrJq1Sreffdd4H7SsQoVKjB79mw6duyY7ZnwO3fu4OTkxNtvv83YsWP55JNPmDt3brZrPij3d+3bt2fWrFlmU61zYmdnx4cffsjEiRN56623/sFdP9ypU6eIj49nxowZpszvf19aLj+sWrUKb29vvv/+e7PtO3bsYPr06UybNo0qVapw5coVzpw5k+NoepUqVdixY8dD28vNzc0seV1iYiIXL17MtW4///wztWvXZtCgQaZt58+fN/1sb2+Pr68vO3bs4PXXX8/xHC4uLrRu3Zrw8HD2799vmsL/NBXqp+bVq1czfPhwQkNDqVOnDosXL6ZZs2acOHGCkiVL5nhMhw4duH79OmFhYZQpU4a4uDgyMwtuSS/xdAQUCyCgWM7PBGVkZRBzN4YryVc4FHuIbZe3cSnxEjujd7IzeidwfxpUKcdSlHYqjb+zPzU9a+Lv7J/vgXtB0BgMFOvWlWLduubbOd2GDiFx82bSz53n9jffUKx7dwAyrl4lqk9f0i9dQufmitvQoQ89h6WfH77ffE38osXEL1pE4uYtpBw6jNcnn2D7as6JZO6dOkVU7z6g1eLStw/O77yD1soq3+5LCCEKgoykC/HsWLBgAbVr16ZJkyZMnToVPz8/jh8/zqhRo/D39zdlY9doNISHh9O4cWPq1avHBx98QEBAAMnJyfzwww9ERESwe/duSpQowdy5cxk8eDCJiYl0794dX19frly5wvLly7Gzs8u2DNsDM2bMMD2j/Sj9+/dn7ty5fPPNN9SsmX/J+UqWLInBYODzzz9nwIAB/PHHH0yZMiXfzv9AWFgYb7/9NpUqVTLb7uPjw5gxY9i0aROtWrWiXr16tGvXjjlz5lCmTBlOnTqFRqOhadOmjBs3jsqVKzNo0CAGDBiAwWBg586dtG/fHldXVxo2bMiyZct46623cHZ25sMPPzQlnXuUMmXKsHz5crZu3Yqfnx8rVqzg119/xe8vg2gTJ05kwIABFC9e3JTcbu/evQwZMsRUpm/fvrz55ptkZWXRo0eP/Gu8hyjUCGbOnDn06dOHvn37Ur58eebNm0eJEiXMpkn81Y8//sju3bvZvHkzjRs3xtfXlxo1alC7du0CrrkoSBY6C0o6lKS2V22GvjKUja03sq7lOv5V5V+Ucy6HXqsnNTOV4zePs/H8Rj499Cntf2hPwzUNGffzOH44/wPxqfGFfRuFSufoiNt7wwG48fkCMm/eJO38eS516Ur6pUtYeHnhu2JFrqPiGgsL3IYMxvebrzH4+pIZF0dUv34kbo3IVjbt4kWi+vQl69YtsuLjiZsxk/NvBHNrxUpZd10I8UxJN/65BJsE6UIUeWXLluXXX3+lVKlSdOjQAR8fH5o1a4a/vz979+41S/ZVo0YNDh06ROnSpenXrx/ly5enZcuWHD9+3Gy5r0GDBhEREcHVq1dp06YNAQEB9O3bFwcHh0dmcG/YsCENGzbMdUDRwsKCKVOm5Lps2+Nyc3Nj2bJlrF27lgoVKjBjxgw+/fTTfL3G4cOHOXr0KO3atcu2z97enuDgYNOa6evWraN69ep06tSJChUqMHr0aFP2dn9/fyIiIjh69Cg1atSgVq1abNiwwfSM+7hx46hXrx5vvvkmzZs3p3Xr1mbPzT/MgAEDaNu2LR07dqRmzZrcvHnTbFQdoEePHsybN4/Q0FAqVqzIm2++aUos+EDjxo3x9PSkSZMmeHl5/aO2ehwa9TgLD+aj9PR0bGxsWLt2LW3atDFtHzZsGJGRkezevTvbMYMGDeLMmTMEBQWxYsUKbG1tadmyJVOmTHno8wRpaWmk/SUgSExMpESJEiQkJODg4JD/NyYKXKYxk+ikaC7cucC5O+f4Pf53fo39Nds67OWcy1Hbqza1vGrxivsrL1yWXpWVxaUOHbl3/Di2dety748/yLpzB0Pp0v8oGZwxNZVr4z4g6cf7z7J7TpmCU7u2AGRcu8alLl3JjInBsnx5nDt24OaXX5Fx7RpwP1u++wcf4NAkON/vU4hnUWJiIo6OjtI35aP8bNPxe8az8fxGRgSOoFelpz/NUYjCdu/ePS5evIifnx9Wz8EMuJCQEObMmUNERAS1csjNI0RuUlJS8PLyYunSpbRt2/ah5R71f+dx+qVCm+4eHx9PVlYW7n9bE9rd3Z3Y2Ngcj7lw4QJ79uzBysqK7777jvj4eAYNGsStW7ceupj89OnTc30WRDzb9Fo9fo5++Dn60cjnfmbG9Kx0jt44yt6re9l3bR8nb53k9O3TnL59mvDj4VjrrelVqRd9KvXJ88hIpjHTdL1nkUanw33CeC536szdPXsAsKpcmRJfLkbv7PzY59NaW/PS7E+JsbMl4T/riBk/HuPdZByaNyeqV28yY2Iw+PlRcslX6F1ccGrbljvr1xO/cBGZ169zddgwklu1xH38eHQSlAghirD0LBlJF+JZNmnSJHx9fTlw4AA1a9bMlzXJxYvBaDQSGxvL7NmzcXR0pGXLlgVy3UL/C/175uxHLXNgNBrRaDSsWrWKGjVq0Lx5c+bMmcOyZctITU3N8Zhx48aRkJBgekVHR+f7PYiix6AzUN2jOsMDh7PmrTXs7ribma/NpFXpVhS3Lk5qZiqhkaG029iOgzEHH3oeozJyKPYQk/ZPov7q+rz69avMOTSHhLSEAryb/GPz8ss4/vntn82rr1IyPPwfBegPaHQ6PKdMoVjPngBc/3g6F9q0If3yZSy8vCgZvhS9i8v9sgYDzu+8Q+mIrbj07w9aLQkbNnKhVWvu7t//xPcmhBBPizyTLsSzr1evXgwfPlwCdPFYoqKieOmll1izZg1Lly41Tb9/2gptSNDV1RWdTpdt1DwuLi7b6PoDnp6evPTSSzg6Opq2lS9fHqUUV65coWzZstmOsbS0xNLyxZrWLLIrZlWM5qWa07xUc5RS/HjpR2YenMmlxEv0iehDy9It6RzQmeSMZG6n3SbhXgLRSdFEXI4g5m6M2bnCj4ez9sxaelXqRdfyXYv8ciR/5zllMk5vt8O6cuVc137PC41GQ/Exo9E62BM//3OybsSjc3OlZPjSHKfQay0tKT7iPewaNODa2LFkREUR1as3zp074Tp4MPpixZ64TkIIkZ8ejKS/aI9JCSHEi87X1/eRy889LYUWpBsMBgIDA9m2bZvZM+nbtm2jVatWOR5Tp04d1q5dS3Jysinpw5kzZ9BqtXh7exdIvcWzT6PR0MyvGXVeqsP83+az5vQaNp7fyMbzG3Msb2dhxxs+b9CiVAvSstL47LfPOHP7DJ8f+ZxVJ1dR06MmjpaOppe3nTf1vOuh0+aecbIwaHQ6bF55JX/PqdHgNmgQejc3kiK2UXzUSAw+Po88xuaVlyn13Xquf/opd775lttff0PC9xso1rMnxXr1fOS68kIIUZBMieNkCTYhhBAFoNASx8H9Jdi6devGokWLqFWrFl9++SVfffUVx48fx8fHh3HjxnH16lWWL18OQHJyMuXLl+fVV19l0qRJxMfH07dvX+rXr89XX32Vp2tKch7xd7/f+J1PD31KVGIUTpZOOFk54WTphLOVM7U8a1G/RH2z0ROjMvLjxR9ZELmA6KScH58IdA/k47of42WXPfujUoo7aXdwtvrn08yfN3f37+f6rFmknTgJ3M9G79K/H86dO6N9SFJIIZ4n0jflv/xs0y6bu/D7jd/57PXPaFiyYT7VUIii63lLHCdEQXnmE8cBdOzYkZs3bzJ58mRiYmKoVKkSmzdvxufPEbiYmBiioqJM5e3s7Ni2bRtDhgwhKCgIFxcXOnTowNSpUwvrFsRzoIpbFZY3W57n8lqNlualmvOG7xvsjt5NzN0YEtISuJN2h4S0BP535X8cvn6YdhvbMf7V8bTwa4FGoyE1M5Ufzv/A1ye/5nzCeVqVbsUHNT945qbLPw22tWrh95//kBSxjRvz55N+4QJxsz4lYeMP+H3/3UPzVAghREHIyMoA5Jl0IYQQBaNQR9ILg4xWiKctOjGacXvGcfTGUQCa+TbDw86DdWfWkZieaFbW18GXWfVnEVAsoDCqWiSpzEwSfvgv8Z9/jnO3brj06lnYVRLiqZO+Kf/lZ5u2+r4VFxIusLTJUqp7VM+nGgpRdMlIuhD/TH6NpEt6QyHyWQmHEixruox3q72LTqNjy6UthP8RTmJ6Ii/ZvcSooFGENgqluE1xLiVeovOmzqw6ucosKUWGMYOEtASMyliId1I4NHo9Tm1aU+rHLTh36VzY1RFC5JPQ0FDTh5bAwEB+/vnnPB23d+9e9Ho91apVe7oVfARZgk0IIURBejYXfBaiiNNr9QyoOoA6XnWYdmAa9gZ7OgV0or53fVNCuXVvrePDfR+yK3oXMw7OYPnx5WQaM0nKSCI1M9V0Hg8bD7zsvPC09cTX0Zc6XnUIKBbwj6aAJ6QlEPZHGDujdtLevz1dyncpsgnutAb5MCzE82L16tUMHz6c0NBQ6tSpw+LFi2nWrBknTpygZMmSDz0uISGB7t2706hRI65fv16ANTZnCtIlcZwQQogCINPdhShESim+OfUNnx76lAxjRp6Pc7N24zXv16jjVQeAy4mXTa+7mXep4VGD+t71CXIPwkJnQUpGCitOrGDZ8WUkZySbzlPZtTITa0/E39k/3+9NCJF3z3vfVLNmTV555RUWLlxo2la+fHlat27N9OnTH3rcO++8Q9myZdHpdHz//fdERkbm+Zr52aZ1v61LQloCG1ptoJRTqSc6lxDPApnu/vzatWsXr7/+Ordv38bJyYlly5YxfPhw7ty589BjJk6c+NjvwS8qme4uxHNAo9HQuXxntrTdQlhwGKvfXM3mtpvZ884eDnU9xNZ2W1nWdBkf1/2YIS8PoUGJBljrrbmReoP1Z9fz/u73eX/3+8w/Mp8N5zcQeSOSs7fPsurkKvpv60+91fUY9tMwmq1vxoLIBSRnJOPv7M+gqoOws7DjWPwxOv7QkQVHFphGioQQIj+lp6dz+PBhgoODzbYHBwezb9++hx4XHh7O+fPnCQkJydN10tLSSExMNHvllwfvjxY6i3w7pxDi6YmOjqZPnz54eXlhMBjw8fFh2LBh3Lx5M1vZc+fO0atXL7y9vbG0tMTPz49OnTpx6NAhs3I7d+6kefPmuLi4YGNjQ4UKFXj//fe5evUqcD/41Wg0VKpUiaysLLNjHwTDD/j6+qLRaPjll1/Myg0fPpwGDRrkeE+HDx9Go9GwZ8+eHPc3adKEli1b5tY02XTs2JEzZ8489nEPExwcjE6ny3Zv4vHIdHchigB3W3fcbd2zbfey88LLzotA90DTtvSsdA5dP8TPV37mYOxBrPRW+Dr4UtK+JD6OPug0OvZc3cPu6N3cvHeTn6J/AqCkfUnerfYuTf2aotVoaeffjmm/TOOn6J9Y/PtiIi5HML7meGp61iyw+xZCPP/i4+PJysrC3d38Pc7d3Z3Y2Ngcjzl79ixjx47l559/Rq/P20eV6dOnM2nSpCeub04eBOl/XY5TCFE0XbhwgVq1auHv788333yDn58fx48fZ9SoUWzZsoVffvmFYsWKAXDo0CEaNWpEpUqVWLx4MQEBASQlJbFhwwbef/99du/eDcDixYsZNGgQPXr0YN26dfj6+hIVFcXy5cuZPXs2c+bMMV3//PnzLF++nF69ej2ynlZWVowZM8Z0jdwEBgZStWpVwsPDqVu3rtm+6Ohotm/fzvr16x+nqQCwtrbGOp+Wu42KimL//v0MHjyYsLAwXn311Xw57z+VkZGBhcUz+uWqesEkJCQoQCUkJBR2VYR4qrKMWepo3FG1KHKR+v7s9yo9Kz1bGaPRqLZe3Krqf1tfVVpWSVVaVkmN2jVKxSbHFkKNhXhxPc9909WrVxWg9u3bZ7Z96tSpqly5ctnKZ2ZmqqCgILVw4ULTtpCQEFW1atVHXufevXsqISHB9IqOjs6XNs3IyjC9P95Ovf1E5xLiWZGamqpOnDihUlNTTduMRqPKunu3UF5GozHPdW/atKny9vZWKSkpZttjYmKUjY2NGjBggOl+KlasqAIDA1VWVla289y+fVsppVR0dLQyGAxq+PDhOV7vQbmdO3cqQI0aNUqVKFHCrO0cHR1VeHi46XcfHx81bNgwZTAY1KZNm0zbhw0bpurXr//Qe5s/f76ys7NTycnJZtsnT56s3N3dVUZGhlqxYoUKDAxUdnZ2yt3dXXXq1Eldv37dVPZBPR/UOzw8XDk6Opqdb/r06ap48eLKzs5O9e7dW40ZMybX92CllJo4caJ655131MmTJ5W9vX22et6+fVv169dPFS9eXFlaWqqKFSuqH374wbR/z549ql69esra2lo5OTmp4OBgdevWLVObzZ071+x8VatWVSEhIabfAbVw4ULVsmVLZWNjoz766COVmZmpevfurXx9fZWVlZXy9/dX8+bNy1b3sLAwVaFCBWUwGJSHh4d69913lVJK9erVS7Vo0cKsbEZGhnJ3d1dhYWHZzpPT/50HHqevl5F0IZ5TWo2WKm5VqOJW5aFlNBoNwb7BvOr1KguOLGD16dVsubSF3Vd2M7DqQEo4lCD2biwxyTHE3I0hLSsNT1tPvO29ecnuJbzsvMg0ZnIz9SY3793k1r1b3L53m5TMFFIyUribcZeUzBQqulRkQNUB2BvsC7AFhBBFgaurKzqdLtuoeVxcXLbRdYCkpCQOHTrEkSNHGDx4MABGoxGlFHq9noiICBo2bJjtOEtLSywt83+k+6+PAkl2d/EiU6mpnH4lMPeCT0G53w6jsbHJtdytW7fYunUr06ZNyzY67OHhQZcuXVi9ejWhoaFERkZy/Phxvv76a7Ta7E8AOzk5AbB27VrS09MZPXp0jtd8UO6B4cOHs3LlShYsWMDIkSMfWldfX18GDBjAuHHjaNq0aY51+LsuXbowatQo1q5dS8+ePYH7+Y2WLVtGjx490Ov1pKenM2XKFMqVK0dcXBzvvfcePXv2ZPPmzbmeH2DNmjWEhITwxRdf8Nprr7FixQrmz59PqVKPzsehlCI8PJwvvviCgIAA/P39WbNmjWlGgdFopFmzZiQlJbFy5UpKly7NiRMn0OnuJzCOjIykUaNG9O7dm/nz56PX69m5c2e2RwdyExISwvTp05k7dy46nQ6j0Yi3tzdr1qzB1dWVffv20b9/fzw9PenQoQMACxcuZMSIEcyYMYNmzZqRkJDA3r17Aejbty/16tUjJiYGT09PADZv3kxycrLp+KdBgnQhBA4GBz6o+QGty7Rm2oFp/H7jd2Yfnp1v5z98/TA/XvyRCa9O4PWSr+fbeYUQRZ/BYCAwMJBt27bRpk0b0/Zt27bRqlWrbOUdHBw4duyY2bbQ0FB++ukn/vOf/+Dn5/fU6/xXf03qKUG6EEXb2bNnUUpRvnz5HPeXL1+e27dvc+PGDc6ePQtAQEBArud0cHAwBWi5sbGxISQkhA8++IB+/frh6Oj40LITJkwgPDycVatW0a1bt1zPXaxYMVq3bk14eLgpSN+1axcXLlygd+/eAKZ/AUqVKsX8+fOpUaMGycnJ2NnZ5XqNefPm0bt3b/r27QvA1KlT2b59O/fu3Xvkcdu3byclJYUmTZoA0LVrV8LCwkxB+vbt2zl48CAnT57E39/fVL8HPvnkE4KCgggNDTVtq1ixYq71/bvOnTubtQFg9iiUn58f+/btY82aNaYge+rUqbz//vsMGzbMVK569eoA1K5dm3LlyrFixQrTFzXh4eG0b98+T+35T0mQLoQwqeBSgRXNVrDh3AZWnlyJQWvA084TT9v7L4POwLXka1xNvmp6WeoscbFyoZh1MVysXHC2csbOwg4bCxts9Pe/9V5ybAlRSVEM3TmUJr5NGFtjLJnGTA7GHuRgzEF+jf2Vm/duYmdhh73BHlsLW+wMdgS6BzKw6sBCbhUhxJMaMWIE3bp1IygoiFq1avHll18SFRXFgAEDABg3bhxXr15l+fLlaLVaKlWqZHZ88eLFsbKyyra9IKRlpQGg0+jQa+Vjk3hxaaytKffb4UK7dn5Qfy5qZTAYTD/ntqStUuqxl73t06cPc+bMYebMmXz88ccPLefm5sbIkSP56KOP6NixY57PHRwczLlz5yhTpgxLly6lTp06lCtXDoAjR44wceJEIiMjuXXrFkajEbj/vHiFChVyPf/JkydN780P1KpVi507dz7yuLCwMDp27GjKI9KpUydGjRrF6dOnKVeuHJGRkXh7e5sC9L+LjIykffv2udYvN0FBQdm2LVq0iCVLlnD58mVSU1NJT0+nWrVqwP1ZXdeuXaNRo0YPPWffvn358ssvGT16NHFxcWzatIkdO3Y8cV0fRXobIYQZrUZLm7JtaFO2Te6F86iZXzNCj4ay/Phytl7ayo6oHWQaM7OVS8tK4+a9/8+8am8h0+OFeB507NiRmzdvMnnyZGJiYqhUqRKbN2/Gx8cHgJiYGKKiogq5ljl7EKTLKLp40Wk0mjxNOS9MZcqUQaPRcOLECVq3bp1t/6lTp3Bzc8PJyckULJ48edIUsOXE39+fhIQEs+nOudHr9UydOpWePXuaHtt5mBEjRhAaGmo2gvwojRs3xsfHh2XLljF69GjWr1/PggULALh79y7BwcEEBwezcuVK3NzciIqKokmTJqSnP71VfG7dusX3339PRkaG2VKbWVlZLF26lJkzZ+aanC63/Vqt1vTFygMZGdmXL7a1tTX7fc2aNbz33nvMnj2bWrVqYW9vz6xZszhw4ECergvQvXt3xo4dy/79+9m/fz++vr689tpruR73JGQJNiHEU2elt2JE4Ai+bvE1AcUCyDRmotVoqexamd6VerO48WK2tN3Cf976D+FNwlnQcAHTX5vOOwHvFHbVhRD5ZNCgQVy6dIm0tDQOHz5MvXr1TPuWLVvGrl27Hnrsg1GhwpCRdf9DoATpQhR9Li4uvPHGG4SGhpKammq2LzY2llWrVpmmiVerVo0KFSowe/Zs02jzXz1YN/ztt9/GYDDwySef5HjNh60v3r59eypWrJjrqhN2dnZ8+OGHTJs2LU9LR2o0Gnr16sW///1v0/P0D6Ztnzp1ivj4eGbMmMFrr71GQEAAcXFxuZ7zr8qXL59t+bTcllNbtWoV3t7eHD16lMjISNNr3rx5/Pvf/yYzM5MqVapw5cqVhy73VqVKlUeOTru5uRETE2P6PTExkYsXL+Z6Pz///DO1a9dm0KBBvPzyy5QpU4bz58+b9tvb2+Pr6/vIa7u4uJgeMwgPD881c3++yDW13HPmec6gK8SzICMrQ526eUolpSUVdlWEKDKkb8p/+dWmd9PvqgPXDqgD1w7kU82EKPoelaG6qDtz5oxydXVVr732mtq9e7eKiopSW7ZsUZUqVVLVqlVTSUn///njwIEDyt7eXtWpU0dt2rRJnT9/Xh09elRNnTpV1atXz1Tuiy++UBqNRvXu3Vvt2rVLXbp0Se3Zs0f1799fjRgxQimVPWu6Ukrt2LFD6fV6pdfrs2V3/2um8vT0dFW6dGllZWX1yOzuD1y+fFlptVrl7Oys+vbta9oeFxenDAaDGjVqlDp//rzasGGD8vf3V4A6cuRIjvX8e3b3b7/9VllaWqqwsDB1+vRp9dFHHyl7e/tHZnevWrWqGjNmTLbtiYmJytLSUn3//fdKKaUaNGigKlWqpCIiItSFCxfU5s2b1ZYtW5RSSp0+fVoZDAY1cOBAdfToUXXy5EkVGhqqbty4oZRSauzYscrDw0P973//U8eOHVOtW7dWdnZ22bK7f/fdd2Z1mDdvnnJwcFA//vijOn36tJowYYJycHAwu59ly5YpKysr9dlnn6kzZ86ow4cPq/nz55udJyIiQhkMBqXT6dTVq1cf2hb5ld1dRtKFEAVKr9VTrlg57AxPL9mGEELkFxsLG2p41qCGZ43CrooQIg/Kli3Lr7/+SqlSpejQoQM+Pj40a9YMf39/9u7da5bsq0aNGhw6dIjSpUvTr18/ypcvT8uWLTl+/Djz5s0zlRs0aBARERFcvXqVNm3aEBAQQN++fXFwcHhkBveGDRvSsGFDMjOzP+L3VxYWFkyZMiXX5GwPlCxZksaNG3P79m2zJGlubm4sW7aMtWvXUqFCBWbMmMGnn36ap3M+0LFjRz766CPGjBlDYGAgly9fZuDAh+cHOnz4MEePHqVdu3bZ9tnb2xMcHExYWBgA69ato3r16nTq1IkKFSowevRoU/Z2f39/IiIiOHr0KDVq1KBWrVps2LDB9Iz7uHHjqFevHm+++SbNmzendevWlC5dOtf7GTBgAG3btqVjx47UrFmTmzdvMmjQILMyPXr0YN68eYSGhlKxYkXefPNNU2LBBxo3boynpydNmjTBy8sr1+s+Kc2f3zq8MBITE3F0dCQhIQEHB4fCro4QQgghfdNTIG0qxD937949Ll68iJ+fH1ZWVoVdnScWEhLCnDlziIiIoFatWoVdHfEMSklJwcvLi6VLl9K2bduHlnvU/53H6ZckcZwQQgghhBDiuTVp0iR8fX05cOAANWvWzNOa5ELA/fXdY2NjmT17No6OjrRs2bJAritBuhBCCCGEEOK5ViDJvsRzJyoqCj8/P7y9vVm2bJlp+v3TJkG6EEIIIYQQQgjxN76+vtmWfisIMtdDCCGEEEIIIYQoIiRIF0IIIYQQQmTzguWXFuKJ5df/GQnShRBCCCGEECYWFhbA/YzWQoi8S09PB0Cn0z3ReeSZdCGEEEIIIYSJTqfDycmJuLg4AGxsbNBoNIVcKyGKNqPRyI0bN7CxsXniBHMSpAshhBBCCCHMeHh4AJgCdSFE7rRaLSVLlnziL7UkSBdCCCGEEEKY0Wg0eHp6Urx4cTIyMgq7OkI8EwwGA1rtkz9RLkG6EEIIIYQQIkc6ne6Jn68VQjweSRwnhBBCCCGEEEIUERKkCyGEEEIIIYQQRYQE6UIIIYQQQgghRBHxwj2T/mCB+cTExEKuiRBCCHHfgz7pQR8lnpz090IIIYqSx+nrX7ggPSkpCYASJUoUck2EEEIIc0lJSTg6OhZ2NZ4L0t8LIYQoivLS12vUC/a1vdFo5Nq1a9jb2z/2+nWJiYmUKFGC6OhoHBwcnlINxQPS3gVH2rpgSXsXnGelrZVSJCUl4eXllS9Lt4h/3t8/K38zzwtp74IjbV2wpL0LzrPS1o/T179wI+larRZvb+8nOoeDg0OR/gN43kh7Fxxp64Il7V1wnoW2lhH0/PWk/f2z8DfzPJH2LjjS1gVL2rvgPAttnde+Xr6uF0IIIYQQQgghiggJ0oUQQgghhBBCiCJCgvTHYGlpSUhICJaWloVdlReCtHfBkbYuWNLeBUfaWjwu+ZspWNLeBUfaumBJexec57GtX7jEcUIIIYQQQgghRFElI+lCCCGEEEIIIUQRIUG6EEIIIYQQQghRREiQLoQQQgghhBBCFBESpAshhBBCCCGEEEWEBOl5FBoaip+fH1ZWVgQGBvLzzz8XdpWeC9OnT6d69erY29tTvHhxWrduzenTp83KKKWYOHEiXl5eWFtb06BBA44fP15INX5+TJ8+HY1Gw/Dhw03bpK3z19WrV+natSsuLi7Y2NhQrVo1Dh8+bNov7Z0/MjMzmTBhAn5+flhbW1OqVCkmT56M0Wg0lZG2Fnkl/X3+k76+8Ehf//RJX18wXri+Xolcffvtt8rCwkJ99dVX6sSJE2rYsGHK1tZWXb58ubCr9sxr0qSJCg8PV3/88YeKjIxULVq0UCVLllTJycmmMjNmzFD29vZq3bp16tixY6pjx47K09NTJSYmFmLNn20HDx5Uvr6+qkqVKmrYsGGm7dLW+efWrVvKx8dH9ezZUx04cEBdvHhRbd++XZ07d85URto7f0ydOlW5uLio//73v+rixYtq7dq1ys7OTs2bN89URtpa5IX090+H9PWFQ/r6p0/6+oLzovX1EqTnQY0aNdSAAQPMtgUEBKixY8cWUo2eX3FxcQpQu3fvVkopZTQalYeHh5oxY4apzL1795Sjo6NatGhRYVXzmZaUlKTKli2rtm3bpurXr2/quKWt89eYMWNU3bp1H7pf2jv/tGjRQvXu3dtsW9u2bVXXrl2VUtLWIu+kvy8Y0tc/fdLXFwzp6wvOi9bXy3T3XKSnp3P48GGCg4PNtgcHB7Nv375CqtXzKyEhAYBixYoBcPHiRWJjY83a39LSkvr160v7/0PvvvsuLVq0oHHjxmbbpa3z18aNGwkKCqJ9+/YUL16cl19+ma+++sq0X9o7/9StW5cdO3Zw5swZAI4ePcqePXto3rw5IG0t8kb6+4Ijff3TJ319wZC+vuC8aH29vrArUNTFx8eTlZWFu7u72XZ3d3diY2MLqVbPJ6UUI0aMoG7dulSqVAnA1MY5tf/ly5cLvI7Pum+//ZbffvuNX3/9Nds+aev8deHCBRYuXMiIESP44IMPOHjwIEOHDsXS0pLu3btLe+ejMWPGkJCQQEBAADqdjqysLKZNm0anTp0A+dsWeSP9fcGQvv7pk76+4EhfX3BetL5egvQ80mg0Zr8rpbJtE09m8ODB/P777+zZsyfbPmn/JxcdHc2wYcOIiIjAysrqoeWkrfOH0WgkKCiIjz/+GICXX36Z48ePs3DhQrp3724qJ+395FavXs3KlSv5+uuvqVixIpGRkQwfPhwvLy969OhhKidtLfJC/k6eLunrny7p6wuW9PUF50Xr62W6ey5cXV3R6XTZvkWPi4vL9k2N+OeGDBnCxo0b2blzJ97e3qbtHh4eANL++eDw4cPExcURGBiIXq9Hr9eze/du5s+fj16vN7WntHX+8PT0pEKFCmbbypcvT1RUFCB/2/lp1KhRjB07lnfeeYfKlSvTrVs33nvvPaZPnw5IW4u8kf7+6ZO+/umTvr5gSV9fcF60vl6C9FwYDAYCAwPZtm2b2fZt27ZRu3btQqrV80MpxeDBg1m/fj0//fQTfn5+Zvv9/Pzw8PAwa//09HR2794t7f+YGjVqxLFjx4iMjDS9goKC6NKlC5GRkZQqVUraOh/VqVMn2xJDZ86cwcfHB5C/7fyUkpKCVmvenel0OtOyLNLWIi+kv396pK8vONLXFyzp6wvOC9fXF0a2umfNgyVZwsLC1IkTJ9Tw4cOVra2tunTpUmFX7Zk3cOBA5ejoqHbt2qViYmJMr5SUFFOZGTNmKEdHR7V+/Xp17Ngx1alTp2d2OYWi5q8ZX5WSts5PBw8eVHq9Xk2bNk2dPXtWrVq1StnY2KiVK1eaykh7548ePXqol156ybQsy/r165Wrq6saPXq0qYy0tcgL6e+fDunrC5f09U+P9PUF50Xr6yVIz6MvvvhC+fj4KIPBoF555RXTsiHiyQA5vsLDw01ljEajCgkJUR4eHsrS0lLVq1dPHTt2rPAq/Rz5e8ctbZ2/fvjhB1WpUiVlaWmpAgIC1Jdffmm2X9o7fyQmJqphw4apkiVLKisrK1WqVCk1fvx4lZaWZiojbS3ySvr7/Cd9feGSvv7pkr6+YLxofb1GKaUKZwxfCCGEEEIIIYQQfyXPpAshhBBCCCGEEEWEBOlCCCGEEEIIIUQRIUG6EEIIIYQQQghRREiQLoQQQgghhBBCFBESpAshhBBCCCGEEEWEBOlCCCGEEEIIIUQRIUG6EEIIIYQQQghRREiQLoQQQgghhBBCFBESpAshCpxGo+H7778v7GoIIYQQ4imRvl6If06CdCFeMD179kSj0WR7NW3atLCrJoQQQoh8IH29EM82fWFXQAhR8Jo2bUp4eLjZNktLy0KqjRBCCCHym/T1Qjy7ZCRdiBeQpaUlHh4eZi9nZ2fg/vS0hQsX0qxZM6ytrfHz82Pt2rVmxx87doyGDRtibW2Ni4sL/fv3Jzk52azM0qVLqVixIpaWlnh6ejJ48GCz/fHx8bRp0wYbGxvKli3Lxo0bn+5NCyGEEC8Q6euFeHZJkC6EyObDDz+kXbt2HD16lK5du9KpUydOnjwJQEpKCk2bNsXZ2Zlff/2VtWvXsn37drOOeeHChbz77rv079+fY8eOsXHjRsqUKWN2jUmTJtGhQwd+//13mjdvTpcuXbh161aB3qcQQgjxopK+XogiTAkhXig9evRQOp1O2dramr0mT56slFIKUAMGDDA7pmbNmmrgwIFKKaW+/PJL5ezsrJKTk037N23apLRarYqNjVVKKeXl5aXGjx//0DoAasKECabfk5OTlUajUVu2bMm3+xRCCCFeVNLXC/Fsk2fShXgBvf766yxcuNBsW7FixUw/16pVy2xfrVq1iIyMBODkyZNUrVoVW1tb0/46depgNBo5ffo0Go2Ga9eu0ahRo0fWoUqVKqafbW1tsbe3Jy4u7p/ekhBCCCH+Qvp6IZ5dEqQL8QKytbXNNiUtNxqNBgCllOnnnMpYW1vn6XwWFhbZjjUajY9VJyGEEELkTPp6IZ5d8ky6ECKbX375JdvvAQEBAFSoUIHIyEju3r1r2r937160Wi3+/v7Y29vj6+vLjh07CrTOQgghhMg76euFKLpkJF2IF1BaWhqxsbFm2/R6Pa6urgCsXbuWoKAg6taty6pVqzh48CBhYWEAdOnShZCQEHr06MHEiRO5ceMGQ4YMoVu3bri7uwMwceJEBgwYQPHixWnWrBlJSUns3buXIUOGFOyNCiGEEC8o6euFeHZJkC7EC+jHH3/E09PTbFu5cuU4deoUcD8b67fffsugQYPw8PBg1apVVKhQAQAbGxu2bt3KsGHDqF69OjY2NrRr1445c+aYztWjRw/u3bvH3LlzGTlyJK6urrz99tsFd4NCCCHEC076eiGeXRqllCrsSgghig6NRsN3331H69atC7sqQgghhHgKpK8XomiTZ9KFEEIIIYQQQogiQoJ0IYQQQgghhBCiiJDp7kIIIYQQQgghRBEhI+lCCCGEEEIIIUQRIUG6EEIIIYQQQghRREiQLoQQQgghhBBCFBESpAshhBBCCCGEEEWEBOlCCCGEEEIIIUQRIUG6EEIIIYQQQghRREiQLoQQQgghhBBCFBESpAshhBBCCCGEEEXE/wFkS2tNGfRJxQAAAABJRU5ErkJggg=="
+      "image/png": "iVBORw0KGgoAAAANSUhEUgAAA+kAAAHUCAYAAABGRmklAAAAOnRFWHRTb2Z0d2FyZQBNYXRwbG90bGliIHZlcnNpb24zLjEwLjAsIGh0dHBzOi8vbWF0cGxvdGxpYi5vcmcvlHJYcgAAAAlwSFlzAAAPYQAAD2EBqD+naQABAABJREFUeJzs3Xd4FNUawOHf9k3vDUih995BQOSKoKCoKFaavSM2uCp2ufaOigWsiKggojQFpIOAoXdII430vtky94+TQkgCAQIJ8L3Ps0+ys2dmzixhZ79TvqPTNE1DCCGEEEIIIYQQdU5f1xUQQgghhBBCCCGEIkG6EEIIIYQQQghRT0iQLoQQQgghhBBC1BMSpAshhBBCCCGEEPWEBOlCCCGEEEIIIUQ9IUG6EEIIIYQQQghRT0iQLoQQQgghhBBC1BMSpAshhBBCCCGEEPWEBOlCCCGEEEIIIUQ9IUG6uKjpdLoaPVasWHFG53n++efR6XSnte+KFStqpQ5ncu6ffvrpnJ/7dGzbto1x48bRuHFjrFYrnp6edOnShddff52MjIy6rp4QQogakHtzzc2fPx+dTkdAQAA2m61O63I+ysnJ4ZVXXqFbt254e3tjsViIiopi/PjxbNmypa6rJy5ixrqugBB1ad26dRWev/TSSyxfvpxly5ZV2N6mTZszOs+dd97JkCFDTmvfLl26sG7dujOuw4Xus88+4/7776dly5Y88cQTtGnTBrvdzqZNm/jkk09Yt24dc+fOretqCiGEOAm5N9fcF198AUBGRgbz5s1j1KhRdVqf88nBgwcZPHgwqamp3Hvvvbzwwgt4enoSExPDjz/+SNeuXcnKysLHx6euqyouQjpN07S6roQQ9cXYsWP56aefyMvLO2G5goIC3N3dz1Gt6s6KFSsYOHAgc+bMYeTIkXVdnWqtW7eOfv36cfnllzNv3jwsFkuF14uLi1m0aBFXX331GZ+rsLAQq9V62r0vQgghTo3cm6uWnJxMeHg4/fv3Z+3atfTr148lS5bUdbWqVN/+bZxOJ507dyY2NpY1a9bQrl27SmUWLlzIgAEDzrjemqZRVFSEm5vbGR1HXFxkuLsQJ3HppZfSrl07Vq5cSZ8+fXB3d2f8+PEAzJ49m8GDBxMWFoabmxutW7dm0qRJ5OfnVzhGVUPqoqKiGDZsGIsWLaJLly64ubnRqlUrvvzyywrlqhpSN3bsWDw9PTlw4ABXXnklnp6ehIeH89hjj1Ua7paQkMDIkSPx8vLC19eXW2+9lX/++QedTsfMmTNr5T3asWMH11xzDX5+flitVjp16sRXX31VoYzL5eLll1+mZcuWuLm54evrS4cOHXjvvffKyhw9epS7776b8PBwLBYLQUFB9O3blz///POE53/11VfR6XRMnz69UoAOYDabKwToOp2O559/vlK5qKgoxo4dW/Z85syZ6HQ6lixZwvjx4wkKCsLd3Z3Zs2ej0+n466+/Kh3j448/RqfTsW3btrJtmzZt4uqrr8bf3x+r1Urnzp358ccfT3hNQgghqif3Zvjqq69wOBw8+uijXHfddfz111/ExsZWKpeVlcVjjz1GkyZNsFgsBAcHc+WVV7Jnz56yMjabjRdffJHWrVtjtVoJCAhg4MCBrF27FoCYmJhq63b8PbX0fd2yZQsjR47Ez8+Ppk2bAup+eNNNNxEVFYWbmxtRUVHcfPPNVdb7yJEjZd8JzGYzDRo0YOTIkaSkpJCXl4evry/33HNPpf1iYmIwGAy88cYb1b538+bNY/v27UyePLnKAB1g6NChZQH62LFjiYqKqlSmqr8hnU7Hgw8+yCeffELr1q2xWCx8/vnnBAcHc/vtt1c6RlZWFm5ubkycOLFsW05ODo8//jiNGzfGbDbTsGFDJkyYUOlvWFy4ZLi7EDWQlJTEbbfdxpNPPsmrr76KXq/at/bv38+VV17JhAkT8PDwYM+ePbz22mts3Lix0rC8qmzdupXHHnuMSZMmERISwueff84dd9xBs2bN6N+//wn3tdvtXH311dxxxx089thjrFy5kpdeegkfHx+mTJkCQH5+PgMHDiQjI4PXXnuNZs2asWjRolodDrd371769OlDcHAw77//PgEBAXz77beMHTuWlJQUnnzySQBef/11nn/+eZ555hn69++P3W5nz549ZGVllR3r9ttvZ8uWLbzyyiu0aNGCrKwstmzZQnp6erXndzqdLFu2jK5duxIeHl5r13Ws8ePHc9VVV/HNN9+Qn5/PsGHDCA4OZsaMGQwaNKhC2ZkzZ9KlSxc6dOgAwPLlyxkyZAg9e/bkk08+wcfHhx9++IFRo0ZRUFBQoVFACCFEzV3s9+Yvv/ySsLAwhg4dipubG99//z0zZ87kueeeKyuTm5vLJZdcQkxMDE899RQ9e/YkLy+PlStXkpSURKtWrXA4HAwdOpRVq1YxYcIELrvsMhwOB+vXrycuLo4+ffqcUr1KXXfdddx0003ce++9ZcFlTEwMLVu25KabbsLf35+kpCQ+/vhjunfvzq5duwgMDARUgN69e3fsdjv//e9/6dChA+np6SxevJjMzExCQkIYP34806dP5/XXX68wJH3atGmYzeayRpuqlI44GDFixGld28nMmzePVatWMWXKFEJDQwkODubw4cN88sknfPTRR3h7e5eVnTVrFkVFRYwbNw5Qow4GDBhAQkJC2bXv3LmTKVOmsH37dv78808ZzXcx0IQQZcaMGaN5eHhU2DZgwAAN0P76668T7utyuTS73a79/fffGqBt3bq17LXnnntOO/6/W2RkpGa1WrXY2NiybYWFhZq/v792zz33lG1bvny5BmjLly+vUE9A+/HHHysc88orr9RatmxZ9vyjjz7SAG3hwoUVyt1zzz0aoM2YMeOE11R67jlz5lRb5qabbtIsFosWFxdXYfvQoUM1d3d3LSsrS9M0TRs2bJjWqVOnE57P09NTmzBhwgnLHC85OVkDtJtuuqnG+wDac889V2l7ZGSkNmbMmLLnM2bM0ABt9OjRlcpOnDhRc3NzK7s+TdO0Xbt2aYD2wQcflG1r1aqV1rlzZ81ut1fYf9iwYVpYWJjmdDprXG8hhLgYyb25spUrV2qANmnSpLLrbNy4sRYZGam5XK6yci+++KIGaEuXLq32WF9//bUGaJ999lm1ZQ4fPlxt3Y6/p5a+r1OmTDnpdTgcDi0vL0/z8PDQ3nvvvbLt48eP10wmk7Zr165q9z148KCm1+u1d955p2xbYWGhFhAQoI0bN+6E5x0yZIgGaEVFRSeto6apf9vIyMhK26v6GwI0Hx8fLSMjo8L2bdu2aYA2ffr0Ctt79Oihde3atez51KlTNb1er/3zzz8Vyv30008aoP3xxx81qrM4v8lwdyFqwM/Pj8suu6zS9kOHDnHLLbcQGhqKwWDAZDIxYMAAAHbv3n3S43bq1ImIiIiy51arlRYtWlQ57Ot4Op2O4cOHV9jWoUOHCvv+/fffeHl5VUqMc/PNN5/0+DW1bNkyBg0aVKkXe+zYsRQUFJQlAOrRowdbt27l/vvvZ/HixeTk5FQ6Vo8ePZg5cyYvv/wy69evx26311o9z8T1119fadv48eMpLCxk9uzZZdtmzJiBxWLhlltuAeDAgQPs2bOHW2+9FQCHw1H2uPLKK0lKSmLv3r3n5iKEEOICczHfm0sTxpX2Fut0OsaOHUtsbGyFqVgLFy6kRYsW/Oc//6n2WAsXLsRqtZ6w5/l0VHXvzMvL46mnnqJZs2YYjUaMRiOenp7k5+dX+LdZuHAhAwcOpHXr1tUev0mTJgwbNoxp06ahlaTY+v7770lPT+fBBx+s1Ws5VZdddhl+fn4VtrVv356uXbsyY8aMsm27d+9m48aNFd77BQsW0K5dOzp16lThe8MVV1xRL1YUEOeGBOlC1EBYWFilbXl5efTr148NGzbw8ssvs2LFCv755x9++eUXQCUYO5mAgIBK2ywWS432dXd3x2q1Vtq3qKio7Hl6ejohISGV9q1q2+lKT0+v8v1p0KBB2esAkydP5s0332T9+vUMHTqUgIAABg0axKZNm8r2mT17NmPGjOHzzz+nd+/e+Pv7M3r0aJKTk6s9f2BgIO7u7hw+fLjWrul4VV1f27Zt6d69e9nN1ul08u2333LNNdfg7+8PQEpKCgCPP/44JpOpwuP+++8HIC0t7azVWwghLmQX6705NzeXOXPm0KNHD4KCgsjKyiIrK4trr70WnU5XFsCDyvXSqFGjEx7v6NGjNGjQoGy6QG2p6t/nlltu4cMPP+TOO+9k8eLFbNy4kX/++YegoKAK729N6g3wyCOPsH//fpYuXQrARx99RO/evenSpcsJ9ytthDlb3x2qunZQjSrr1q0rywdQ2rh/bANNSkoK27Ztq/S9wcvLC03T5HvDRULmpAtRA1XN/Vm2bBmJiYmsWLGirIUeqDDHuq4FBASwcePGSttPFPSezjmSkpIqbU9MTAQom19mNBqZOHEiEydOJCsriz///JP//ve/XHHFFcTHx+Pu7k5gYCDvvvsu7777LnFxccyfP59JkyaRmprKokWLqjy/wWBg0KBBLFy4kISEhBrd1C0WS5XryVY39726uV/jxo3j/vvvZ/fu3Rw6dIikpKSyOWXHXvvkyZO57rrrqjxGy5YtT1pfIYQQlV2s9+ZZs2ZRUFDAxo0bK/XWAsydO5fMzEz8/PwICgoiISHhhMcLCgpi9erVuFyuagP10oaH4++dJ8oZc/y/T3Z2NgsWLOC5555j0qRJZdttNhsZGRmV6nSyeoPqsW7Xrh0ffvghnp6ebNmyhW+//fak+11xxRVMnz6defPmVahLdaxWa5XfG6oLmKv73nDzzTczceJEZs6cySuvvMI333zDiBEjKvw7BgYG4ubmVilZ4bGviwuf9KQLcZpKP4CPzyb+6aef1kV1qjRgwAByc3NZuHBhhe0//PBDrZ1j0KBBZV+KjvX111/j7u5Or169Ku3j6+vLyJEjeeCBB8jIyCAmJqZSmYiICB588EEuv/xytmzZcsI6TJ48GU3TuOuuuyguLq70ut1u57fffit7HhUVVSH7Oqgvdidb3ud4N998M1arlZkzZzJz5kwaNmzI4MGDy15v2bIlzZs3Z+vWrXTr1q3Kh5eX1ymdUwghRPUuhnvzF198gZeXF3/99RfLly+v8HjjjTew2Wx89913gMpQvm/fvhMmzBs6dChFRUUnzCofEhKC1WqtdO/89ddfa1RnUP82mqZV+rf5/PPPcTqdleq0fPnyGk0Je/jhh/n999+ZPHkyISEh3HDDDSfd55prrqF9+/ZMnTqVHTt2VFlm8eLFFBQUAOp7Q2pqatkIOVDLuy5evPik5zqWn58fI0aM4Ouvv2bBggUkJydXmmYwbNgwDh48SEBAQJXfG6rKMi8uPNKTLsRp6tOnD35+ftx7770899xzmEwmvvvuO7Zu3VrXVSszZswY3nnnHW677TZefvllmjVrxsKFC8tuKjUd2rZ+/foqtw8YMIDnnnuOBQsWMHDgQKZMmYK/vz/fffcdv//+e4WMq8OHD6ddu3Z069aNoKAgYmNjeffdd4mMjKR58+ZkZ2czcOBAbrnlFlq1aoWXlxf//PMPixYtqrYXulTv3r35+OOPuf/+++natSv33Xcfbdu2xW638++//zJ9+nTatWtXNk/w9ttv59lnn2XKlCkMGDCAXbt28eGHH1bIDlsTvr6+XHvttcycOZOsrCwef/zxSu/pp59+ytChQ7niiisYO3YsDRs2JCMjg927d7NlyxbmzJlzSucUQghRvQv93rxjxw42btzIfffdV+V8/L59+/LWW2/xxRdf8OCDDzJhwgRmz57NNddcw6RJk+jRoweFhYX8/fffDBs2jIEDB3LzzTczY8YM7r33Xvbu3cvAgQNxuVxs2LCB1q1bc9NNN6HT6bjtttv48ssvadq0KR07dmTjxo18//33Nb5ub29v+vfvzxtvvEFgYCBRUVH8/ffffPHFF/j6+lYo++KLL7Jw4UL69+/Pf//7X9q3b09WVhaLFi1i4sSJtGrVqqzsbbfdxuTJk1m5ciXPPPMMZrP5pHUxGAzMnTuXwYMH07t3b+677z4GDhyIh4cHsbGx/PTTT/z2229kZmYCMGrUKKZMmcJNN93EE088QVFREe+//36lxoWaGD9+PLNnz+bBBx+kUaNGlfIFTJgwgZ9//pn+/fvz6KOP0qFDB1wuF3FxcSxZsoTHHnuMnj17nvJ5xXmmbvPWCVG/VJdBtm3btlWWX7t2rda7d2/N3d1dCwoK0u68805ty5YtlTKgVpdB9qqrrqp0zAEDBmgDBgwoe15dBtnj61ndeeLi4rTrrrtO8/T01Ly8vLTrr79e++OPPzRA+/XXX6t7Kyqcu7pHaZ22b9+uDR8+XPPx8dHMZrPWsWPHShlg33rrLa1Pnz5aYGCgZjabtYiICO2OO+7QYmJiNE3TtKKiIu3ee+/VOnTooHl7e2tubm5ay5Ytteeee07Lz88/YT1LRUdHa2PGjNEiIiI0s9mseXh4aJ07d9amTJmipaamlpWz2Wzak08+qYWHh2tubm7agAEDtOjo6Gqzux+fYfVYS5YsKXs/9u3bV2WZrVu3ajfeeKMWHBysmUwmLTQ0VLvsssu0Tz75pEbXJYQQFzO5N5ebMGGCBmjR0dHVlpk0aZIGaJs3b9Y0TdMyMzO1Rx55RIuIiNBMJpMWHBysXXXVVdqePXvK9iksLNSmTJmiNW/eXDObzVpAQIB22WWXaWvXri0rk52drd15551aSEiI5uHhoQ0fPlyLiYmpNrv70aNHK9UtISFBu/766zU/Pz/Ny8tLGzJkiLZjx45K919N07T4+Hht/PjxWmhoqGYymbQGDRpoN954o5aSklLpuGPHjtWMRqOWkJBQ7ftSlaysLO2ll17SunTponl6emomk0mLiIjQbrvtNm3NmjUVyv7xxx9ap06dNDc3N61Jkybahx9+WG129wceeKDaczqdTi08PFwDtKeffrrKMnl5edozzzyjtWzZUjObzZqPj4/Wvn177dFHH9WSk5NP6RrF+UmnaSXpEIUQF41XX32VZ555hri4uBrN4RZCCCHE2SX35tNTXFxMVFQUl1xyCT/++GNdV0eIWiHD3YW4wH344YcAtGrVCrvdzrJly3j//fe57bbb5EuAEEIIUQfk3nzmjh49yt69e5kxYwYpKSk1SgAnxPlCgnQhLnDu7u688847xMTEYLPZiIiI4KmnnuKZZ56p66oJIYQQFyW5N5+533//nXHjxhEWFsa0adNOuuyaEOcTGe4uhBBCCCGEEELUE7IEmxBCCCGEEEIIUU9IkC6EEEIIIYQQQtQTEqQLIYQQQgghhBD1xEWXOM7lcpGYmIiXlxc6na6uqyOEEEKgaRq5ubk0aNAAvV7az2uD3O+FEELUJ6dyr7/ogvTExETCw8PruhpCCCFEJfHx8bL8Ui2R+70QQoj6qCb3+osuSPfy8gLUm+Pt7V3HtRFCCCEgJyeH8PDwsnuUOHNyvxdCCFGfnMq9vk6D9JUrV/LGG2+wefNmkpKSmDt3LiNGjDjhPt999x2vv/46+/fvx8fHhyFDhvDmm28SEBBQo3OWDnnz9vaWm7YQQoh6RYZl1x653wshhKiPanKvr9OJb/n5+XTs2JEPP/ywRuVXr17N6NGjueOOO9i5cydz5szhn3/+4c477zzLNRVCCCGEEEIIIc6+Ou1JHzp0KEOHDq1x+fXr1xMVFcXDDz8MQOPGjbnnnnt4/fXXz1YVhRBCCCGEEEKIc+a8SiHbp08fEhIS+OOPP9A0jZSUFH766Seuuuqqavex2Wzk5ORUeAghhBBCCCGEEPXReRekf/fdd4waNQqz2UxoaCi+vr588MEH1e4zdepUfHx8yh6S6VUIIYQQQgghRH11XgXpu3bt4uGHH2bKlCls3ryZRYsWcfjwYe69995q95k8eTLZ2dllj/j4+HNYYyGEEEIIIYQQoubOqyXYpk6dSt++fXniiScA6NChAx4eHvTr14+XX36ZsLCwSvtYLBYsFsu5rqoQQgghhBBCCHHKzque9IKCAvT6ilU2GAwAaJpWF1USQgghhBBCCCFqTZ0G6Xl5eURHRxMdHQ3A4cOHiY6OJi4uDlBD1UePHl1Wfvjw4fzyyy98/PHHHDp0iDVr1vDwww/To0cPGjRoUBeXIIQQQgghhBBC1Jo6He6+adMmBg4cWPZ84sSJAIwZM4aZM2eSlJRUFrADjB07ltzcXD788EMee+wxfH19ueyyy3jttdfOed2FEEIIIYQQQojaptMusnHiOTk5+Pj4kJ2djbe3d11XRwghhJB701kg76kQQoj65FTuS+fVnHQhhBBCCCGEEOJCJkG6EEIIIYQQQghRT0iQLoQQQogzMm3aNBo3bozVaqVr166sWrXqhOU/+ugjWrdujZubGy1btuTrr7+u8PrMmTPR6XSVHkVFRWfzMoQQQoh64bxaJ72++XlzAp+tOsSlLYOZNLRVXVdHCCGEOOdmz57NhAkTmDZtGn379uXTTz9l6NCh7Nq1i4iIiErlP/74YyZPnsxnn31G9+7d2bhxI3fddRd+fn4MHz68rJy3tzd79+6tsK/Vaj3r1yOEEOL8lJZnY8HWRP7ak0ojP3fG9omiZahXlWVziuxsic1kU0wmB1LzCPG2EO7vToS/OyHeVjTA6XLhcGo4XRrtGvngbTWds2uRIP0M2Bwu9iTn0sDXra6rIoQQQtSJt99+mzvuuIM777wTgHfffZfFixfz8ccfM3Xq1Erlv/nmG+655x5GjRoFQJMmTVi/fj2vvfZahSBdp9MRGhp6bi5CCCHEeSW3yE5SdhFJ2UUkZBbw564UVu5Pw+kqz4k+a2McfZsFcHuvKMxGHXuT89ifksuupBz2puRyKunTf76vN10j/c/ClVRNgvQzEO6vgvO4jII6rokQQghx7hUXF7N582YmTZpUYfvgwYNZu3ZtlfvYbLZKPeJubm5s3LgRu92OyaR6KvLy8oiMjMTpdNKpUydeeuklOnfuXG1dbDYbNput7HlOTs7pXpYQQoizKM/m4N+4TPzczYT7uePtZkSn051wn6yCYlYfSGPlvqOs3JdGck7V0586NvLhyvZhRMdnsXhnMmsOpLPmQHqVZaMC3OkW5U+rUC/S84uJyyggPqOAtFwber0OQ8nDqNdhNhjO+LpPhQTpZyDczx2AhMwCNE076R+XEEIIcSFJS0vD6XQSEhJSYXtISAjJyclV7nPFFVfw+eefM2LECLp06cLmzZv58ssvsdvtpKWlERYWRqtWrZg5cybt27cnJyeH9957j759+7J161aaN29e5XGnTp3KCy+8UOvXKIQQ4sxpmsa2hGx++CeO+dGJ5Bc7y17zshgJ93enU4QvXSP86BrpR4CnmU2xmaw/lM76QxlsT8jCdVzPt4+biTAfK6E+Vjo09OGazg1pGuRZ9npCZgHfrI9lwdYkvKxGmod40TLEkxYhXnSK8CXYq/5OoZIg/Qw08HVDr4Miu4ujebZ6/Q8thBBCnC3HN1KfqOH62WefJTk5mV69eqFpGiEhIYwdO5bXX38dQ0lPRa9evejVq1fZPn379qVLly588MEHvP/++1Ued/LkyUycOLHseU5ODuHh4Wd6aUIIIU6TzeFkS2wWaw6k8deeVHYnlY9wCvOxYne6SMsrJtfmYFdSDruScvh+Q1y1x2sR4kn/5kH0bxFEl0g/PC0nDmUb+bkzeWhrJg9tXWvXdK5IkH4GzEY9YT5uHMkqJD6jQIJ0IYQQF5XAwEAMBkOlXvPU1NRKveul3Nzc+PLLL/n0009JSUkhLCyM6dOn4+XlRWBgYJX76PV6unfvzv79+6uti8ViwWKxnP7FCCGEKKNpGpoGen3lBle708XupBzsTo2oAHf8PczodDqcLo0dR7JZdyidNQfS+CcmgyK7q2w/s1HPle1CualHBD0b+6PT6SgsdnIkq4ADqXlsictic2wm2xOyKXa6iAxwp1fjAHo19adXkwDCfC6ePGASpJ+hcP/SIL2QrpF1XRshhBDi3DGbzXTt2pWlS5dy7bXXlm1funQp11xzzQn3NZlMNGrUCIAffviBYcOGoddXvTKspmlER0fTvn372qu8EEKIMpqmEZteUDK8XA0xT8+30TjQg+YhXrQI9sKpaWyKyeDfuCwK7eXD1T1LhqsnZBaQW+SocNxATwuXNAugT7NALm8dgp+HucLrbmYDzYK9aBbsxZB2YYDqgc8tchDoefE2vEqQfobC/dxZT4YkjxNCCHFRmjhxIrfffjvdunWjd+/eTJ8+nbi4OO69915ADUM/cuRI2Vro+/btY+PGjfTs2ZPMzEzefvttduzYwVdffVV2zBdeeIFevXrRvHlzcnJyeP/994mOjuajjz6qk2sUQoj6yuXSyCq0czTXRmpuUclPG0dzbaTl2dABFqMBs1GPu9lA6zBvujf2p2HJ6lTJ2UXMiz7CL1sS2JeSV+n4+1Ly2JeSx+8kVdju42bCw2wgKaeIPJujbCi7l9VIryYB9G4SQN9mgbQI8TzlvF0WowGL57lN1FbfSJB+hiL8VfK4eAnShRBCXIRGjRpFeno6L774IklJSbRr144//viDyEg1vCwpKYm4uPI5hk6nk7feeou9e/diMpkYOHAga9euJSoqqqxMVlYWd999N8nJyfj4+NC5c2dWrlxJjx49zvXlCSFEvaBpGot2JLPuUDopOUWk5NhIySkiLc+G3XkKa4mVaOBjJczXjS1xmWVLkZkMOjqH+9GriRpe3sjPnYNH89iXksu+lDw0NLpG+tE9yp9mQZ7o9TqK7E4SMguIyygg0NNC2wY+GKoYIi9OjU7TTmWFuPNfTk4OPj4+ZGdn4+3tfcbHm/fvESbMjqZnY39m39O7FmoohBDiYlPb9yYh76kQ4sKRb3Pw9NztzItOrLaMv4eZIE8Lwd4WgjwtBHlZCPA0o0NHsdOFze4ku9BOdHwWOxJzKqwn3iPKn+u6NGRo+zB83Ezn4pIuSqdyX5Ke9DNUulZ6QmZhHddECCGEEEIIcSHZnZTDA99v4dDRfAx6Hbf1jKBZsCfB3lZCvK0Ee1kI9LRgNlad06MqBcUOouOyiM0ooG/TQCIC3M/iFYjTIUH6GQovGe6emF1IscN1Sv9BhBBCCCGEEOJ4mfnFzN4UzztL92FzuAj1tvLhLZ3pFuV/xsd2Nxvp0yyQPrVQT3F2SJB+hoI8LVhNeorsLhKzCokK9KjrKgkhhBBCCCHOE06Xht3potjpYueRHH74J46FO5Ipdqjlyy5tGcTbN3bC/7jM6OLCJUH6GdLpdIT7ubM/NY/4zAIJ0oUQQgghhBAn5HJpvLZ4DzPWxJQF48drE+bNmD6R3NA1vMr1ysWFS4L0WhDur4J0WYZNCCGEEEKIC0t8RgFrD6ax9mA6Gw5l4GExcEO3cEZ2bVS2lveB1DzmbI7nr92pRAW4c1uvSPo3D6oyuC6yO3l0djQLdyRXes3TYmR4xzBu7hFB+4Y+p7x8mbgwSJBeC8qXYZPkcUIIIYQQQlwI4tILePTHaDbHZlZ67X8L9/Dm4r38p3UIR/NsFcocSM3jz2OC9UtbBtM40AODXkdmfjF3fb2JTbGZmA16XhvZnktbBGMy6jEZdJj0euk1FxKk14ZGfirDu6yVLoQQQgghxNmVZ3MAqte5OjaHkxV7j/Lb1kRW7jtKkcOFpmm4NLUeeJ+mgQxpF8rlrUPwq2Ku929bE/nvL9vJtTkw6nV0CvelT9MAejUJICGzkO83xhEdn8Winao33KDXMbBlEMM7NiA6PoufNicQk17Ay7/v5uXfd+NmMtAqzIv0vGLiMgrwthqZProbvZoEnJ03SZzXJEivBWU96ZkSpAshhBBCCHE2pOfZ+GDZAb7bEItOp2NYhzBu7xVJp3BfdDodR3NtrD2Yxsp9aSzZlUxukaPK4zhdGsv2pLJsTyoGvY5ukX60DvOmaZAHTYI8WbAtkVkb4wHoHuXHuzd1pqGvW4Vj3Ng9nN1JOfy2NRFfdxMjOjUk2NsKwDWdGvLEFS2Z928iv2xJYEdiNoV2J//GZQHQwMfKzPE9aBHidfbeLHFekyC9FpQuwyZz0oUQQgghhKhdBcUOvlx9mE/+PlTWiw4av2w5wi9bjtAmzBuXprEnObfCfmE+VoZ1COPK9mGEeFvR63TodZCeX8zSXSks3JHM7qQcNhzOYMPhjAr76nTw4MBmPDKoOUZD1Usstw7zpnWYd5WvuZuN3NIzglt6RuB0aRxOy2NnYg4Z+cVc1SGMYC/rGb8v4sIlQXotKA3Sswrs5BTZ8baa6rhGQgghhBBCnJ82x2bw1+5U9qfmcTA1j9iMApwuDYB2Db2ZPLQ1bmYD366PZcG2JHYl5ZTt2ybMm77NAri8TSjdIv2qnN8d7G2ldZg3Dw9qTmx6PusPpXPwaD6HjuZx8Gg+FqOeKcPa0KdZYK1cj0Gvo1mwF82Cpedc1IwE6bXA02LE38NMRn4x8RkFtG3gU9dVEkIIIYQQ4pwrsjv5c3cK6w+l06tJAFe1D6uQoTy3yM5ri/awaEcKV7YPZVzfxjQuWcI4Nj2fqX/sKZvnfazIAHcmXt6C4R0alAXeXSL8ePaqNizemYyHxUifpgEElGRbr6nIAA8iA2QJZVG/SJBeS8L93EqC9EIJ0oUQQgghxAUpq6CYf+Oy+Dc+C6fLRYCHhUAvCx5mA8v3pjI/OpGckrng366P4+uoWKYMb0O7hj4s35vK079sJzG7CICv18XyzfpYBrUKppGfO99tiMXu1NDr1Lzujo18aB7iRbNgT4K9LFUuR+bnYeamHhHn9D0Q4myTIL2WhPu7szUhmwRJHieEEEIIIeoxTdP4cNkB5kYf4bHLW3JVh7AKr2cX2Hnlj11sjs3EzWzA3WzE3WzgSGYh+1PzTnr8MB8rfZoG8vv2RDbGZDD8w9V0ifArW6YsMsCd+wY0ZcmuFJbtSeXP3all+/ZvEcQzV7WWpGrioiZBei2R5HFCCCGEEOJ88OGyA7y1dB8AD3y/hfWHInn6qtZYTQY2xWTwyA/RHMkqrHb/JoEedI7ww9NiIC2/mLRcG1kFdlqHeTGyazi9mwZg0Ot4bHAL/rdwD/O3JrI5NhOdDu7o25jHBrfEzWzgph4RHDqax8y1MSRmFXJrr0gGtgw+V2+DEPWWBOm1pGwZNgnShRBCCCFEHUjKLsTXzYyb2VBtmS9WHy4L0C9tGcSKvUf5Zn0sW+Iy6d8iiE//PohLU73d/72yNWaDnoJiJwXFDvzczXSO8K3xvO8Gvm68f3Nnbu8dya/RR7iuSyO6RPhVKNMkyJMXr2l3+hctxAVIgvRaEu4nPelCCCGEEOLc0jSNtQfTef+v/Ww4nIGH2cDlbUIY3rEB/ZoHYTaWLx82a2McLy3YBcBjl7fgoUHNWb43lYmzo9mZmMPORJUl/brODXlxRDs8LbUTKnSP8qd7lH+tHEuIi4EE6bWktCc9IbMQl0urcrkHIYQQQgghToWmacSkF7By31FWH0jD5nDR0NdKAx83fD3M/LIlgX/jssrK5xc7mRedyLzoRLwsRvw8zGWvxZfkTrp3QFMevKwZAANbBvPHI/2YOHsru5JyeG54G67r0uicXqMQoiIJ0s9U4r/gGUqYbwh6HdgcLo7m2QjxttZ1zYQQQgghxHnC4XSx6kAa2+KzyS2yk1vkINdmZ/uRbOIzqp8fDmAx6rm5RwR3929Cck4R86MT+X17EkdzbeTaHBXKjukdyVNDWlbIlB7m48asu3vhdGkYpKNJiDonQfqZWPYyrHwDut+F6ao3CfNx40hWIfEZBRKkCyGEEEKIk9qXkstPmxOY++8RjubaqixjMujoFunPgJZB+HuYScwqJDGrkOQcG23CvBl/SRTBXuq7ZwNfN7V++LA27EnOweZwlR3H22qiWbBntXWRAF2I+kGC9DPRuL8K0rd8Bf0mEuHvzpGsQuIyCugm826EEEIIIUQ1UnOK+O/c7RWWH/P3MHNZq2ACPMx4WY14WU008nOjV5MAPE5xfrhBr6NtA5/arrYQ4hyQIP1MRPWDiN4Qtw7WvEe4/y2sOwR7U3LrumZCCCGEEKIe0jSN+VsTmfLrTrIL7Rj1Oi5rFczIro24tGVwhURvQoiLk3wKnAmdDgY8pX7fPJPLGmkAfLMuluTsojqsmBBCCCGEqG8Sswq579stPPJDNNmFdto39OGPR/oxfXQ3BrcNlQBdCAFIkH7mmlwK4T3BUcQVmbPpEuFLQbGTV/7YXdc1E0IIIYQQdUzTNDbHZvDA91vo9/pyFu1MxqjXMfHyFvxyfx9ahHjVdRWFuLBkxoL9xMkW6zsJ0s+UTgcDnlS/bp7BK5erLO+/bU1k7cG0Oq6cEEIIIYSoK/tSchnx0Rqu/3gdv29LwunS6NXEn18f7MvDg5pjMshXcSFq1Zr34L0OMGMoOO11XZvTVqefDCtXrmT48OE0aNAAnU7HvHnzTrqPzWbj6aefJjIyEovFQtOmTfnyyy/PfmVPpOkgaNgNHIW0PjSD23pFAvDcrzuxO10n2VkIIYQQQlxo0vJsjJvxD1sTsjEb9dzYrRELH+nHD3f3loRuQtQ2TYO/XoKlU9TzxH9hwycn38/lhNQ9kJsCrvoTt9Vp4rj8/Hw6duzIuHHjuP7662u0z4033khKSgpffPEFzZo1IzU1FYfDcfIdzyadDi6dBN+NhH++4PF7HuD3bUnsT81j5poY7urfpLyspqnyQgghhBDiglTscHH/t1s4klVI40APZt/Tq2yJNCFELXO5YNEk2Pipet7scjiwFJZPhTYjwDe8+n2XPAvrP1K/Gyzg0xB8IyGsIzTsAg26gE+jcx6/1WmQPnToUIYOHVrj8osWLeLvv//m0KFD+PurJc6ioqJOuI/NZsNmK19zMicn57TqelLN/gMNOkPiv3jPvYUX+7/CAwuLeffPfQxqHUwTw1FY8Cgkb4ORM6DJgLNTDyGEEEIIUaee/20nG2My8LIY+Wx0NwnQxYXF5YRlL4MtB9pep1a70p/CAO2ts2HzDOh8G3S69cwCYKcdfnsEor9Tz698E7rdATOvVCtwLZoEN31X9b7pB8sDe3TgtEHGIfU4tLy8nEcQXDcdml52+vU8RefVRJj58+fTrVs3Xn/9dRo2bEiLFi14/PHHKSysPjHA1KlT8fHxKXuEh5+gJeVM6HQw9A2w+kLiv1y55kYmBG+hoNjO9x88jeOjXuofuyAdZt0MCZvOTj2EEEIIIUSd+WZ9LN9viEOng/dv7kyzYM+6rpIQ4HRAzOrqE6oVZMCBv8BRfPJj/fk8rH4b/vlcBcPvtlfDzHfOhT1/qOPErIHCrIr72fJg7n0w924VQP/6AHwzAjJjysvYC2HfEoiepep8IoVZaiRz9HegM8C1n0KPu1SDwVVvg94IexbA3oVV77/sZXA5oPlgePYoPLINxv4Bw9+DLmMgtL06bv5R8Ao7+ftSi3Sapmnn9IzV0Ol0zJ07lxEjRlRbZsiQIaxYsYL//Oc/TJkyhbS0NO6//34uu+yyauelV9WTHh4eTnZ2Nt7e3rV9GZCdAD/fBXFrAUg1hBDsTAFgt6UDTYK8sCSsUcH8uD8gpG3t10EIIcR5JScnBx8fn7N3b7oIyXsqzoWsgmJeW7SHA6l55BY5yCm0k5xThEuDp4a04r5Lm9Z1FYVQvc0/joG9v0PDrnDbz+DmV/56Vhx8NVwFy96NoM+D0GU0mD0qHyv6e5h3n/q95VUQs0r1qFdFb1Kjh1tfDQFN4bcJkL4fdHrVA79nATiKwOQO3e+EtH1w6G9wlDQktBoG138BpipGomTGwPej4OgeMHnAyC+g5XEjtJc+B2veBZ8IeGB9xes5sgU+Gwjo4N7VENqu6muwF0LydvW+6Q1Vl6mhU7kvnVdB+uDBg1m1ahXJycn4+KiEG7/88gsjR44kPz8fNze3k57nnNy0XU5Y+Sb8/T/QXNgNbrxafDMz7ZcRaHbyk+cbRBbsQPMIRjd+kfqjBTWfQqc7O3MeshMgfgO0Gg5Gc+0fXwghxGmTgLL2yXsqzrYiu5PbPt/AptjMSq/d0LURr4/sgE7yEF1YXC6IXQ3ugRDS5uyfLzFa9eI2+8/J44P4jaqHO6QtDJoClpKl/VwumHcvbJtdXja0A4z+Fdz91dDur66G7PiKx3MPgB73QJfbwbuB2ha3Ab4aBs5i6P8EXPYM2Itg/2LYOQ/yUsBhU4+ibMiOq1xPrwZw/ecQ1VcNN5//sHpPj+XdEPLT1PDzyEvg5u/Bekyyxdi18OPo8h7uW2arOeTHK86Hj3qperQbCdd8CCY3lSPs66vh8EroeDNcW4MEc7Xggg3Sx4wZw5o1azhw4EDZtt27d9OmTRv27dtH8+bNT3qec3rTjt8Ie36HbuM4YA/ksTlb2RqfhTd5/GB+hTb6WJw6E3qjCZ2zWA238AyB9jdAx5vUEIvakH4QvhwC+amq1er6z8+4JUgIIUTtkYCy9sl7Ks4mp0vjge+2sGhnMl5WIy9c3ZYgLwteVhP+7mYiAtzruoqiOkXZsPs3aHmlClJrojhf9SBv+ATSD4DZCx7dAW6+Fcu5XLD1e2jUA4JanH4dEzarzr79S9TzqH5q+HZVx3TYYPmrsPZ90Eqyk/tGwDXTIOoS+P0x2PSFGrY9+CVY/Y4KboPbwlVvwU/jITcRAprDrXPU9Nw175UPQdfp1UpWbUeoRoD8o6qH+8ZvTj4PPW0/7PpVPZK3QYuhcM1H4BFQ8T379xuV6K1BZ2h+hWpoiFmtpggX50JIe7j6fTj8N2ybA6k71b6hHVSAXtqIUJX9S+G7GwBNxVY3fg0Zh+Hb68Bghoc2q/frHLhgg/Tp06czYcIEUlNT8fRU83t+/fVXrrvuOvLy8upPT3o1XC6N9YfS+XFTPBt37GGG/mVa6hOq3yG4rWplsvqq1iOrDwS3hrBOYKhhzr+seBWg5xxznq7jYNg7kmVeCCHqCQkoa5+8p6LU4bR85myKx9vNRNMgT5oEeeBtNbEtIYvNsZlsis0k3+bgyvZhjOzaiBDvEyd50zSN5+bv5Ot1sZgNer6+owe9mgSccB9Rj8y+HXbPr9ijeyIbP1Nzl4uyKm6/5iOV+OxY236EX+5S39nH/n7qHW7JO+CvF1XPNKjA2mBSQ8L1JrhkAvR+UAXmthzV+73ov3B0tyrfZgQkblHD10EldItbB+jUtbYfCUf3qp7zvOTy8wa1gtHzwStEPXc6YNc82PQlxK6pWMeQ9jB+EVhOMddCUQ5YT/GzOGkrfHu9ahg4lsEM7a5XSeJqUo+Dy+DnO1VuMIsPuPupRoheD8CQV0+tTmfgvAnS8/LyynrFO3fuzNtvv83AgQPx9/cnIiKCyZMnc+TIEb7++uuy8q1bt6ZXr1688MILpKWlceeddzJgwAA+++yzGp2zvty0swvtLIiOZ+WGDexKKaRYM+HAwKUesdxgWk032waMmr3qnS3eqmWscX/1ny+krfoPfLzcFJgxRA1jCWgOve6F3x8HNLhkIvznudq9KE1TCfG2zlKjCMJ7QIdR6qc0CAghRLXqy73pQiLvqQDYlZjDbV9sICO/BsmwAINex8CWwfRq4k9aXjGpOUWk5Bah1+kI9bYS5mMlo6CYb9erxHAf3NyZYR1O0IsnTs+SZ+HQCojsA00uhci+NQ/wsuJg8X+hw03QeljF19IPwgddgZLwR6eHAZOg/+NVjzI9shk+K8no7dcYet0HuckqaVrjATBmfsXyM64qH7rtEQTjFkFgs/LX7YWQtE11uh17PU676uH++zU1slanV/Xv/7hKfvbHE+WBe1U8glSys1ZXgS0XljwDm2eWvz7sXeg2ruL78NVwyDmigu7R88AjsOpjpx9UidmiZ6lOwrG/n7Oe57Lzf3u9CqqjLlEjjttcXXFOfU1kH4Gfxqnpv6DiqYejK/bqn2XnTZC+YsUKBg4cWGn7mDFjmDlzJmPHjiUmJoYVK1aUvbZnzx4eeugh1qxZQ0BAADfeeCMvv/xyjXrRoX7etLclZPH9hjjmb02koNgJgDd5XGHYRKQuBR9dAeFudsKthTQs2IXVkVthf5fRiq5BZ3QNu6phO0Y3MFpUxsXUXeo/0rhFat2/TTNgwQS148CnVa+6Z5B6rmlqKMr2n1QLo6apD8Zmg1SDwPH/GTQNchJV692RLar1MH1/5Qv0jVTD93s/eOotaEIIcRGoj/em8528pxeP6PgsXlqwi3YNvHngsmZly51tT8jmti82kF1op1WoF81DvDh0NI9DR/MptDtpGuRBt0h/ukaq7zc/boqvcn55daYMa8P4SxqflWs679jyVJDZ5FJo3O/MjnVoBXx9TcVtOgNE9FK9p22vrX6YelE2fDFYJROz+qgg7Niyvz+mvh83uVT1pG/9Xm2P6gejvqn4XVfT1LESNqrA8NpPVSCfGQPvdQR08Nge8ApV5dMPwgddVIAd2FJ9P/ZupHqdPQLVd/A176p52yZ31evd5XZw81dzxhP/VcdpNQwuf7E8Z1VpXXb/ppYTyzmizm3xVt+ro/rB4JcrB5v7/4RVb6ne8+53VH6vchJh7x/qPa1pwKtpddPxZi9Uj5pOT6iO066G7P/zOQyZCt3G10r1auq8CdLrQn2+aefbHOxLySUuo4CYtAIOp+Xxb3wWsekFZWX0uGiji6Gvfie99bvorN+Pj66g2mNmGwOY2/ELfBo1J9zPnRBvK2E7PsG47IXyQu4BENRafWhUFWSD+sDxDFHBv9GqhtxkxYEtu2I5kzu0Hq7mrhxarj5QivPUa/5N1BrxDTqd5jskhBAXpvp8bzpfyXt6cVi0I4kJs6Mpsqu5uG4mA+P6RtG7aQD3f7eF3CIHncJ9+Wp8D3zc1KhDl0vD5nDhZq7cc3ogNY85m+NJyCgkyMtCiLeVEG+L6sfIKSIpu5DkbBu9mvhzZ78m5/Raa+RsBVGapuYN6/SV17V22OD7G1Vw7R6glrE61aHQpZx2+KSfCnBbDwePYHXcjIPlZfRGlUit23i1dFZpXZwO+P4GNbS5VI974MrX1e/56fBOW5U5fMxvqgNq6w+wYCLY81XP+G0/l49O3TYHfrlTZQ5/aDN4H7ME1+eXq+D9ileh9wNq258vqB72ZpfDiI9hxlD1vdo3Qr1HeWq1J0zuYK/iu7vVRw3fbn9D9f+GLpeao232OrV1yUVFLmed5OeSIP0EzsebdnJ2ERsOp7MpJhObw4nVZMDNZECn07HrSCbpcbto7dxLW10M7tiw6oqxYKcACx86RnBIqzwMa4LbQkbplhLiSkFP+Z+AU28mPrA///oMohALzXM30CR7AwGFMVVXTmeAgGYQ3Ep9ULa5pjybJEBxAUU7fsO4/AWMuUfUHJLBr6g1DGUIvBBCAOfnvam+k/f0wpKcXYTd6aKhrxt6vQ5N0/hs1SGmLtyDpkG/5oHk2Rz8G5dVYb9ukX7MGNcdL2sV0wIvNIdXqYzX3e+Ey56u/LrLBWl7wSf81IPov1+H5a+o39tdr+Zjm9xUsPPTeDV/udSgKdDvsdO7hvWfwKKnVO/yw1vKe3gzY1Xyse1z1KjPUk0HwdDX1HfRPx5XPaQmdxj4XzXkW2eA+9erhGul1xDWEe7+u/x7aNI2lb/Jng897oYr31CJ4j7srnqtL3tGZTI/1sbP1PkadIa7V6gGgnfbQW4S3PCVSrKWfUQdtzTDuW8E9HtcZRNP/Fc1euycqzqzml2uEqOdKAGaOO9JkH4CF+JN2+F0sTMxh+1HVK+22ajHYtSjaXAkq5C49ALiMgpIyCogJdtGsdNVtq8bRTTVJdJCl4AdI8tdncijckbSYDIJ1GVjpRiLzo4FOxn6AGy+TQn286aRnxu+7ma8rEa8rCYMOh3bj2Tzb1wme1Ny8dby+Mp/Jp0K1PrxNB+s1kxs0EklqzCY1AdcXrJaLi4vRSWJyE9XSR7c/aFRN7VGYekHtqZBQYbKSGkvApddtcCiqZEBpckvjudy1U7ro6apD2/3wKrXbxRCiBq6EO9NdU3e0/NfWp6NP7Yn8Wt0IptLhqF7mA00D/HCy2pk1f40AG7vFclzw9tg0OtYuiuFNxbvZX9qHj0b+/Pl2O54WGqYbPd8lhkD0wdCYQYYLDBhe+XvQX8+r4aklw7HbthFDfs+Uc8tqMD395KgW2cAzamC05u+V0sOb/pCjbDscrtKNGb1Ub3px2c+P1b6QUjdrdbQLu3cyU+D97uoUZrHz6E+1tG9sOVr2DhdLQWmN0HTy0rmbOtg1LdqLvqsm9Vw7uZXqIze77SFgjS17nb7kRWPued3+OFWQFNZ1PNS1Pxw3wh4YKNqkDhW3lF4q6V6Lx7crHr6v79RjSSYuKd8ueOMw6pxIKKXCs6PXwbZlqdyR4W2l86ri4AE6Sdwsd+0NU0js8BOcnYRcRkF7E/JZV9qHvtTcskvdhDsZSXYy0KwlwWr2YDDqeFwuih2amTmF5OUU0RydiGpuTZO5S9Hp1PnvteylCf032LQHOUvGq2qxTQvRX3YnUxAyVJ72QlqyFJ1fCNV0rqgViqYProP0vappeiMVtXSavZUc/Ib9YDI3ioRn8VLzeVP3gEpO9UNwCtUDff3DFYfuHHrIG69+rA3WtWHb5NLIaq/CtiLC1SLbHG+amQoSC9Z77FYJfpr2K2kceIi+OIghDipi/3edDbIe3r+2nEkm/f/2s9fe1JxutSXDZ0OTHp9hY4GnQ6euaoN4/tGVViP3OnS2JWYQ6swL0yGi2BIcHG+mjudsqN82yWPwn+eL3+ekwjvdVLrTh9vyGsquXBVds6FOeMADQY8pYaIz75dNQaYPNR3HXQw8ks1mnJab9VbP+Ap1Zt9LE1Ty2qtnwZ7F6pjegSrcp1vhz8eU8nOQjuo3umTDUdOP6gSxO1bVL7t8heh7yPq97QDMK2nSsTWbiTs+EmNInj436oTLq96S2VW1xvVw1FU3itelW9HqmXDBkxS7/2eBec8W7g4v0iQfgJy064ddqeL5Owi4jMLSMgs5EhmIdmFdnKLHOQW2bE5XLQK9aJzhB9dInzJLLAz6Zdt/BuXRWtdLHd4b6CLMZZGRfswO/PLjqvpjei8G4BXmMpU6R6gHjlHIOEf1dp4PPdAMHuoD1y9SQXCGYeA0/zT1unL15mszbLHM7lDcBs1SsDqo5ba8w6D8J5qxMCxrbZOu1pmw+VSSUKsPio/wOnKjFXvUaNuFacnVCU/veQGqEF4L5XIRFp7hahVcm+qffKenn/2JOfw7tL9LNpZvjRUh0Y+XNOpIcM7hOHnYSYmLZ89ybkcTsunV5MAejQ+w0RSdU3T1Pxks8fJyzodqmfY5VBzsq3eav+fxqlg2iNIBcd/PK6WmXp0R3nC3gUTVY93ZF8VUCf+C7sXQPS3qhPika2Ve4sPLlfrS7vsav73VW+r+39mjOqlTt2lyl31lhpiD7BzHswZo+ZMT9hWnujr8EpY/HTFoeoewarjBCCwhVpTG00lO47sXfP3cN8SWPWmurZBUyp+R1k4CTZ8XP588CvQ58Gqj6Npavm07XPU88i+KpN5dd95ts6GuXeDd0PV0eRywH3rIKRNzesuLioSpJ+A3LTrjsul8d2GWF5btJc8m+pJ1+EiSpeCD/kkagFk6X0J8nYn0NOMt5sJL6sRb6v66WExEqDLJcK2D4vFCj6NMPs1wt3DA71Oh0vTcLnUZ2kDNzs+adEqsE8/oIYrBbaAwOaqFdVRpFqeiwsg85ie8ZSdgKYC/9B2ENJO3Thzk9UHcF6KuplFlPS6N+ikblaHVqibWcJGQAfmkl56k3t5Q4NHgHotaau6Odpyqn+z9CY1Z8rNVwXTWXHqw/9YRquqi2+EuibvBirzZX4q5KWq4/s3VUPSGnRSN8N9C9UNNHFL+XkieqnpB+E91TAsnUG1Xif+Czt+Udd27AgHjyC1T2ljQljH8i8XBRnqpp1xWNUnpK2qo06nRj7s+V0lE0zephoayt6bYPBppB6+4ap1viBNjT4oSCtfdjCoVc0bCGx5am5YYaZKWnj80iLpB2H/EtW70GKI+vesbhqEvRASo9X7ZvaEFleUZ3OtSkGG6inY8zs06KKWbAltV7N6i4uS3Jtqn7yn9Z/LpbE7OYdV+9NYtf8oaw6kA+pjfniHBjx4WTNahJykIfl85nLBnNGwd5HqAe51X/X3uAN/wuJnytfDNpihyUB1T47+Vt3Px/ym7s3Teqne7NJe5cwY+KCbCrbH/lG+LrijGN7vDDkJMPQN6Hl3+flyEuGjXmroeZsRKrA/tmfblgsr31D35U63VLymT/tDynboO0E1Gvz5PGz8VL1udINON0PP+8AvSg2PXzG1fA3y9jeo9bxrS0GGusaiLPVd4tGdJ15pyF4IX49Q31PuWHLidc5tefBGs/JRnQ27wl3Lqi8vLnoSpJ+A3LTrXmpuEX/vPUp8hporH5eheuPT8my4avGv0d/DTGSAO2E+VnKLHKTnFZORX0yx00VkgDtNgzxpGuRJiLcFu9OFzeFCK8jGqBVj9gnB292Mt9VEsdNFak4RR/NsHM214XRpmA16TEZ92TA6TdNwaRo6dEQEuNM61JvmIZ5YTdUM1XK5VMbPtH1QmKWWDCnKUq3IcevV3PzjGa3qpnyi4L6mdHo1WiHnSM3Kh7ZXreJHNlceKqfTqykIRVnlmUuP5R6gvkQc3XPG1cY9UK2Z6huhRjBoLpW0xpargvHCTDUELzdFZT89lncj1aDgGaxa9I/NFAvqmB1uUl9ecpNVw0hWbMm0hx2VG0kadoWWQ1UjiEewaozQG2HDJ+px/L9T4/7QZYyq69G96v3ITVKNFW7+qrfBYFa5GEobhDRNfYnxb6IevhHq380rVD2KCyArRo2MyI5X74XRqkZZmNzUtfqEq1Z+i6f6O8uMVddVmKnmJIa2K29k0TR1nKStqhErpB0Etaw8LNDlLPmbLfm7LcpR1x/Y/NxmS9U09X66HOq9Lx1Nc6rTSFwu9b7nJKhRK2GdznmeCbk31T55T+uvgmIH05Yf5Id/4kjLq7h++VUdwnhkUPMLOzgvteY9WDql/Hnn21RvdelIOU1Tn8d/vQgH/1LbrL7qnnr8ajzD3ilfTurfb+HXB8AzVPVmL5ioAvkmA9V62McqnW/u3VANAzda1Hm/H6XmeDfoDOMXn9rovb0LYdZNqqPCK6z8ftt1nOrpPn4ZrcJMWPW2+g40/L3qcwqdrs1fwW8Pq6WHBzx58vJOhxrGb/U5edmfxsOOn9XvJ5pHLwQSpJ+Q3LTrL4fTxdE8G8nZRWTkF5NTZCen0EF2oZ08m4M8m4N8m4O8IkfZ89witU0D9DrQ6XQ4XRoZ+cUnPd/ZZtDriPR3x8NixGTQYTLo8bAYaRHiRfuGPrRr6E2Ev3uFeXSAujlmxULcBtXj799EDTH3DFU9vaVBaVG2aunOjlcBZU6iCrY8g1XQaPZQLe6J0eqRm6iGbrW9Vi1r4hmsepMP/Kl6lI/uUz3mLqf66Rmi5pe1vQ4Cm6m6OWyqhz1unQrYj2ypHOj7RoJ/Y5XVNOPgMdMBdKoHvtUwFbA6bKqXvCBdBcXZCSWPeDX0zyNIBeUegeoccRtOnIOgKmYv1WJeVWOE3qgCfq8Gqsf7+KD+eJ4hKpdAXgoc2XTyc4e0U6sYHPpbZaStSb6Fs8noVs37p1PBtWeIaowoPG6NYINFjYhwL8kbkZeqAtqqpnkY3VTQH9peBcvF+Sprrb2ACkksNFdJokeH+olONTr4NFJfFM3u6m8hK049inJU4K03quPaC8v/dpxV/F9381ONMqWjMwKbq8aGoFbq7yplp/objl2jvgDnJFY8jtGqRlY0HagadjSt5P+GS72HRTmqEcaWq14bOPl0/kUqkHtT7ZP3tP7RNI3FO1N4acEujmSpzyMPs4FeTQLo1zyQS1sGExVYg2HfF4L4f2DGENXI2PpqNZ9Zc6lpZUP/p5YR2zanvOdcb1KZx/s/rj6PU/fA7vmwf6nKiXNsNndHsVrHOzdR9Wav/UB9ht35l5rmdiyHTc1Vz00sD/Sjv4d596mG43tWqRV8ToWmweeD1PcEUPfZaz5QQ/TrSl6q+vyv7el6exfBrFHq/vf43poF9uKiJUH6CchN++KQb3MQk55PTFoBKTlF+LiZ8Pc0E+BhxqDXEZNWwMGjeRw8mkd6XjEWox6LSY/FaMDh0sgtspNTaCe70I7JoCfY20qQp4UgLwtmg45ip4bd6aLY4UKnA71Oh04HDqfGwaN57E7KIbPAftJ6Wk16/N3N+Lib8XUzEehloZGfG+F+7jTycyPE24qHxYCXxYSHxYDxTBLgnK01IXOSVHDn5q8CoWOXdbEXql7jnCPQqLtqGDhdjmI13Dx2req91enLHxYvFZiVPjxDVNBXOt++KEfVMWmrCv7Ce6ovNaVD3ooL1Dy/rbPUUH2fhqrX2jdSNZA06q56pEtv7rnJap7+/qWqZzovRQWNmkvlGbh0ErQaXj58PiteZaE98JeaBhDUUj18GqlAryBDjQBw2tWXCM8Q9dCcqj6Zh1WDSs4R1fuem6wacED1UvhGqvoazGq706auKTdZ9Q4XZZe/j+4BqrzVR2XWPX7Uht4Ewa3VsMDk7WqoY3VMHuo4Fi/1vtrzqy9bX+iNlUdFAGUNBS5n+RzJmjC5w9NJZ1wtuTfVPnlP65e49AKmzN/Bir1HAWjo68YzV7VmUOsQzMaLILnbsQqz4NN+qhGy7bUwcobqKZ8zvvJnrsGsGtYHPq3uRzW19gO1BFmpFkPgltlVl93wKSx8Ut3nxv6u1im3ZcOg56DfxFO+PEA1QswZq+61V7xy4kzv5zOXS62NHtRKZZQX4gQkSD8BuWmLc0HTNFJzbRw8mofN4cLucGF3amQWFLMrKYedR7LZnZxLsePUks4Z9TqMBh0mvRpubzaoxgWr0YDZqC/Jog8aaui9r7uJAA8zAZ4WvKxG8orUyISsQjv5pXkBdKBDh7vZQNNgT5oHe9I82At3i4GkrCISswtJzi7CZNATGeBOhL87kQHueJiNOFxqmL/DpWHU6zAb9Oj1p9ZK7XC6cLi06qcGnC9Kh4C7+Z39xHqaphoqjNbKiX6qUpSjer89gysnCsxNUXPv8o+qBobg1hWHWmbGQFK06hUvXeHAM0Q1yhy7lIzLqfInJG1VPdU6nRrNYfZSdTy+gchgLh+i7nKoxp6cBDUCw15Ynp/AN0K9py5Hec+70apGWJSOtDCYVQOHy64ac/JSykdlZMWpIZRH96jGDs2l6hResqJDeE81pcArTNVF01TD0qEVcGi5un6dQTW46Azq3FZv1YhhKRmp8Z8XzvjfXO5NtU/e0/rB6dKYseYwby7ZS5Hdhdmg5+7+TXhgYDPczOf55/7p0DS1lvnu+eqz556V5b2vaftVQrb0A9C4H7S/UQXopxPg2nLVkmOljbT3rFQjg6piL1Q973kpatReXrLKp3LHUlmFRohaJEH6CchNW9QXdqeLpKwisgqLySywk1VQTGqOjYTMAuIzC4nPKCAjv5g8mwPbKQbzdcmg12Ey6NCXBC061EgDN7MBD4sRd7MaEZBdoK47u1CNOPBxMxHqbSXEx4rVqCerwE5mQTGZBcVoGvi6m/BzN+PrbsbLasRqMmA16XEzGfD3MBPqYyXU20qQl4Uiu4uskuPnFtkxGnRYjAbMBj1WkwEfNxO+7urhaTGi0+nQNE0l2XW5KCx2UlDyKLI7sTmc2Owqb4G72UAjf3dCva0YTrFB4kRKz3+qjRziFNiL1JdP70b17oun3Jtqn7ynde9Aai5P/KRWdgHo3SSAV65tR5MgzxPveCHSNIjfqOaAb/9RNVLesUTlNzmWo1hNEzp+3vbpWPYKrHxdJX678asTl133kVrODEqGua9UjbZCiFpzKvel+vUtRYiLiMmgJyLAnQjcT1q22OEivyRYtzvVw+HSKHa4SoJIFzaHE00r7xl3aRpZBXbS822k5xWTU+TA22rE2608OIXyqcJZBcUcOJrH/pQ8DqSqEQChPlbCSh7FDhexGQXEpReQfoI5/06XVrau7bFybQ7IrWJ91hLZJdML9qZUPTdcnbP+DKc26nWE+ljxczfjZjJgNRuwGPUUO1wUFDvIs6kA381kwNvNiJfVhLvZQE6hncySBoicQjvFDhfFTjXSwqjX0dDPjQh/NWIh0NNS1kFbOtohwFONjAj0NBPsZSXAwyyBfU2ZrKrnSghx1q0/lM7oLzZS7HThaTHy3ytbc3OP8Mp5WC509kI13WnLNxWTvV3+UuUAHdQIJWMtLSs34CmVI6TpZScv23UcrH5Hjaq6dJIE6ELUMQnShTgPmI16zMcOLa5jeTYHxQ4XBr1OPXQ6HC4VaBaXNCSUDrsHFbgX2lXPdL7Ngd2pqTwBHiZ83c2Y9HpScotIzi4iOacIm92Jv4cFPw/Vew6QWVBc1rteYHNSaFdBcEGxk7Q8Gyk5at+juTbcTAb83M34uJvwtppwaZrqCXe6KCx2kFPoILOg+IQjFEwGHW4mA25mg+q1L5lSkFtk50hWIXanRkJmIQmZp5jM7gQcLo3Y9AJi0wtqvI/ZoC9rTPGyGtXfikFf8jejViAwG/QYS0Y36HQ6dKjGHJcGaBouTZ3b5nBSZFcNPmjgaTXiZTXiaTHhaTFgMRpKcjfoATX6wKmpRhmjvvycZoMeDTUywFUyQsBi1ONuNuJm1uNmNmI16rGY1OgGk0F38X1xF+IC9836WIqdLno18eftGzvRwLcGU3POFVueSmoWdcnZXZHCUayGrx9arp6b3FWvdpfRp7YO+OkyGKHN1TUra3ZXc9YT/4UuY89qtYQQJydBuhDilHlajFBpNZYz+6Lj424650vuFNmd5JXOzUetDmDQqx5r0wmS9LlcKudAQmYBuUWOsgaDIrsLs1GPp8WAu1kNyS+0O8kptJNTZKew2Im3VY1k8PMw4+NmwnJMMF3scJUtSxiXXkBmgRqxUBJLU1CslhJMy7ORlldMer6NYmf5Pucrg16Hl9WIj5tqVHEzGXC41GgRh1Oj2KmmIBTZVeOMDnAzqwYUN5OaRuFpUQ0KHmYjFpMeo15ftqqCl9WEX9n0BhO5ReWjGfJtDpVPoeTfwGLUlzXMuJkMGPQ6bMeMWHGW5F8wlOaHMOgZ1qFBXb+FQtQrLpfG2gNpADw+uGX9CtDthTBjqMrF0XgAXP8FeAbVfH9Ngy1fqTwYzQeDX2TV5VwumP+gCtBNHip5WrvrT7xGd11r2LXq3n0hxDknQboQ4qKl5rWfeuOCvmSoe6hP7a+l3cDXjV5NAmpU1u50kZJTRFK2ehTYHBSXrDpw7NSI0hEOmqbGNpT2but1urKlCw16HRajvmyuv6ZRtsxhaQND6bSKIrs6lkGvesANJUsflp672Okqy0VQ2kFe7HCVjaYoLHZS7CwfxeB0qakZWTVYEaFUfnEdL2lXws1kkCBdiOPsKlnhxMNsoGO477k5aeoetfznidbz1jT4bYIK0AEO/62yrN8wUy0R6nJCzGq17rXeAEP+V/l40d/Db4+UPw9qDS0Gq2XUGnYtTyL51wuwbbZKOHnj19C8DpcfE0KcdyRIF0KI85TJoKeRnzuN/E6e16C+cZUE9TaH6iXPLVKjDXIK1ciE0gSEhpIecXezUc39N6kRDoV2FewXFjvJL3aSZ7OTW6QaFYodLtUT79SwOVzkFjnILknQmFfkwNNqLOlZN+NpMeLStLLGBZvDha2kx76w2InTpWExGcoaMPQ6cLrAWdLTbz6TZRGFuECtKelF79Uk4ISjkmrNP1/A7xMhpD2MX1h5FYuycp/Dth9U4DzsbVg3DdL2wsyroO11ELNKLXNZyuSuesBL5afBkpL1yAOaqxUtju5WjzXvgV9jaH+DSgq35l1V7uoPJEAXQpwyCdKFEEKcc3q9DqveUJZt/2yMShBC1I3VJUF632aBZ/9kKbvKs5KnbIef74Sbvq881zxuPSyapH6//EXoOhbajVS94jt+UhnXQS2H1rg/7P4N1n0ITQdCs5Ige8kzUJgJIe3g7hUqC/uBv2DvQvXIPKyyqZe67FnofOvZvHohxAVKugCEEEIIIUStKLI7+ScmA4BLmh8TpCdtgz+egKN7a3agPX/A5pK539WxF8LPd4CjCBp2A6MV9i2CJc9WLJd+UK1N7nKoHvPeD6jtFk+4/nPV293xFhj1HTy+H0Z9C93vVGXm3gd5R+HQCtg6C9DB8PfAYAI3P2g/EkZ+AU/sV/PbWwxRS5j1fhD6PVazaxVCiONIT7oQQgghhKgVW+IyKbK7CPKy0Dy4ZD30nCT49jq1vNe/38LQ16HzbeXzt4+3bQ78UhIkb/gEhr2j5owfb8mzkLoLPILh5h/UcPWfxsH6jyCwGQS3Uet/71kAmks9v+bDiufV6VS29S6jKx578MsQu1Ydf+49qpccVPDeqFvlupg9VMDefiRl66EKIcRpkp50IYQQQghRK0rno1/SLFAtrei0w5yxKkA3WsFeoLKe/3IX2HIrHyBmDfx6v/rdYFFB8pdXwPyHVI94YZZK8LbnD/jnM1Xu2k9UhvZ218HAZ9S2BY+q/XbPVwF608tUIG/2qNmFmNxUz7jRCgf/UvPPvcJg0JST7ysBuhDiDEmQLoQQQgghTkmR3cmmmAyyj1uVYfWBdOCY+eh/Pg/x68HiDfeuUUGuzgDb58Anl8C/34HDpsoe3Qc/3ALOYmg9HCbugs63q9e2fA0fdIHXIuFFf5hdMte794PQbFB5Bfo/Dh1uUr8bLKqH/P71cPvc6pdLq05IG9WjXmro6/V7CTUhxAVDhrsLIYQQQoiT2peSy9JdKazen8bmuEyKHS6aBHnw6wN98bKayC6wsz0hC4C+zQJg5zyVfA1gxDQ1BL3fYxB5iZpLnhmjes3/fB66jVdzvouyoFF3uO4z1Zt9zYfQ6VaVHC51l5p/Dqp3vGFXGPRcxUrqdHDNR9DhBgjteGproFel+52qTnqjajgQQohzQIJ0IYQQQghxQl+tjeGF33bi0sq36XVw6Gg+T8zZxse3dWHdoXRcGjQN8iDMkQi/PqgK9nmoYoAb0RPuW6OWTtv4GeQmwt//U6/5NVbD0k1u5eUje8Pdy9XvDhsU5UBxLng3BKO5cmUNxvKM7GdKp4P+T9TOsYQQooYkSBdCCCGEEFVyujRe/n0XM9bEANCveSCD24TQt1kg2YV2Rn26nkU7k/l05SGOZBYCaj46Gz9TgXRE78q93aCWOus3UQXwO+epBHGFmXDLj+BxgqXbjJaS3vEz7CEXQoh6TOakCyGEEOKMTJs2jcaNG2O1WunatSurVq06YfmPPvqI1q1b4+bmRsuWLfn6668rlfn5559p06YNFouFNm3aMHfu3LNVfVGNfJuDe77ZVBagPzWkFV+P78HtvaNoEuRJ5wg/nru6DQCvL9rDb9sSgZL56HHr1EG636mWK6uOwaSGpt/1Fzy8RQ2JF0KIi5wE6UIIIYQ4bbNnz2bChAk8/fTT/Pvvv/Tr14+hQ4cSFxdXZfmPP/6YyZMn8/zzz7Nz505eeOEFHnjgAX777beyMuvWrWPUqFHcfvvtbN26ldtvv50bb7yRDRs2nKvLuiBl5BezZGcyNofzpGX3Jucy8pN1/Lk7FbNRz0e3dOG+S5uqjO3HuKVHBCO7NsKlQVaBHb0OeoVbIHm7KhDe82xcihBCXNB0mqZpJy924cjJycHHx4fs7Gy8vSVDpxBCiLp3Pt+bevbsSZcuXfj444/LtrVu3ZoRI0YwderUSuX79OlD3759eeONN8q2TZgwgU2bNrF69WoARo0aRU5ODgsXLiwrM2TIEPz8/Jg1a1aN6nU+v6dnw4HUXEZ/sZHE7CKaBXvy2vXt6RrpX6mcw+li+qpDvLt0P8VOF/4eZj4b3Y2ukX7VHrvI7uT6j9eyMzGHTuG+zLvSCV8NV3PGJ+46m5clhBDnjVO5L0lPuhBCCCFOS3FxMZs3b2bw4MEVtg8ePJi1a9dWuY/NZsNqtVbY5ubmxsaNG7Hb1XJe69atq3TMK664otpjlh43JyenwkMom2MzGfnJOhKzVWb0A6l5jPxkHc/O20FukZ08m4MDqXn8ve8oIz9Zx+uL9lLsdDGoVTCLHul3wgAdwGoyMH10N67t3JBJQ1tBXMmIB+lFF0KI0yKJ44QQQghxWtLS0nA6nYSEhFTYHhISQnJycpX7XHHFFXz++eeMGDGCLl26sHnzZr788kvsdjtpaWmEhYWRnJx8SscEmDp1Ki+88MKZX9QF5q/dKTzw/RaK7C46hfvyzqhOTFt+gDmbE/hmfSzfb4zD6ao4qNLLauS54W25vkvDSsPbq9PQ1413RnVST9auVz8lSBdCiNMiPelCCCGEOCPHB3KaplUb3D377LMMHTqUXr16YTKZuOaaaxg7diwABoPhtI4JMHnyZLKzs8se8fHxp3k1FwZN0/hqbQx3f7OZIruLgS2D+P6unjQO9OCNGzry/Z09iQxwLwvQvSxGmgZ5cE2nBix5tD8juzaqcYBegcsF8f+o3yMkSBdCiNMhPelCCCGEOC2BgYEYDIZKPdypqamVesJLubm58eWXX/Lpp5+SkpJCWFgY06dPx8vLi8BAtfRWaGjoKR0TwGKxYLFYzvCKLgx5NgeTft7Ggm1JAFzfpRH/u749JkN530yfZoH8NXEASdlF+HuY8bDU0lfCo3vAlg0mDwhpXzvHFEKIi4z0pAshhBDitJjNZrp27crSpUsrbF+6dCl9+vQ54b4mk4lGjRphMBj44YcfGDZsGHq9+lrSu3fvSsdcsmTJSY8pVFb2qz9czYJtSRj1Op65qjVv3tChQoBeymjQE+7vXnsBOkB8yVD3Rl3BIH1BQghxOuTTUwghhBCnbeLEidx+++1069aN3r17M336dOLi4rj33nsBNQz9yJEjZWuh79u3j40bN9KzZ08yMzN5++232bFjB1999VXZMR955BH69+/Pa6+9xjXXXMOvv/7Kn3/+WZb9XVTmcml8tyGWV/7YTZHdRai3lY9u7VxlBvezKn6j+inz0cUZWByzmPjceMa0HYNJb6rr6ghxzkmQLoQQQojTNmrUKNLT03nxxRdJSkqiXbt2/PHHH0RGRgKQlJRUYc10p9PJW2+9xd69ezGZTAwcOJC1a9cSFRVVVqZPnz788MMPPPPMMzz77LM0bdqU2bNn07OnBH5ViUsv4Mmft7L+UAYA/ZoH8u6oTgR41sHw/7jSpHG9zv25xXlP0zQ++PcDPtv+GQDphek81eOpSuWWxi7F1+JL99Du57qKQpwTsk66EEIIUcfk3lT7LsT31OXS+H5jHEdzbXhYDLiZjWTlFzNtxUEK7U7cTAaeGtKS0b2j0OuPS/qmabD+YzjwJwx7G/yiar+CeanwZnNAB0/FgJtv7Z9DXLDsLjsvrnuReQfmVdj+6iWvMrzp8LLnP+79kZfXv4yHyYPZw2YT4R1xjmsqxOk5lfuS9KQLIYQQQpwHpq86xP8W7qnytd5NAnjt+g5EBLhXftGWB7/eD7t+Vc//fAFumFH9iTQNCtIhOx68G4JncM0qWNqLHtxaAnRxQrnFuRzIOoBep8eoN2LUGXlvy3usOrIKvU7PlF5TSMxPZPq26byw7gWa+DahjX8bPtn6CdO2TgPgysZX0tCzYR1fiRBnhwTpQgghhBD13ObYDN5YvBeAK9uHYjUaKCh2Uux0cXmbEEZ1C6/cew6QfhB+uBWO7ga9CVx22DkXBj4Ngc0qll3zPmz5CrITwFGkthnMMOAp6PsIGE4yNzh+g/op89EvekWOIv5J/gen5sRisOBmdMPusrMpeRNrE9eyPW07Ts1ZaT+rwcobA97g0vBLcWku9mbs5e+Ev5mwfAJ9G/Tl5/0/A3Bfx/u4r+N9p7dMoBDngToN0leuXMkbb7zB5s2bSUpKYu7cuYwYMaJG+65Zs4YBAwbQrl07oqOjz2o9hRBCCCHqSlZBMQ/Pisbp0ri6YwPeu6lTzYKTxGj46mq1JJpnKIz6Bla9BfsWwZp34ZoPy8vuWwJLnz1mZx24+UFhBix7CXbOg2s+gAadqz9faZAeIfPRLzRZRVl8ueNLsouz6dugL70b9MbL7FWp3O703fyy/xd+P/Q7ufbcEx4z1CMUo86I3WXH4XLg7+bPc72fo2NQRwD0Oj1T+03llt9vISYnhp/3/4wOHf/t+V9uanXTWblOIeqLOg3S8/Pz6dixI+PGjeP666+v8X7Z2dmMHj2aQYMGkZKSchZrKIQQQghx7tgcTmwOF95W1WutaRqPz9nGkaxCogLceeXadugcNji4DHbPh9i10Ot+6HVv5YNt/EwF6A27wk3fg1co9HtMBelbf4BLJ4FPI7DlwoJH1T5dx0LfCWqYu8EE2+fAwqcgZTt8Ngj+85zqVT+evVA1CoD0pF9ANE1jwaEFvPHPG2TaMgH4Zf8vGHVGuoR0IcQ9hFx7LnnFeaQVphGTE1O2b6hHKEFuQRQ6CrE5bbg0F20D2tKnQR96N+hNA88GJz2/l9mL9wa+x61/3EqRs4ip/aYyJGrI2bpcIeqNOg3Shw4dytChQ095v3vuuYdbbrkFg8HAvHnzar9iQgghhBDnWJ7NwdUfrubQ0XwaB3rQsZEPZqOeP3enYDbo+eiGVngtnqB6tYvzynfcPKPqID1tn/rZ+0EVoAOE94CofhCzCtZ+AENfg2UvQ04C+EbAFa+C2aP8GB1uhCYDYdFTsONn+PN5aH01+DeueK7EaDWU3iP47CSlE+dcfG48L617iXVJ6wBo5tuMXmG9WH1kNTE5MWxM3lhpH5PexKCIQVzX/Dp6hvVEr9OfcT2a+DZh/oj5ODUnoR6hZ3w8Ic4H592c9BkzZnDw4EG+/fZbXn755ZOWt9ls2Gy2suc5OTlns3pCCCGEEKflld93cehoPgCH0/I5nJZf9tqUK5vRdtUDcPAvtcG7oQqeo79V886dDjAc87VO08qD9MAWFU/U7zEVpG/+CppcChs+VduHvVMxQC/lGQQjv4TCLHX+dR/CVW9VLLPla/Uzsg/IPOHz3tajW7n/z/vJKc7BrDdzb8d7Gdt2LCaDiad4iricONYkrqHIUYSn2RMvkxeeZk/aBbTD1+pb6/UJcg+q9WMKUZ+dV0H6/v37mTRpEqtWrcJorFnVp06dygsvvHCWayaEEEIIcfpW7E1l1sZ4AD4b3Q2zUU90XBbbj2TRItiDW5P+pwJkk7uaW950kArEd/4C9gLIjKmYCK4gHYqyAB0ENK14siaXQoMukLhFJZVDgw43QbP/nLiSl0xQdfj3WxgwSQXvAKm7YdsP6vc+D5/xeyHq1oakDTy07CEKHYW0D2zP1H5TifSOrFAmwjtClj4T4iw68zEo54jT6eSWW27hhRdeoEWLFiffocTkyZPJzs4ue8THx5/FWgohhBBCnJrsQjuTft4OwNg+UVzeJoQBLYJ45D/N+Xx0N57UZqDb8RPojXDjNyqY1ulAr4eAksA8bW/Fg5b2ovuGg8mt4ms6HfR/XP2uOcE9EIZMPXlFo/qp4N5RBBs/Ld++7GXQXNB6ODTqehrvgKgvlsct5/4/76fQUUivsF58PvjzSgG6EOLsO2+C9NzcXDZt2sSDDz6I0WjEaDTy4osvsnXrVoxGI8uWLatyP4vFgre3d4WHEEIIIUR98cJvO0nOKaJxoAdPDWlV8cWVb8DG6YAOrv0Umh/X2x3UUv08enyQvl/9DGhe9UlbDIWQ9ur3oa+Bu//JK6rTqd50KElKlwfx/8CeBaDTw2XPnnB3UXd2pu3k6dVPM3zucP638X8k5iVWeD2rKIsvtn/BoysepdhVzGXhl/HhoA9xN7nXUY2FuLidN8Pdvb292b59e4Vt06ZNY9myZfz00080bty4mj2FEEIIIeqnJTuT+WXLEfQ6ePOGjriZDeUvHlwOy19Rvw99HdqPrHyAwJIgvbTnvFR189FL6fVw+y+QGQvh3Wte4VbDwL8pZBxUa6rvXai2d7qlvMFA1AsOl4MlMUv4bs93bDu6rWx7TE4MP+z5gaGNhzKg0QCWxi5lefxy7C47AMObDOfFvi9i1J83YYIQF5w6/d+Xl5fHgQMHyp4fPnyY6Oho/P39iYiIYPLkyRw5coSvv/4avV5Pu3btKuwfHByM1WqttF0IIYQQor5buD2JR3+MBuCu/k3oGulX/qItF+Y/pH7vfif0vLvqgwSW9JQfH6Snl3y/Onae+vE8g9XjVOgN0Pdh+O0RWP6qyjJvMKs56qJecGkuFscsZlr0tLIl0Yx6I1dEXUHfBn359eCvbEjawIJDC1hwaEHZfq39WzOyxUhGthhZK1nZhRCnr06D9E2bNjFw4MCy5xMnTgRgzJgxzJw5k6SkJOLi4uqqekIIIYQQtU7TNN7/6wDv/KkC6/4tgnj0P8f1eC95FrLjwTcS/nOCBLhlw933qURypZnVS4P26oa7n4kON6kAPS9FPe9+l5r7LuqUS3OxMmElH/z7Afsy1b+/n8WPm1vfzA0tbiDQLRCA4U2HszN9JzN3zGRPxh76NuzLiGYjaOXf6kSHF0KcQzpN07S6rsS5lJOTg4+PD9nZ2TI/XQghRL0g96baV1/f08JiJ4//tJXftyUBMK5vFE9f2Rqj4Ziey4PL4ZsR6vcxC6Bxv+oP6CiGV0JVAriJu8G7wXHb9oB3WO1fyOp31JrpZi94JBo8Amv/HKJGkvOTmXdgHvMOzONI3hEAPE2ejGk7htvb3I6HqYpl9YQQ59yp3JdksokQQgghxDky6Zdt/L4tCaNex0sj2nFzj+OWsbLlwvySZcy633XiAB3AaAb/xmp4+9G9KkjPPKwCdLMneIWenQvpcQ/kJkPjARKg15ECewHPrnmWP+P+xKW5APAyeXFDyxsY3248PhafOq6hEOJ0SZAuhBBCCHEOJGYV8ttWlVV75rgeXNL8uODW5YSFT0F2XMkw9+drduDAFipIT9sHTQcekzSuefnw99pmdldZ4UWd+Xz75yyJXQJA99DuXNvsWi6PvByr0VrHNRNCnCkJ0oUQQgghzoHvN8Th0qBXE//KAXpmLMy9B+LWqefXfAgWz5odOLAF7P2jPDg/2fJr4ryXmJfIVzu/AuDNAW9yRdQVdVwjIURtkiBdCCGEEOIsszmc/PCPSoY7undU+QuaBtt+hD8eB1uOGqJ+1dvQuH/ND378WumlQXp1y6+J897bm9+m2FVMj9AeDI4cXNfVEULUMgnShRBCCCHOskU7kknLKybE28LlbULKX1j8X1g/Tf3eqAdcN13NMT8Vx6+Vnl4apJ9g+TVx3tqSsoXFMYvR6/Q82f1JdGdrSoMQos5IkC6EEEIIcZZ9sy4WgFt6RGIqzeSevL08QL/0v9DvMTCcxlez0rXS81KgMOuYOenSk36hcWkuXvtH5QK4rvl1tPRvWcc1EkKcDfqTFxFCCCGEEKdrV2IOm2IzMep13NzjmPXEl09VP9teB5c+dXoBOoDVG7xKllmLXQtF2YAO/JucUb1F/TP/4Hx2pe/C0+TJg50erOvqCCHOEulJF0IIIYQ4i75ZHwPAkHahBHuXZN4+shn2/g46PVw6+cxPEtgCcpNgz+/quW8EmNzO/LiizmUWZbI+aT3rEtexNHYpAPd0uIcAt4A6rpkQ4myRIF0IIYQQ4izJLrQz71+17NrtvSLLX1j+qvrZYRQE1cKw9KCWcPhvleUdyofAi/OGw+VgR9oOdqXvIj43nrjcOOJy4ojNiUVDKyvX2r81t7S+pQ5rKoQ42yRIF0IIIYQ4S37anECh3UnLEC96NPZXG+PWw4E/QW+EAU/WzolK558XZqifsvzaeSEhN4G1iWtZm7iWjUkbybXnVlmuhV8Leof1pk+DPnQL7YbZYD7HNRVCnEsSpAshhBBCnAV2p4svVx8GYHSfyPIs3MteVj873Vp788aDjksgJj3p9ZJLc7E+aT2rElax+shqYnJiKrzuY/Ghc3BnGns3ppFXIyK8I2jm24xAt8C6qbAQok5IkC6EEEIIcRbMj05Elx3H827LGZWzGdb4gr0QYlaBwQz9n6i9kx2fyV2C9HpH0zQmrZzEwpiFZdsMOgMdgzrSt2Ff+jToQ2v/1hj0hjqspRCiPpAgXQghhBCilrlcGp/8fZAJxp8Zqa2ENccV6DoWfMOr2vX0eIaAxQds2eq5LL9W73yx4wsWxizEqDMyrOkw+jXsR68GvfA2e9d11YQQ9YwE6UIIIYQQtezP3SnsT82jrSVObWh5JVi8oChHZV0fMKl2T6jTqQR0Cf+A2UsF7aLeWJWwive3vA/A5J6TubHljXVcIyFEfSZBuhBCCCFELdI0jWkrDqLHRTN9EmjAFa+c/XXLA1uqID2wuQraRb0QmxPLUyufQkNjZIuREqALIU5KX9cVEEIIIYS4kKw/lEF0fBZNjGmYNBsYreAbefIdz1RYB/UztN3ZP5eokbTCNB5e9jC59lw6BXVico/JdV0lIcR5QHrShRBCCCFq0bQVBwAY27wIDqN6ts9FMrDOt4PRAi2Gnv1ziWo5XU7WJa3jl/2/sDx+OQ6Xg2C3YN6+9G1ZOk0IUSMSpAshhBBC1JIdR7JZtT8Ng17HsLAcFaQHtTo3Jze7q4R0os7E58Zzz9J7iM+NL9vWLqAdz/V5jiD3oDqsmRDifCJBuhBCCCFELflpcwIAV7UPwzfvV7XxXAXpos59vv1z4nPj8TZ7M7zpcK5tdi0t/VuefEchhDiGBOlCCCGEEKcrKw7mjINe90H7kazcfxSAK9uHwZo9qowE6ReFAnsBiw4vAuC9ge/RLbRbHddICHG+ksRxQgghhBCn69/v4MgmWP0uCZkFHDqaj0Gvo3cTP0jbp8pIkH5RWByzmAJHAZHekXQN6VrX1RFCnMckSBdCCCGEOF0JG9XPlB1s2HUQgE7hvvjYksBeAAYz+EXVXf3EOTP3wFwARjQbgU6WwBNCnAEJ0oUQQgghTofLCQmbSp5opO74G4B+zQPh6F61OaA5GGR24YXuUPYh/k39F71Oz9VNr67r6gghznMSpAshhBBCnI6je8CWU/bUPWk9AP2aB6nXAIIkadjFYN6BeQD0a9iPYPfguq2MEOK8J0G6EEIIIcTpiC8Z6q5XPeUdXTvxshrp2MinvCc9uHUdVU6cK3aXnfkH5gNwbfNr67g2QogLgQTpQgghhBCnozRIb38jAO10MQxs7IbRoIfU3eo16Um/4K1KWEV6UTr+Vn/6N+pf19URQlwAJEgXQgghhDgdpUnj2l5LiiEUo87FNf4JoGnlPemS2f2ComkaW1K28FfcXxzKPoTdZWfufpUw7uqmV2PSm+q4hkKIC4FkMhFCCCGEOFX56ZB+AIC84M6sLm7B9YZkurELshPAnq+Gwfs3qeOKilOVkp9Cli2Llv4VR0EU2At4ZcMrzD84v2ybQWfApbkAuLaZDHUXQtQO6UkXQgghhDhVCf+on4EtWJ/oYr1L9Zj7pP5zTGb3ZmCQntXzyc60nVz767WM/G0kYxaOYfWR1WiaxqGsQ9z6x63MPzgfvU5PS7+WuBndcGpONDR6hPagia80yAghaof0pAshhBBCnKr4Depnox6s2n+UDa6SBHFHNkPiv+p3GepeL/0V9xe/HfyNW1vfSvfQ7mXbd6bv5K6ld5FrzwVgS+oW7vvzPlr6tSQuN45CRyGBboG83v91uod2R9M0UgpSSMxLpLlf87q6HCHEBUiCdCGEEEKIU1Xakx7eg1Ur0ojTgilyC8FamALR36nXJEivd/5J/ofH/34ch8vBX3F/MazJMB7r9hipBancveRucotz6RTUiRf7vshP+35izr457M1UIyN6hvbkf/3/R6BbIAA6nY5Qj1BCPULr8pKEEBcgCdKFEEIIIU6F06F6zIFknw4cOpqIQa/H0Lgv7PoFMg+rcpLZvV45nH2YCcsn4HA5aOrTlEPZh1hwaAF/x/+NTqcjpziHjkEd+fg/H+Np9uSJ7k9wZ/s7+WnfT7gZ3bi51c0Y9Ia6vow6k/H1NzjS0/G5ejiWpk3P2XldNhsZM2bgzMnF55qrsbaU/1fiwidBuhBCCCHEqUjZAfYCsPiwNjsASKRjIx9MTfqpIL2U9KTXG5lFmTzw1wPkFOfQIagDXwz+ggNZB3hp/UvsSt8FQIegDnzyn0/wNHuW7edn9eOuDnfVVbXrjfwNG0l59VUA0j/9FLeOHfG5/jp8hg1D7+5+1s5bsHkzSU8/Q3FMDAAZX36JtW1bde6rrsLg43PWzi1EXZIgXQghhBDiVJSuj96oG0eybAA0D/aCyL7lZXQGCDh3vY2ieplFmTyy/BHic+Np6NmQ9we+j9VopV1gO76/8nt+OfALh7IOcX+n+ysE6KJc2ocfAmBq1Ah7UhKFW7dSuHUrWT/9TNQPs9DpazcXtTMvn6Nvv03m998DYAgKxK1jR/L+XknRzp0U7dxJ6v9eo+nSJZhCQmr13ELUBxKkCyGEEEKcitKkceE9SckqAiDE2wKBLcA9EArS1NJrRksdVvLilW/PZ9aeWWw9upU9GXtIzk8GwMvkxbRB0whwCygra9AbuKHFDXVV1fNC/oaNFPzzDzqTichvvkZnNJI9/zfSpk2jaNs28pYtw+s//6m187kKC4m97TZse/YA4DPyekKeeAKDjw+OzExyfvuNrJ9+Rme1SoAuLlgSpAshhBBCnIqEkp708O6kxKme9GBvK+h0ENkHds+HYBnqfrYcLTjKVzu/ok1AG65scmWF1+xOO48se4QNyRsqbG/s05gpvaZcUMukOTIyiLnpZtw6dqTBa/+r9d7sUmkffQSA7w0jMYWFARBwx3icOTmkf/opadM/w3PQIHQ6XYX9NKeT4tg4bPv2qcehQzjSjuJMz8CRno7ObCb0mWfwHnJFhf2SX34Z2549GPz9afjWm3j07l32mtHPD//Ro/G7/XZcOTln5XqFqA/qNEhfuXIlb7zxBps3byYpKYm5c+cyYsSIasv/8ssvfPzxx0RHR2Oz2Wjbti3PP/88V1xxRbX7CCGEEELUmtxkyIoDdNCwG6k52wAI8baq17uMgX2Loc2IOqvihcqluZizdw7vbnmXPHseAEn5SdzR/g4ANE3jxfUvsiF5A+5Gdx7o9ABtAtrQ0r8lXmavuqz6WZG7ZAn2uDjscXGYIyIIeujBU9rfZbPhzM7GFBxcbZn8jRsp2LgRTCYC7qo4N99/9O1kzJxJ0bZtFGzYgEevXuX7rd/AkQkTcGZlnbAORx5/HJ3JiNegQQBkzZtH9s+/gF5Pw7ffqnDMY+l0OpmPLi5odRqk5+fn07FjR8aNG8f1119/0vIrV67k8ssv59VXX8XX15cZM2YwfPhwNmzYQOfOnc9BjYUQQghxUUvaqn4GtQKrNyk5qic9xLtkaHvz/8CzqXVUuQvXvsx9vLDuBbYdVY0iDT0bciTvCO9ueZd8ez4PdX6Iz7Z/xrwD89Dr9Lwx4A36N+pfx7U+u/JWry77Pe2jj7C2a4vXwIEn3Kdo927y16whf+06CjZvRisupuE7b+M9ZEiV5dM+mgaA78jry3rRSxkDAvC9/noyv/+e9OmflQXU9qQkjjz6KM6sLHRWK5ZmzbC0aIGleXNMoSEY/AMwBgaQ9umn5Mz/jYQJjxL+4QeYGjYk+YUXAQh84P5qA3QhLgZ1GqQPHTqUoUOH1rj8u+++W+H5q6++yq+//spvv/0mQboQQgghzr7MGPUzoClOl8bRvNIg3Vp3dbrAxeXEcdsft1HoKMTD5MHDnR9mVMtRzNw5k3e3vMtn2z9jZ/pO1iauBWByj8kXfICu2e0UrFsPgHvvXhSsW0/ik0/ReM6PmKOiqtwn9d13Sf/k00rbk597HrcuXSr1qOdv3EjBhg1gMhF4991VHtN//HgyZ88mf+1aCnfsxNqiOUcmPIozMxNrmzZEfv8demvV/zcavPoqWrGd3EWLSHjoYYzBwWiFhXj06U3gvfeewrshxIXnvJ6T7nK5yM3Nxd/fv9oyNpsNm81W9jxH5q8IcUFxOp3Y7fa6roYQJ2QymTAYLt71lS8ombHqp18U6fk2nC4NvQ4CPMx1W68LlKZpvLLhFQodhXQM6shbA94ixEMlC7uj/R14mDx4ZcMrZQH66DajuanVTXVZ5XOiMDoaV34+Bj8/wj/5hLix4yj8918SHnqYqB9moffwqFA+f926sgDd89JL8ejTB/eePUmaPJmiXbtInvIcjT6eVjav3FVQwNG33gbA9/rrKvWilzI3aoj3VVeSM/830qdPxxgcTOHWrei9vWn4/nvVBugAOqORhm+8ToLdTt5ff2FPSMAYFESDN95AJ5+X4iJ3Xgfpb731Fvn5+dx4443Vlpk6dSovvPDCOayVEOJc0DSN5ORksk4y302I+sLX15fQ0NBKyZXEeaa0J90vitSSoe6BnhaMhrOTtOtityhmEWsT12LWm3n1klfLAvRSN7W6CXeTOy+vf5nLIi5jYteJdVTTcytv9RoAPPr0QW+x0PDddzk88nps+/eTMOFRGr37Tlmg7szKInHSZAB8R40i7IXny44T9r+pxFw/krwVK8j+ZS6+11+HIz2d+Pvup2jbNnRubtX2opcKuPNOcub/Ru6SJWXbGrz2P8yNGp30OnQmEw3feZvEJ56kYP16Gr7zNsaAgJPuJ8SF7rwN0mfNmsXzzz/Pr7/+SvAJEl5MnjyZiRPLP7BzcnIIDw8/F1UUQpxFpQF6cHAw7u7uEviIekvTNAoKCkhNVfOUw6rpkRLniaySnnTfSFJySpdfk6HuZ0NOcQ6vbXwNgLs63EWEd0SV5a5uejVDo4ZiMpjOZfXqVH7JfHSPfpcAYAoJptF77xE3/g7yV60idsxYwj/5GENAAEnPPY8jJQVzVBQhTz1Z4TjWFi0IeuRhUt98i5RXX8XUIIyk557HHheHwceHRh9/jKlBgxPWxdqiBZ4DB5K3fDkAAffcc9K58cfSm800eu9dNIcDnfG8DU2EqFXn5f+E2bNnc8cddzBnzhz+c5J1GS0WCxaLrFMqxIXE6XSWBegB0uIuzgNubm4ApKamEhwcLEPfz1eaVqEnPeXQcUnjxGkpdBQyc+dM4nLiuKHFDXQO7oxOp+P9Le+TXpROlHcU49uNP+Ex6kuAbk9JIfm55/G75WY8+5+defGO9HSKdu4EwLNv37Lt7l26EDlzhuoF37GDmJtuxmfECHIXLwajkQZvvone3b3S8fzHjSP3z78ojI4mbpx6n02NGhE+fTqWJo1rVKfABx4gf80a3Hv1JOjhh07ruiRAF6Lcefe/YdasWYwfP55Zs2Zx1VVX1XV1hBB1oHQOunsVXzaEqK9K/17tdrsE6eergnQoVkt/4RtBSk4cULJGujhlmqaxJHYJb256k+T8ZAAWHFpAh6AOXB5xOT/u/RGAZ3s9i9lwfsz5z/zmG/JWrKAwOpqmixdVuUyY5nLhzM7GmZ6OIz0Dvbsb1nbtajwiLH+tmn9vadUKY1BQhdfcOnUiatb3xN11N/b4eNI+/BCAoIcewq1d2yqPpzMYaPC/qRwacS1aURHWtm0J//QTjIGBNb5ut3Ztab5mNXp3d5lPLkQtqNMJVHl5eURHRxMdHQ3A4cOHiY6OJi5O3fQmT57M6NGjy8rPmjWL0aNH89Zbb9GrVy+Sk5NJTk4mOzu7LqovhKhjMsRdnE8u5L/XadOm0bhxY6xWK127dmXVqlUnLP/dd9/RsWNH3N3dCQsLY9y4caSnp5e9PnPmTHQ6XaVHUVHR2b6UEytNGucVBiYrqbklw929JEg/FXannXWJ67hzyZ08/vfjJOcnE+YRxvAmwzHrzWw7uo23Nr+FhsbwJsPpEdajrqtcY7nLVwBqHvjRjz6q9HrGd9+xt2s39vfuw6Fhw4kbM4aYG24kadIkXDX8+y4d6u55Sd8qXzdHRRH1wyys7dsD4NatKwF33nHCY5qjogj/9FMCH36IyK+/OqUAvZTBy0sCdCFqSZ32pG/atImBx8xZKZ07PmbMGGbOnElSUlJZwA7w6aef4nA4eOCBB3jggQfKtpeWF0IIIcS5NXv2bCZMmMC0adPo27cvn376KUOHDmXXrl1ERFSeQ7x69WpGjx7NO++8w/Dhwzly5Aj33nsvd955J3Pnzi0r5+3tzd69eyvsaz1BpuhzIvOw+ukXBVB5jXRRJU3TSMpPYnPKZlbEr2Bt4lry7GpEgsVg4Y52dzC23VjcjG5M7DaRWXtm8cOeH3A3ufNYt8fqtvKnoDgujuKDB0GnA00j8/tZ+N10E5YmTQDIW7WalJdfUdMmAL23N0Z/f4rj48n+dT5F+/cT/oFaLxzAfuQIOYsWobNa8bv5ZnR6PZrLRd4a1ZPucckl1dbFGBBA5NdfkbdqFZ59+9YoePbo2QOPnudPg4gQF7I6DdIvvfRStJIPqqocH3ivWLHi7FZICCHEGZk5cyYTJkyQrPsXkbfffps77riDO++8E4B3332XxYsX8/HHHzN16tRK5devX09UVBQPP/wwAI0bN+aee+7h9ddfr1BOp9MRGhp69i/gVByTNA6QxHEnkF6Yzo/7fmTb0W3sSt9FRlFGhdcDrAFcGn4pd3W4i4aeDcu2B7oF8lDnh7iv4324NNd5M8wdKEuc5t69O3p3d/JWrCDlf/8jYvp0iuPjOfL446Bp+N5wA6HPPoPOrK4tf/0Gjjz6KLZduzl8/Uj8x4wmb/UaCjdvLjt20bbthL38ErYDB3CmpaFzd8etS5cT1kfv5ob34MFn74KFEGeNrBcihBDnyPDhw6tNdrlu3Tp0Oh1btmwp2/bzzz9z2WWX4efnh7u7Oy1btmT8+PH8+++/FfYtLi7mjTfeoEuXLnh4eODj40PHjh155plnSExMrPJ8Y8eOrXI48bGP0zFq1Cj27dt3WvuWWrFiBTqdTgL980BxcTGbN29m8HGBwODBg1lbMm/2eH369CEhIYE//vgDTdNISUnhp59+qpRnJi8vj8jISBo1asSwYcMq/d0fz2azkZOTU+FR645JGgflPenB0pNewfK45Vw3/zqmRU9j9ZHVZBRlYNQZaRPQhrva38V3V37HshuX8Xyf5ysE6Mcy6o3nVYAOkFvSmeQ5cCDBTz0JJhP5K1eRs3gJCQ89jCs7G2uHDoQcE6ADePTqSeOff8Latq0aJv/e+ypA1+lw69wZDAayf/2VhAcfInfpUrVPjx7ozefX+yOEqDkJ0oUQ4hy54447WLZsGbGxsZVe+/LLL+nUqRNdSnpGnnrqKUaNGkWnTp2YP38+O3fuZPr06TRt2pT//ve/ZfvZbDYuv/xyXn31VcaOHcvKlSvZvHkzr7/+Ounp6XzwwQdV1uW9994jKSmp7AEwY8aMSttKFRcX1+ga3dzcTrgspriwpKWl4XQ6CQmpuHZ1SEgIycnJVe7Tp08fvvvuO0aNGoXZbCY0NBRfX98Kf6utWrVi5syZzJ8/n1mzZmG1Wunbty/79++vti5Tp07Fx8en7HFWllstnZPuF4nd6SI9v3S4u/SkAxTYC3h+7fM8vPxhMooyaO7XnMk9JvPdld+x/tb1zB42m4e7PEyHoA7odefuK6gjM5P0GTPJXbEC5xnmMdI0jfSZM8ldtqzCdmduLgX/bALAa+ClWBo3xv+22wBUL/mePRj8/Wn0/ntVBtemBg2I/O5b/G65GbfOnQl+8kmarVhO1KzvafThB+isVvL+/pu0aR8DJx7qLoS4AGgXmezsbA3QsrOz67oqQojTVFhYqO3atUsrLCys66qcErvdroWEhGjPP/98he35+fmal5eX9sEHH2iapmnr1q3TAO29996r8jgul6vs96lTp2p6vV7bsmXLScueCKDNnTu37PmAAQO0Bx54QHv00Ue1gIAArX///pqmadpbb72ltWvXTnN3d9caNWqk3XfffVpubm7ZfjNmzNB8fHzKnj/33HNax44dta+//lqLjIzUvL29tVGjRmk5OTnV1mX58uUaoGVmZlb5ekZGhnb77bdrvr6+mpubmzZkyBBt3759Za/HxMRow4YN03x9fTV3d3etTZs22u+//1627y233KIFBgZqVqtVa9asmfbll1/W6D06Uyf6uz1f701HjhzRAG3t2rUVtr/88stay5Ytq9xn586dWlhYmPb6669rW7du1RYtWqS1b99eGz9+fLXncTqdWseOHbWHHnqo2jJFRUVadnZ22SM+Pr7239N32mvac96aFrNGS8wq0CKfWqA1nfy75nTW7P/ZharAXqD9vO9nbchPQ7R2M9tp7We21978503N5rDVddU0TdO0+Acf1Ha1bKUerVprB4dfrSW//rrmLCg45WPl/v23Ok679prt8OGy7dl//KHtatlKOzBkaNk2R06Otrd3H1W+TVstb/2G076G/M2btT3de5Rdx7HnFkKcH07lXn/eLcEmhBBV0TSNQruzTs7tZjLUaHi40Whk9OjRzJw5kylTppTtM2fOHIqLi7n11lsBtZKFp6cn999/f5XHOfZcs2bN4vLLL6dz584nLXuqvvrqK+677z7WrFlTlj9Er9fz/vvvExUVxeHDh7n//vt58sknmTZtWrXHOXjwIPPmzWPBggVkZmZy44038r///Y9XXnnltOo1duxY9u/fz/z58/H29uapp57iyiuvZNeuXZhMJh544AGKi4tZuXIlHh4e7Nq1C09PTwCeffZZdu3axcKFCwkMDOTAgQMUFhaeVj0EBAYGYjAYKvWap6amVupdLzV16lT69u3LE088AUCHDh3w8PCgX79+vPzyy4SFhVXaR6/X07179xP2pFssFiyWszjs3OmA7AT1u29k+VB3Lwt6/YWbuf9E4nPi+WHvD8w7MI+cYjW9INQjlFcveZXuod3ruHZK0d595C79E3Q6zBERFMfGYtu3D1vJtJyQkr/Dmsr8US0Lh91OyutvED5NZXDPLZmP7nlMQmSDlxehzzxN4tPPEPLkE2eUlM29Sxciv/2GhAcfwhwRgSky8rSPJYSo/yRIF0JcEArtTtpMWVwn59714hW4m2v2cTp+/HjeeOMNVqxYUba6xZdffsl1112Hn58fAPv27aNJkyYYjeXHfPvtt5kyZUrZ8yNHjuDj48O+ffu49NJLK5zj2muvZWnJvMUOHTpUOzf4ZJr9n737Dq/xfAM4/j3Ze0qIlQSxR4giVu1VWwmKRqKKlsao1drUqBGjRlUSra06/Fpq1opRK9oKYiYhIUSG7OTk/f0RORUJgixxf67rXHXe8bz3e0RP7vd5nvupVClbMS8vLy/Nnx0dHZk1axbDhw9/bpKenp6On58fpqamAAwcOJADBw68UpKemZz7+/vTuHFjIGM5r3LlyvHLL7/Qu3dvQkJC6NWrF7UeLz9U4XFlZYCQkBDq1q1L/fr1AXBwcHjpGMR/9PT0cHFxYd++ffTo0UOzfd++fXTr1i3HcxISErL8bAOadeOVZxSTVRSFgIAAzd9poYi9DYoatPXA1I57tyOAt2+NdEVROBdxjvUX13Mo9BAKGX9nZUzK4FbFjfcrv4+pnmnhBvmEyDVrADBt356y3ktIi4wk9vffuffVXKI2bsLa3T3bWuNJQUGoo6MxbpA1qU69F0Hc4yXW0NYm7uBB4vz9MW7YkPjDRzKu07JFlnPMOnXCrFOnPLkXg8qVqbjnj2K9nKMQIoPMSRdCiAJUtWpVGjdujI+PD5DRy3z06FE8PDyyHPf0L2EeHh4EBASwZs0a4uPjsyQzTx+7cuVKAgIC8PDwICEh4ZVjzUxkn/Tnn3/Stm1bypQpg6mpKYMGDSIyMpL4+PhntuPg4KBJ0AHs7OyIiIh4pZguXbqEjo4ODRs21GyztramSpUqXLp0CYBRo0Yxe/ZsmjRpwrRp0/j77781xw4fPpwtW7bg7OzM+PHjX/kBhvjPmDFj+O677/Dx8eHSpUuMHj2akJAQhg0bBsCkSZMYNGiQ5vguXbrw008/sWrVKm7cuIG/vz+jRo2iQYMGlC5dGoAZM2awZ88ebty4QUBAAJ6engQEBGjaLBSZ89EtyoOWFhGayu5vT9G4/cH76fd7P9z/cOfP0D9RUGhSpgnftP6G33v8zuCag4tUgp584yaxu3cDUGLYx0DG0mSWAwdi6OyMkpTEg2/XZjkn5dYtbvXtR8igD4k/9VeWfTE/7QC1GsN69bAakDHyKWLePBLOnEUdE4OWuXlGobd8JAm6EG8H6UkXQhQLhrraBM5sX2jXfhmenp58+umnfPPNN/j6+mJvb0/r1q01+52cnDh27Bipqano6uoCYGFhgYWFBbdv387SlpOTE5cvX86yLXO4sJWV1avcjoaxsXGW98HBwXTq1Ilhw4Yxa9YsrKysOHbsGJ6enqSmpj6zncx7yKRSqUhPT3+lmJ7X05r5y+uQIUNo3749v//+O3v37mXu3LksWrSIkSNH0rFjR4KDg/n999/Zv38/rVu35pNPPmHhwoWvFI/IqOgfGRnJzJkzCQ8Pp2bNmuzatQv7x8Nxw8PDCQkJ0Rzv7u7Oo0ePWLFiBWPHjsXCwoJWrVoxf/58zTHR0dEMHTqUu3fvYm5uTt26dTly5AgNGhTiGs7PqOz+NhSNS1Wn8tVfX/Fj0I9AxtrmXSp2YWD1gVQwr/CCswtP5Nq1oCiYtGyJQdWqmu0qlQqbz0YRMtiD6C1bsPYYjK6dHUpKCnfGjkN5/HDz3uzZOP60A5WuLopaTfT2jPu3dOuDSYsWxPy6k+Sr1wj/8ksATJo3R6Ujv1oLIV6f9KQLIYoFlUqFkZ5OobxetmejT58+aGtrs2nTJtavX8/gwYOztNGvXz/i4uKeO4T8yWP37dv3wuWp8sKZM2dIS0tj0aJFNGrUiMqVKz9zibf8Ur16ddLS0jh16pRmW2RkJEFBQVSrVk2zrVy5cgwbNoyffvqJsWPHsnbtf71lNjY2uLu7s2HDBry9vfn2228L9B6KoxEjRnDr1i2Sk5M5e/YszZs31+zz8/Pj0OOlqTKNHDmSixcvkpCQQFhYGBs2bKBMmf+W4lqyZAnBwcEkJycTERHBnj17cHV1Lajbydlbukb6g8QHeO715MegH1GhwqOmB3vf38s012lFOkFPuX2HmJ07gf960Z9k1KgRRg0aoKSm8mDVagAili4l6eJFtMzN0bawIPnqVaI2bQIg/vhxUsPC0DIzw7R9e7TNzbH5bBQAqY8fnj491F0IIV6VPO4TQogCZmJigpubG5MnTyYmJgZ3d/cs+11dXRk7dixjx44lODiYnj17Uq5cOcLDw1m3bh0qlQotrYxnrKNHj+b333+nVatWTJ8+nWbNmmFpaUlQUBC7d+/WzPXNCxUrViQtLY3ly5fTpUsX/P39Wb16dZ61/7R//vknyzB5AGdnZ7p168ZHH33EmjVrMDU1ZeLEiZQpU0YzB9rLy4uOHTtSuXJloqKiOHjwoCaBnzp1Ki4uLtSoUYPk5GR+++23LMm9EM/0dE/6o/8Kx72pYpJjuBV7i1olauW4JNo/9//B65AXEQkRmOqaMq/5PJqXbZ5DSwUr/tRfhH/5JfoVK2I3a2a2OeUAkd+tBbUa48aNMaxTJ9v+zN704A8GEP3TTxjUqMHDdRnTkOxmz0IdFcXdqdO4v3wFZp06Ef24YJx5t25oGWQ8mLHo3ZuoTZtJvnoVdHRkWTQhRJ6RnnQhhCgEnp6eREVF0aZNG8qXL59t/8KFC9m0aRPnz5+nc+fOODk50bt3b9LT0zlx4gRmZmYAGBgYcODAASZOnIivry9NmzalWrVqeHl50aRJE3755Zc8i9nZ2ZnFixczf/58atasycaNG5k7d26etf+05s2bU7du3SwvyFjP3cXFhc6dO+Pq6oqiKOzatUszrF6tVvPJJ59QrVo1OnToQJUqVTSjEvT09Jg0aRK1a9emefPmaGtrs2XLlny7h6LKwcGBmTNnZhmGLl5As0a6A8ATc9LfzJ70tPQ0PPd4MmDXAHr/rzf7g/eTrmRMQ7ny8ApjD43lg10fEJEQgaO5I5ve21ToCbqiKDz8/ntCPDxIDQ0l7tAhbnTvQZy/f5ZjEv/+m5gdPwFQYviz6xgYubhkJNZpadydNg0ACzc3zNq2xaJXLwxq1iQ9Lo7wL6fw6GBG9XbLPr0156t0dCj55Zegq4tp69ZoP/7/shBCvC6V8qwJfsVUbGws5ubmxMTEaH7JFUK8WZKSkrh58yaOjo4YGLyZvyCLt8/zfm4L+rtp+fLl+Pn5ceHCBVq2bImnpyc9evTI3yXMClief6YLKkLCA/j4CNjVoe7MvUQlpLLHqzlVShWdYmm5te3KNmadnJVlW2XLytgZ23H49mHNto4OHZnqOhUTPZMCje/+smUkXvgbwzp1MKrvgn7VqtybN4/Ynf8DwKxTR5KvXc9YSk2lwtrTAy0TU2J++x8p164DYOjigsPGDc+9TuI//3Crdx8A9CpVxHH7drQMDTP2/f03t/q4aY41rFcPh00bs7WRdv8+WubmaOnp5cm9CyGKp5f5XpLh7kIIIcRbZuTIkYwcOZILFy7g4+PDqFGjGDFiBP3798fDw4N69eoVdohFS3JcRoIOYOlAcpqaqISMYolvYnX3RymP+CYgY33vkXVHkqJOYeOljQRFBREUFYQKFR0cOjCk9hAqW1Yu8PhS793jwcpVAMQ/0UsOgLY2JSdMwHLgAJTkZO7Nm0f0lq1EfrdOc4hKVxeTFi0oOWniC69lWKsW5u/3Iu7PQ5RZtEiToAMY1q6N+fu9iPlxBwAWT/SiPymn4fZCCPE6JEkXQggh3lJ16tRh6dKlLFy4kJUrVzJhwgRWrVpFzZo1+eyzz7IVNXxrZRaNM7AAA3MiHmZU/9bT0cLcUPfZ5xVR3/3zHQ+THuJg5sDgmoPR1dJlYPWBbLuyjdiUWHo69cTR3LHQ4os/fgIA3XLlMKzrTOKZs6SGhaFtbU2ZJYs165erDAywmz4d40auRCxYgG758ph36Yxp27YvNfS89OzZWVaIeJLtmDHEHz6CSlcXs/aFs4KIEOLtI0m6EEII8ZZKTU3l559/xtfXl3379tGoUSM8PT0JCwvjiy++YP/+/Wx6XN36rfb0fPRH/62R/qY9xLj96DY/BP4AwLj649DVynjIYK5vzke1PyrM0DTijx8HwKxTJ2xHewGQ9uAB2ubmqHSzPxQx69Aesw6vl0A/6+9Rx8qKCrt3ZRTsfKKXXQgh8pMk6UIIIcRb5ty5c/j6+rJ582a0tbUZOHAgS5YsoeoTa0m3a9cuy1JqbzVNZffM5dcer5Fu+ubVxFh6bimp6ak0tGtYqIXglPR0lNRUtJ6qg6AoCvEnMnrSjRs31mzXKVGiQON7krZJwc7HF0IISdKFEEKIt8w777xD27ZtWbVqFd27d9dUxn9S9erV6du3byFEVwQ9vfzaG1jZXVEUjtw+wh+3/kCFis/rf/7SowDijx9Hu0QJDCpnn6eenpTEQ7/1gIK2lRU6JUqgU6IEeo6OaD+xlGJaZCTRP+4geutW0iIjsd+4EcOaNTT7k4OCUD94gMrQEMO6zq96u0II8UaTJF0IIYR4y9y4cQN7e/vnHmNsbIyvr28BRVTEZc5Jt8jak277BhSNu/LwCn/c+oM9t/YQ+igUgB5OPahiVeWl2on94w/ueI1G28oKp0N/onqqknnUhg3c9/bO8Vyd0nYYOFVGpafHo0OHIDVVs++hnx9lFn6teR/vnzHU3eid+lItXQjx1pIkXQghhHjLREREcPfuXRo2bJhl+6lTp9DW1qZ+/fqFFFkR9VRPeuYa6aWKcE96upLO/L/ms+nyfzUFDLQNaG3fmjEuY16qrdQ7dwifMhUA9cOHxB09imnr1pr9iqIQ8+uvABg3dkWlp09aZCRp9+6RFhFBWlg4cWHh/8VRuzYm7zbnwfIVPNqzh7QvJqNjaQn8Nx/9yaHuQgjxtpEkXQghhHjLfPLJJ4wfPz5bkn7nzh3mz5/PqVOnCimyIkhRshWOu/eoaA93T1fSmX1yNtuDtqNCRavyrejg0IHmZZtjpGv0Um0paWnc+Xw86Y8egbY2qNXE/LozS5KefOUKyVevodLVpYy3d5bK6uqYGJKDgkgKCkId+RCTVq00w9vjDhwkKTCQmF9/xdrdnfTkZBLOnAEkSRdCvN20CjsAIYQQQhSswMDAHNdCr1u3LoGBgYUQUREWFwFpiYAKzMsBcDcmI0kvisPd05V0ZpyYoUnQZzedjXdLbzo4dnjpBB3gweo1JJ47h5axMWW8lwAQ9+efqGNjNcfE/O9/AJi0aJFt6TNtc3OM3nkHqw8+wGbUyCzzzy369AEgett2FEUh8XwASlISOjY26Ds5vXSsQghRXEiSLoQQ4pX5+flhYWGheT99+nScnZ2fe467uzvdu3fP17jE8+nr63Pv3r1s28PDw9HRkUF2WcTczviveVnQyZgjHZFZ3b2I9aSr09VM9Z/KT1d/QkulxVfNvqJrxa7PPD7+5CnuL1tOelJSjvsTzp7lwcqVAJSaPh3TNm3Qd3JCSU0l9o8/gIwq7bG//Q6AWdcuLxWvWef3UBkZkXLjBolnzz4x1N31jVvaTggh8pIk6UIIUUC6dOlCmzZtctx34sQJVCoV586d02zbsWMHrVq1wtLSEiMjI6pUqYKHhwfnz5/Pcm5KSgpff/019erVw9jYGHNzc+rUqcOXX35JWFhYjtfbsWMH2trahISE5Li/atWqjBo16qXvcdy4cRw4cOClz3vSoUOHUKlUREdHv1Y74tnatm3LpEmTiImJ0WyLjo5m8uTJtG3bthAjK4LKusDkMHDPSETjk9N4lJwGFL0kfeWFlfx6/Ve0VdrMbzafzhU6P/PY9JQU7owdy4OVK7k7e3a2/WkPHnBn3OeQno55t66Yd+mMSqXCvFtG0h+7M6P3POGv06Tdu4eWmRkm7777UvFqm5hg/l4nAKK2bpP56EII8Zgk6UIIUUA8PT05ePAgwcHB2fb5+Pjg7OysGYI8YcIE3NzccHZ2ZufOnVy8eJFvv/2WihUrMnnyZM15ycnJtG3blq+++gp3d3eOHDnC2bNnWbBgAZGRkSxfvjzHWLp27Yq1tTXr16/Pts/f358rV67g6en50vdoYmKCtbX1S58nCtaiRYsIDQ3F3t6eli1b0rJlSxwdHbl79y6LFi0q7PCKHj1jzRrpEY8yetGN9bQx0S86ow7O3TvHd/98B8CsJrPo4Njhucc/2rMHdWQkADE/7tAUfoOM5dRuf/IpaeHh6NnbU/Jx0TgAs/feA5WKhDNnSL1zh5jfMpJ1s/btX6kae+aQ90d//EHSxYsAGLm6vnQ7QghRnEiSLoQQBaRz587Y2tri5+eXZXtCQgJbt27VJMUnT55kwYIFLF68mMWLF9OsWTMcHR159913+eKLL9i1a5fm3CVLlnDs2DEOHjzIqFGjcHFxoVKlSrRv355Vq1bx1Vdf5RiLrq4uAwcOxM/PD0VRsuzz8fHBxcWFOnXqsHjxYmrVqoWxsTHlypVjxIgRxMXFPfMenx7urlarGTNmDBYWFlhbWzN+/Phs13tZUVFRDBo0SDPCoGPHjly9elWzPzg4mC5dumBpaYmxsTE1atTQfGZRUVF88MEH2NjYYGhoiJOT01u5zFiZMmX4+++/WbBgAdWrV8fFxYWlS5fyzz//UK5cucIOr0jLXCPdtgj1oj9KecSko5NIV9LpWrErXSq+eNh51MaMqu96Dg4AhE+fQfK1ayjp6YRPnkzihQtomZtTdvUqtE2MNefp2tlh1KABANE7dvDojz0AmHV5dq/98xjUrIl+tWooqamgKOg7OaFra/tKbQkhRHEhSboQonhQFEiJL5xXLpNOHR0dBg0alC0x3r59OykpKXzwwQcAbN68GRMTE0aMGJFjO0/O1dy8eTNt27albt26Lzz2aZ6enty4cYPDhw9rtsXHx7Nt2zbNAwMtLS2WLVvGv//+y/r16zl48CDjx4/P1f1CRo+tj48P69at49ixYzx8+JCff/451+fnxN3dnTNnzrBz505OnDiBoih06tSJ1MdrL3/yySckJydz5MgR/vnnH+bPn4+JiQkAU6ZMITAwkN27d3Pp0iVWrVpFiRIlXiueN5WxsTFDhw7lm2++YeHChQwaNAhdXd3CDqvIe5SUMdTdzLDofFZzTs0hLD6MMiZlmNRg0guPTwoMJDEgAHR1Kb9+PcaNXVESE7nt5UXEokXE7toNOjqUXbYMfUfHbOebd80Y8h659jvS4+LQsbPD6BWX7VOpVFj26a15b9xYetGFEKLojNMSQojXkZoAX5UunGtPDssYDpsLHh4efP311xw6dIiWLVsCGT3XPXv2xPLxOsFBQUFUqFAhSwGvxYsXM3Xqf0NO79y5g7m5OUFBQbRo0SLLNXr06MG+ffsAqF27Nscfz/N8WvXq1WnYsCG+vr6aNrZt24ZaraZfv34AeHl5aY53dHRk1qxZDB8+nJWPi0m9iLe3N5MmTaJXr14ArF69mj179uTq3JxcvXqVnTt34u/vT+PH81Y3btxIuXLl+OWXX+jduzchISH06tWLWrVqAVChQgXN+SEhIdStW1ezDrjD417Et1VgYCAhISGkpKRk2d6167OLjb3tElIyknQjXe0Cv3ZSWhLz/ppHanoq9UvWx6WkC/88+Iffb/yOtkqbec3mYaJn8sJ2Hm7K6EU3a9sW3ZK2lF6wgBs9epBy7ToPr10HwG7GDIwbNsjxfNN2bbk7cyZKcsbQf/PO76HSevV+H7MuXbi34GuUxESZjy6EELxikh4aGopKpaJs2bIA/PXXX2zatInq1aszdOjQPA1QCCGKk6pVq9K4cWN8fHxo2bIl169f5+jRo+zduzfLcU/3gHt4eNC1a1dOnTrFgAEDsvTEP33sypUriY+PZ9myZRw5cuS58Xh6euLl5cWKFSswNTXVPDDIrNj+559/8tVXXxEYGEhsbCxpaWkkJSURHx+PsfHzH0zExMQQHh6O6xPzS3V0dKhfv/4rD3m/dOkSOjo6Wdb3tra2pkqVKly6dAmAUaNGMXz4cPbu3UubNm3o1asXtWvXBmD48OH06tWLc+fO0a5dO7p3765J9t8mN27coEePHvzzzz+oVCrN30fmz5JarS7M8Iq0xJSMz8ZIr+CT9B+DfmTH1R0A7Ly+M8u+obWH4mzr/MI21NHRxP7vNwAsB2SM3tEpUYIyixYR4j4Y0tOx/mgIFr16PrMNbVNTTFq15NHujArvZl1erqp7tvZMTCi9YD7Jly5h3KzZa7UlhBDFwSsl6f3792fo0KEMHDiQu3fv0rZtW2rUqMGGDRu4e/dult4eIYQoELpGGT3ahXXtl+Dp6cmnn37KN998g6+vL/b29rRu3Vqz38nJiWPHjpGamqoZfmxhYYGFhQW3b9/O0paTkxOXL1/Oss3Ozg4AKyurF8bSt29fRo8ezdatW2nRogXHjh1j5syZQMbc7k6dOjFs2DBmzZqFlZUVx44dw9PTUzO0vKA9K7lXFEWTYA4ZMoT27dvz+++/s3fvXubOncuiRYsYOXIkHTt2JDg4mN9//539+/fTunVrPvnkExYuXFiQt1HoPvvsMxwdHdm/fz8VKlTgr7/+IjIykrFjx751n8XLSnicpBsWcJKeqk7F92JG/YRW5VoRnRzNPw/+ITU9FWcbZ4bWHkrKrVuETZpMiU8+waRpkxzbif7pZ5TkZPSrVsXwiWkyxg0aUG7NGlJvh2Lh5vbCeCx6vc+j3X9gULMmBpUrv/b9mbVtC7KygBBCAK84J/3ff/+lweOiIdu2baNmzZocP36cTZs2ZSuIJIQQBUKlyhhyXhivl1zPt0+fPmhra7Np0ybWr1/P4MGDs/SG9+vXj7i4uFwNKe/Xrx/79u3LtixbbpmamtK7d298fX3x8fGhQoUKmqHvZ86cIS0tjUWLFtGoUSMqV678zCXdcmJubo6dnR0nT57UbEtLS+Ps2bOvFCtkDNFPS0vj1KlTmm2RkZEEBQVRrVo1zbZy5coxbNgwfvrpJ8aOHcvatWs1+2xsbHB3d2fDhg14e3vz7bffvnI8b6oTJ04wc+ZMbGxs0NLSQktLi6ZNmzJ37txXWnrvbZKYWjg96Tuv7yQiIQIbQxu+fvdr1ndcz/F+x9ny3hbWtF2DjpYO0Tt+IvH8eR6sXpVjG0p6OlFbtgBg2b9ftlE4Js2aYtmvX66Grps0bUJ5Pz/KfvPN69+cEEKILF6pJz01NRV9fX0A9u/fr5m7VrVqVcLDw/MuOiGEKIZMTExwc3Nj8uTJxMTE4O7unmW/q6srY8eOZezYsQQHB9OzZ0/KlStHeHg469atQ6VSofX4l+jRo0fz+++/06pVK6ZPn06zZs2wtLQkKCiI3bt3o6394kTC09OTZs2aERgYyLhx4zS/uFesWJG0tDSWL19Oly5d8Pf3Z/Xq1S91r5999hnz5s3DycmJatWqsXjx4lyvf/7PP/9gamqaZZuzszPdunXjo48+Ys2aNZiamjJx4kTKlClDt27dgIx59B07dqRy5cpERUVx8OBBTQI/depUXFxcqFGjBsnJyfz2229Zkvu3hVqt1hTTK1GiBGFhYVSpUgV7e3uuXLlSyNEVbZo56XoFV9YnLT2Ndf+uA+DDGh+ip52x1JmBjgE1StTQHJd87RoAiRf+Jj0hAS2jrKN84o8dIzUkBC1TU8w7v1o19icZN2r44oOEEEK8tFfqSa9RowarV6/m6NGj7Nu3jw4dMtbiDAsLk/VxhRAiFzw9PYmKiqJNmzaUL18+2/6FCxeyadMmzp8/T+fOnXFycqJ3796kp6dz4sQJzMzMADAwMODAgQNMnDgRX19fmjZtSrVq1fDy8qJJkyb88ssvL4yladOmVKlShdjYWD788EPNdmdnZxYvXsz8+fOpWbMmGzduZO7cuS91n2PHjmXQoEG4u7vj6uqKqakpPXr0yNW5zZs3p27dulleAL6+vri4uNC5c2dcXV1RFIVdu3Zppgao1Wo++eQTqlWrRocOHahSpYpmVIKenh6TJk2idu3aNG/eHG1tbbY87ll8m9SsWZO///4bgIYNG7JgwQL8/f2ZOXNmlkJ7IrvCGO6+L3gfoY9CsdC3oHfl3s88LjlzKcLUVBJyGLESvX07ABY9e2ZL4IUQQhQdKuUVqvccOnSIHj16aH6h8/HxAWDy5MlcvnyZn376Kc8DzSuxsbGYm5sTExOj+SVXCPFmSUpK4ubNmzg6OmJgUHTWKhbieZ73c1vQ30179uwhPj6enj17cuPGDTp37szly5extrZm69attGrVKt9jyG/59ZlO3PE3W06HMq5dZT5t5ZRn7T5LupLO+/97n6tRV/nE+ROG1RmW83EJCVyp56J5bzV4MCUnjM+yP8i1MUpyMo6//IxB1ar5HrsQQoj/vMz30iuN1WrRogUPHjwgNjZWs2QQwNChQzGSJ7NCCCFEkda+fXvNnytUqEBgYCAPHz7E0tIy2zxlkdV/PekFM9z9yO0jXI26irGuMf2q9nvmccnXb2R5H/9ELQiAuKPHUJKT0S1bFv0qVfIlViGEEHnjlYa7JyYmkpycrEnQg4OD8fb25sqVK9ja2uZpgEIIIYTIO2lpaejo6PDvv/9m2W5lZSUJei4kFOASbOlKOmv/zih66FbFDVO1LnfGfa5Z5/xJmfPR9Z0qZby/dIm0qCjN/kf79wNg2qaN/D0LIUQR90pJerdu3fj+++8BiI6OpmHDhixatIju3buzalXOFUWFEEIIUfh0dHSwt7eXtdBfUWJqZuG4/E3S1elqpvhP4e8Hf6Ovrc/A6gOJ2raN2N9+4/7CRShpaVmOT76WMR/dqEFD9B8viZbwuDddSUkh7tAhAEzbtsnXuIUQQry+V0rSz507R7NmzQD48ccfKVmyJMHBwXz//fcsW7YsTwMUQgghRN768ssvmTRpEg8fPizsUN448cmPh7vr5k+SrigKqanJfOH/BTuv70Rbpc3sJrOx1jHn4eMOkvSEBJKDgrKcl1k0Tt+pEsaujTJiPZGRpMf/dZr0R4/QLlECQ2fnfIlbCCFE3nmlJD0hIUGzLM7evXvp2bMnWlpaNGrUiODg4Fy3c+TIEbp06ULp0qVRqVS5qkJ8+PBhXFxcMDAwoEKFCi+9HJAQQgjxtlu2bBlHjx6ldOnSVKlShXr16mV5iWdL1Ax3z/s56YqicGvAAM61aszRf35DR6XDguYL6ODYgdg9e0kL+2+Z24Tz57OcqxnuXqkSRo0yk/QTADzavw8A01atUOViWUYhhBCF65W+YSpVqsQvv/xCjx492LNnD6NHjwYgIiLipSqoxsfHU6dOHQYPHkyvXr1eePzNmzfp1KkTH330ERs2bMDf358RI0ZgY2OTq/OFEEIIAd27dy/sEN5YCY+Hu+fHEmxJN2+QdPYcZsBHe7Uot3Qhre1boygKDx+vpKNdogTqBw9IPHcePvgAAHVcvCaB169UCXR0QVub1NBQUkJDiTtwEJCh7kII8aZ4pSR96tSp9O/fn9GjR9OqVStcXV2BjF71zHVsc6Njx4507Ngx18evXr2a8uXL4+3tDUC1atU4c+YMCxculCRdCCGEyKVp06YVdghvrMR8LBz36/Y51Hn854aX0ylzKRXsIeHUXyQFBqIyMKDU5EncGTOWhPPnNOelXM/oRdexsUHbwgIAw9q1STx/ngdr1pB2/z5aJiYYN2yY5zELIYTIe6803P39998nJCSEM2fOsGfPHs321q1bs2TJkjwL7mknTpygXbt2Wba1b9+eM2fOkJqamuM5ycnJxMbGZnkJIYQQQryK/Kru/kPgD8Q9LvSWZm0OwN0ZM0l7+JBI34xedIuePTFu/i5oaZEWFk7q3btA9srugGZeesyOnwAwefddVHp6eRqzEEKI/PFKSTpAqVKlqFu3LmFhYdy5cweABg0aULVq1TwL7ml3796lZMmSWbaVLFmStLQ0Hjx4kOM5c+fOxdzcXPMqV65cvsUnhBBCvAm0tLTQ1tZ+5kvkTFEUElMz10nPu89pX/A+vv5rAdVDFAAc5y9Cv3Jl1FFR3B45ivjDR0ClwurDQWibGGvWOU98PC89+WpGkq5X6ckk3TUzaECGugshxJvklZL09PR0Zs6cibm5Ofb29pQvXx4LCwtmzZpFenp6XseYxdNreyqPv3yetebnpEmTiImJ0bxCQ0PzNT4hhBCiqPv555/56aefNK+tW7cyceJE7Ozs+Pbbbws7vCIrKTU9M+fFOI8KxwVEBDDxyERKPVSwjAeVnh5G79THbu5XoK1N4tmzQMb65nr29gAYPZ5amHAuM0l/XNn9iSTdsE4dVIaGQEabJo9X5RFCCFH0vVKS/sUXX7BixQrmzZvH+fPnOXfuHF999RXLly9nypQpeR2jRqlSpbj7eGhXpoiICHR0dLC2ts7xHH19fczMzLK8hBBCFE2HDh1CpVIRHR0NgJ+fHxaP59g+y/Tp03GWZaVeSrdu3bK83n//febMmcOCBQvYuXNnYYdXZCWk/Lc2eV4swZaupDP52GRS0lPoGeuU0a6zM1r6+hjWqEGJj4dqjrXyGPzftR9X4Nf0pGuGuztpjlHp6WHk4gKAcZMmaBkbv3a8QgghCsYrJenr16/nu+++Y/jw4dSuXZs6deowYsQI1q5di5+fXx6H+B9XV1f27duXZdvevXupX78+urq6+XZdIYTIS6GhoXh6elK6dGn09PSwt7fns88+IzIyMtux165dY/DgwZQtWxZ9fX0cHR3p168fZ86cyXLcn3/+SadOnbC2tsbIyIjq1aszduxYzXSkzOS3Zs2aqNXqLOdaWFhk+X+3g4MDKpWKk4/nx2by8vKiRYsWOd7T2bNnUalUHDt2LMf97du3p2vXri/6aLJxc3Mj6Kn1oF/WrVu3UKlUBAQEvFY7b4OGDRuyf//+wg6jyMqcj26gq4WWVs4j+F7G2XtnCX0UiomuCR2jywNg1LCBZn+JYcMw69IFqw8/1PSeAxjVdQYg6dIlUu9FkHbvHpC1Jx3A6sMP0S1fHushnq8dqxBCiILzSkn6w4cPc5x7XrVqVR4+fJjrduLi4ggICND84nTz5k0CAgIICQkBMoaqDxo0SHP8sGHDCA4OZsyYMVy6dAkfHx/WrVvHuHHjXuU2hBCiwN24cYP69esTFBTE5s2buXbtGqtXr+bAgQO4urpm+X/omTNncHFxISgoiDVr1hAYGMjPP/9M1apVGTt2rOa4NWvW0KZNG0qVKsWOHTsIDAxk9erVxMTEsGjRoizXv379Ot9///0L4zQwMGDChAm5vi8XFxfq1KmDr69vtn2hoaHs378fT8+XTxQMDQ2xtbV96fPEy0tMTGT58uWULVu2sEMpsjLno+fVGum/XPsFgA4O7Uk+k1Gt3bjBf0m6Sk+PMl8voOSkiVnO0yldGp2SJUGtJubnnzO2lSqFtqlpluNMmjWl0t49mh51IYQQb4ZXStLr1KnDihUrsm1fsWIFtWvXznU7Z86coW7duppl28aMGUPdunWZOnUqAOHh4ZqEHcDR0ZFdu3Zx6NAhnJ2dmTVrFsuWLZPl14QQKIpCQmpCobwya2PkxieffIKenh579+7l3XffpXz58nTs2JH9+/dz584dvvjiC839uLu74+TkxNGjR3nvvfeoWLEizs7OTJs2jV9//RWA27dvM2rUKEaNGoWPjw8tWrTAwcGB5s2b891332n+f5pp5MiRTJs2jaSkpOfG+fHHH3Py5El27dqV63vz9PRk27ZtxMfHZ9nu5+eHjY0N7733Hhs2bKB+/fqYmppSqlQp+vfvT0RExDPbzGm4+7x58yhZsiSmpqZ4enq+8F5eJDk5mVGjRmFra4uBgQFNmzbl9OnTmv1RUVF88MEH2NjYYGhoiJOTk+ZhREpKCp9++il2dnYYGBjg4ODA3LlzXyuegmBpaYmVlZXmZWlpiampKT4+Pnz99deFHV6RFZ/8eI30PBjqHp8az77gjNGB3XXeQf3gASp9fQzq1HnBmRl1eAzrZfzuFL19O5C9F10IIcSb65UeBS9YsID33nuP/fv34+rqikql4vjx44SGhr7UL3QtWrR47i+3OQ2df/fddzl37lz2g4UQb7XEtEQabiqcNYBP9T+Fka7RC497+PAhe/bsYc6cORg+LuiUqVSpUnzwwQds3bqVlStXEhAQwMWLF9m0aRNaWtmfp2Ymrtu3byclJYXx48fneM2nE1wvLy82bNjAihUrnjsKycHBgWHDhjFp0iQ6dOiQYwxP++CDD/j888/Zvn077u7uQMbDBj8/Pz788EN0dHRISUlh1qxZVKlShYiICEaPHo27u3uuvzu2bdvGtGnT+Oabb2jWrBk//PADy5Yto0KFCrk6Pyfjx49nx44drF+/Hnt7exYsWED79u25du0aVlZWTJkyhcDAQHbv3k2JEiW4du0aiYmJACxbtoydO3eybds2ypcvT2ho6BtRoHTJkiVZCq5qaWlhY2NDw4YNsbS0LMTIira8XCN97629JKYl4mDmQLlrsdwDDOvWRSuXy6QZ1a3Lo91/kPp4SsuT89GFEEK82V4pSX/33XcJCgrim2++4fLlyyiKQs+ePRk6dCjTp0+nmVQQFUKIbK5evYqiKFSrVi3H/dWqVSMqKor79+9z9XG15hcta3n16lXMzMyws7PLVQxGRkZMmzaNyZMn89FHH2Fubv7MY7/88kt8fX3ZuHEjAwcOfGHbVlZWdO/eHV9fX02SfujQIW7cuIGHhweA5r8AFSpUYNmyZTRo0IC4uDhMTExeeA1vb288PDwYMmQIALNnz2b//v2v3JseHx/PqlWr8PPzo2PHjgCsXbuWffv2sW7dOj7//HNCQkKoW7cu9evXBzIeYGQKCQnBycmJpk2bolKpsH9cfbuoy/z7ES8nL9dI//V6xmiYbpW6keDzV0a7Dd7J9fmGdetleS896UIIUXy88qSq0qVLM2fOnCzbLly4wPr16/Hx8XntwIQQ4mUY6hhyqv+pQrt2XsgcWaSnp/fC5SWfPOdFxzzN09OTxYsXM3/+fL766qtnHmdjY8O4ceOYOnUqbm5uuW67Xbt2XLt2jUqVKuHj40OTJk2o8nhd5/PnzzN9+nQCAgJ4+PChZtnOkJAQqlev/sL2L126xLBhw7Jsc3V15c8//8xVfE+7fv06qampNGnSRLNNV1eXBg0acOnSJQCGDx9Or169OHfuHO3ataN79+40btwYyEh227ZtS5UqVejQoQOdO3emXbt2rxRLQfL19cXExITevXtn2b59+3YSEhL48MMPCymyoi0hj9ZID40N5ey9s2iptOjs2JnYv/yArPPRX8SgahVUhoYoj0d16DtJki6EEMXFK81JF0KIokalUmGka1Qor9wmyZUqVUKlUhEYGJjj/suXL2NjY4OFhQWVK1cG0CSKz1K5cmViYmIIDw/P9Welo6PD7NmzWbp0KWFhYc89dsyYMSQmJrJy5cpctd2mTRvs7e3x8/MjNjaWn376SVMwLj4+nnbt2mFiYsKGDRs4ffo0Pz8uepWSkpLr+PPSsx6GPPnwo2PHjgQHB+Pl5UVYWBitW7fWTBWoV68eN2/eZNasWSQmJtKnTx/ef//9gr2JVzBv3jxKlCiRbbutre1zH9y87RIfL8H2uoXjMnvRXe1csbwbhzoyEpWBAQYvUddHpauLYa1amvf6FSu+VkxCCCGKDknShRCigFhbW9O2bVtWrlypmdOc6e7du2zcuFEzDNnZ2Znq1auzaNEiTW/zkzLXEX///ffR09NjwYIFOV4z87in9e7dmxo1ajBjxoznxmxiYsKUKVOYM2cOsbGxz79BMpLdwYMHs379es18+j59+gAZDyEePHjAvHnzaNasGVWrVn1u0bicVKtWLdvScE+/fxmVKlVCT08vy9JxqampnDlzJsu0BBsbG9zd3dmwYQPe3t58++23mn1mZma4ubmxdu1atm7dyo4dO15qpZPCEBwcjKOjY7bt9vb2WQq2iqwyh7u/Tk96upLOzusZa9F3q9SN+L8yhrob1nXO9Xz0TJnF43RLl5Z10IUQohiRJF0IIQrQihUrSE5Opn379hw5coTQ0FD++OMP2rZtS+XKlTXV2FUqFb6+vgQFBdG8eXN27drFjRs3+Pvvv5kzZw7dunUDoFy5cixZsoSlS5fi6enJ4cOHCQ4Oxt/fn48//phZs2Y9M5Z58+bh4+OTrRr704YOHYq5uTmbN2/O1T0OHjyYsLAwJk+eTN++fTF+nDyUL18ePT09li9fzo0bN9i5c+dz48vJZ599ho+PDz4+PgQFBTFt2jQuXryYq3OvXLmiWfYz86Wrq8vw4cP5/PPP+eOPPwgMDOSjjz4iISFBMwJg6tSp/Prrr1y7do2LFy/y22+/aRL4JUuWsGXLFi5fvkxQUBDbt2+nVKlS2Qr2FTW2trb8/fff2bZfuHABa2vrQojozaCZk/4a1d1P3z1NeHw4prqmtCzXkoRTGUn6ywx1z2Tati3o6GDS4t1XjkcIIUTR81LjtXr27Pnc/c/qsRFCCJHBycmJ06dPM336dPr06UNERISm+OYPP/yAkdF/VeIbNGjAmTNnmDNnDh999BEPHjzAzs6Oxo0b4+3trTluxIgRVK5cmYULF9KjRw8SExNxcHCgc+fOjBkz5pmxtGrVilatWrF3797nxqyrq8usWbPo379/ru6xfPnytGnThr1792YpFGdjY4Ofnx+TJ09m2bJl1KtXj4ULF9K1a9dctQvg5ubG9evXmTBhAklJSfTq1Yvhw4ezZ8+eF57bt2/fbNtu3rzJvHnzSE9PZ+DAgTx69Ij69euzZ88eTZVzPT09Jk2axK1btzA0NKRZs2Zs2bIFyBhpMH/+fK5evYq2tjbvvPMOu3btylU1/MLUt29fRo0ahampKc2bNwfg8OHDfPbZZzl+TiJDZnV3Y/1XH+6+I2gHAB0dO6ITE0/c0aMAGDVs9NJtGdaoQeVjR6UXXQghihmV8hIL/A4ePDhXx2WuH1sUxcbGYm5uTkxMDGZmZoUdjhDiFSQlJXHz5k0cHR0xMDAo7HBe27Rp01i8eDF79+7F1dW1sMMR+eR5P7cF/d2UkpLCwIED2b59Ozo6GQlneno6gwYNYvXq1ei95LDroig/PtOZ/wvEx/8mw1tUZEKH56+8kJOgqCDe3/k+CgrbOm/Dat1vPPTxwaBmTRy2b3vpIpBCCCHeHC/zvfRSj4KLcvIthBBvqhkzZuDg4MCpU6do2LBhke+FFW8+PT09tm7dyuzZswkICMDQ0JBatWq9MUvIFZbE1MeF415xuPvSc0tRUGjv0B4nxYZrmzYBYDPyU0nQhRBCaLxeeVIhhBB5IrcjlYTIS05OTjg5ORV2GG+M+ORXLxx3+u5pjtw+go5Kh5F1RxL5zXcoSUkY1KmN8eMpB0IIIQRI4TghhBDirfP+++8zb968bNu//vrrbGuni/9oCse95BJsiqLgfdYbgF6Ve1E6yZCox3UNbD4dKb3oQgghspAkXQghhHjLHD58mPfeey/b9g4dOnDkyJFCiOjNoBnu/pI96QdDDvL3g78x1DFkWJ1hRK79DiU5GUNnZ4ybNsmPUIUQQrzBJEkXQggh3jJxcXE5FofT1dUlNja2ECJ6M7zKOulp6Wl4n/MGYGD1gZjHqoneuhUAm1HSiy6EECI7SdKFEEKIt0zNmjXZ+jhRfNKWLVuoXr16IUT0ZkjUDHfPfZK+8/pObsXewkLfgsE1BhO5Zg1KSgqG9V0wktUchBBC5EAKxwkhhBBvmSlTptCrVy+uX79Oq1atADhw4ACbNm3ixx9/LOToiq6EV0jSt1/ZDoBnTU/0wh4QtS3jvc3IUdKLLoQQIkeSpAshhBBvma5du/LLL7/w1Vdf8eOPP2JoaEidOnU4ePBggazT/qbSDHfXffavT4qikHrnDrplynA77jb/Rv6LlkqLzhU7EzFxNqSlYfxuc4wbNiiosIUQQrxhZLi7EEII8RZ677338Pf3Jz4+nmvXrtGzZ0+8vLxwcXEp7NCKrMSUFxeOi/npZ663aUv4xInsufEHAO+Uegejy6E82rMHtLSwHTu2QOIVQgjxZpIkXQghRJFx6NAhVCoV0dHRAPj5+WFhYfHcc6ZPn46zs3O+x1YcHTx4kAEDBlC6dGlWrFhBp06dOHPmTGGHVSQpikJC6uPh7vrPTtLjT5wAIObXnaSvXA9AR/sORHy9EADzHt0xqFw5n6MVQgjxJpMkXQghClhoaCienp6ULl0aPT097O3t+eyzz4iMjMx27LVr1xg8eDBly5ZFX18fR0dH+vXrly2R+vPPP+nUqRPW1tYYGRlRvXp1xo4dy507d4D/kt+aNWuiVquznGthYYGfn5/mvYODAyqVipMnT2Y5zsvLixYtWuR4T2fPnkWlUnHs2LEc97dv356uXbu+6KPJxs3NjaCgoJc+70m3bt1CpVIREBDwWu0UF7dv32b27NlUqFCBfv36YWlpSWpqKjt27GD27NnUrVu3sEMskpLT0lGUjD8/b5305KtXNX9udjiSzqehyQ09Es+dQ2VggM2oUfkdqhBCiDecJOlCCFGAbty4Qf369QkKCmLz5s1cu3aN1atXc+DAAVxdXXn48KHm2DNnzuDi4kJQUBBr1qwhMDCQn3/+mapVqzL2ieGya9asoU2bNpQqVYodO3YQGBjI6tWriYmJYdGiRVmuf/36db7//vsXxmlgYMCECRNyfV8uLi7UqVMHX1/fbPtCQ0PZv38/np6euW4vk6GhIba2ti99nshZp06dqF69OoGBgSxfvpywsDCWL19e2GG9ETLnowMY6ubck66kppJy4wYAd9rWAmDggTRiZy8AwMr9Q3RLlsznSIUQQrzpJEkXQhQLiqKQnpBQKC8ls3stFz755BP09PTYu3cv7777LuXLl6djx47s37+fO3fu8MUXX2jux93dHScnJ44ePcp7771HxYoVcXZ2Ztq0afz6669ARq/oqFGjGDVqFD4+PrRo0QIHBweaN2/Od999x9SpU7Ncf+TIkUybNo2kpKTnxvnxxx9z8uRJdu3alet78/T0ZNu2bcTHx2fZ7ufnh42NDe+99x4bNmygfv36mJqaUqpUKfr3709ERMQz28xpuPu8efMoWbIkpqameHp6vvBeXiQ5OZlRo0Zha2uLgYEBTZs25fTp05r9UVFRfPDBB9jY2GBoaIiTk5PmYURKSgqffvopdnZ2GBgY4ODgwNy5c18rnvy0d+9ehgwZwowZM3jvvffQ1s59lfLnWblyJY6OjhgYGODi4sLRo0efe/zGjRupU6cORkZG2NnZMXjw4GwjSXbs2EH16tXR19enevXq/Pzzz3kS66uKT86Yj66vo4W2Vs5V2VNCQlBSU1EZGfFNy2T+qKdCpYA6MhJtS0ushwwpyJCFEEK8oaS6uxCiWFASE7lSr3AKXlU5dxaVkdELj3v48CF79uxhzpw5GBoaZtlXqlQpPvjgA7Zu3crKlSsJCAjg4sWLbNq0CS2t7M9TMxPX7du3k5KSwvjx43O85tMJrpeXFxs2bGDFihWMGzfumbE6ODgwbNgwJk2aRIcOHXKM4WkffPABn3/+Odu3b8fd3R3IeNjg5+fHhx9+iI6ODikpKcyaNYsqVaoQERHB6NGjcXd3z/XDgG3btjFt2jS++eYbmjVrxg8//MCyZcuoUKFCrs7Pyfjx49mxYwfr16/H3t6eBQsW0L59e65du4aVlRVTpkwhMDCQ3bt3U6JECa5du0ZiYiIAy5YtY+fOnWzbto3y5csTGhpKaGjoK8eS344ePYqPjw/169enatWqDBw4EDc3t9dqc+vWrXh5ebFy5UqaNGnCmjVr6NixI4GBgZQvXz7b8ceOHWPQoEEsWbKELl26cOfOHYYNG8aQIUM0ifiJEydwc3Nj1qxZ9OjRg59//pk+ffpw7NgxGjZs+FrxvqrE1Bcvv5b8eGqG4liWa49uENJenx7WjUncdxCb0V5om5gUSKxCCCHebNKTLoQQBeTq1asoikK1atVy3F+tWjWioqK4f/8+Vx/Pa61ateoL2zQzM8POzi5XMRgZGTFt2jTmzp1LTEzMc4/98ssvuXnzJhs3bsxV21ZWVnTv3j3LkPdDhw5x48YNPDw8APDw8KBjx45UqFCBRo0asWzZMnbv3k1cXFyuruHt7Y2HhwdDhgyhSpUqzJ49m+rVq+fq3JzEx8ezatUqvv76azp27Ej16tVZu3YthoaGrFu3DoCQkBDq1q1L/fr1cXBwoE2bNnTp0kWzz8nJiaZNm2Jvb0/Tpk3p16/fK8eT31xdXVm7di3h4eF8/PHHbNmyhTJlypCens6+fft49OjRS7e5ePFiPD09GTJkCNWqVcPb25ty5cqxatWqHI8/efIkDg4OjBo1CkdHR5o2bcrHH3+cpc6Ct7c3bdu2ZdKkSVStWpVJkybRunVrvL29nxlHcnIysbGxWV556b810l88H/2ObUYi37hcU+yXLqfivr1Y9umTp/EIIYQovqQnXQhRLKgMDaly7myhXTsvZA6b19PT0/xZpcp5WO2T57zomKd5enqyePFi5s+fz1dfffXM42xsbBg3bhxTp07NdW+rp6cn7dq149q1a1SqVAkfHx+aNGlClSpVADh//jzTp08nICCAhw8fkp6eDmQku7lJti9dusSwYcOybHN1deXPP//MVXxPu379OqmpqTRp0kSzTVdXlwYNGnDp0iUAhg8fTq9evTh37hzt2rWje/fuNG7cGAB3d3fatm1LlSpV6NChA507d6Zdu3avFEtBMjIywsPDAw8PD65cucK6deuYN28eEydOpG3btuzcuTNX7aSkpHD27FkmTpyYZXu7du04fvx4juc0btyYL774gl27dtGxY0ciIiL48ccfee+99zTHnDhxgtGjR2c5r3379s9N0ufOncuMGTNyFferSHi8/Jrh83rSHyfpZ4wypnB0cOiASksLvXLl8i0uIYQQxY/0pAshigWVSoWWkVGhvHKbJFeqVAmVSkVgYGCO+y9fvoyNjQ0WFhZUfrxEU2ai+CyVK1cmJiaG8PDwXH9WOjo6zJ49m6VLlxIWFvbcY8eMGUNiYiIrV67MVdtt2rTB3t4ePz8/YmNj+emnnzQF4+Lj42nXrh0mJiZs2LCB06dPa4Y3p6Sk5Dr+vPSshyFPPvzo2LEjwcHBeHl5ERYWRuvWrTVTBerVq8fNmzeZNWsWiYmJ9OnTh/fff79gb+I1ValShQULFnD79m02b978Uuc+ePAAtVpNyaeKoZUsWZK7d+/meE7jxo3ZuHEjbm5u6OnpUapUKSwsLLIUsLt79+5LtQkwadIkYmJiNK+8nnaQmPLi4e5Jj4e7XzCLQl9bnxblWuRpDEIIId4OkqQLIUQBsba2pm3btqxcuVIzpznT3bt32bhxo2Yut7OzM9WrV2fRokWa3uYnZa4j/v7776Onp8eCBQtyvGbmcU/r3bs3NWrUeGHPo4mJCVOmTGHOnDm5Gj6sUqkYPHgw69ev18yn7/N4mO/ly5d58OAB8+bNo1mzZlStWvW5ReNyUq1atWxLwz39/mVUqlQJPT29LEvHpaamcubMmSzTEmxsbHB3d2fDhg14e3vz7bffavaZmZnh5ubG2rVr2bp1Kzt27MhSpf9Noa2tTffu3XPdi/6k5z3keFpgYCCjRo1i6tSpnD17lj/++IObN29mGyHxMm0C6OvrY2ZmluWVlzKHuz+rsnt6YiKpIRkPBkJtVDQv2xxjXeM8jUEIIcTbQYa7CyFEAVqxYgWNGzemffv2zJ49G0dHRy5evMjnn39O5cqVNdXYVSoVvr6+tGnThubNmzN58mSqVq1KXFwc//vf/9i7dy+HDx+mXLlyLFmyhE8//ZTY2FgGDRqEg4MDt2/f5vvvv8fExCTbMmyZ5s2bR/v27V8Y89ChQ1myZAmbN2/OVdGuwYMHM3PmTCZPnkzfvn0xNs5IVMqXL4+enh7Lly9n2LBh/Pvvv8yaNeslPj347LPP+PDDD6lfvz5NmzZl48aNXLx4MVeF465cuZJtW/Xq1Rk+fDiff/45VlZWlC9fngULFpCQkKAZATB16lRcXFyoUaMGycnJ/Pbbb5oEfsmSJdjZ2eHs7IyWlhbbt2/X9Ay/DUqUKIG2tna2Hu6IiIhsPeGZ5s6dS5MmTfj8888BqF27NsbGxjRr1ozZs2djZ2dHqVKlXqrNgvCinvTk6zdAUUgw0SXGWKF+yfoFGZ4QQohiRHrShRCiADk5OXH69GkqVKhAnz59sLe3p2PHjlSuXBl/f39Mnqj+3KBBA86cOUPFihX56KOPqFatGl27duXixYtZ5uaOGDGCvXv3cufOHXr06EHVqlUZMmQIZmZmz63g3qpVK1q1akVaWtpzY9bV1WXWrFm5XuqsfPnytGnThqioKE3BOMjojfbz82P79u1Ur16defPmsXDhwly1mcnNzY2pU6cyYcIEXFxcCA4OZvjw4bk6t2/fvtStWzfLKywsjHnz5tGrVy8GDhxIvXr1uHbtGnv27MHS0hLIqBEwadIkateuTfPmzdHW1mbLli1AxkiD+fPnU79+fd555x1u3brFrl27clUNvzjQ09PDxcWFffv2Zdm+b98+zbz9pyUkJGT7fDKXgsucfuDq6pqtzb179z6zzYKQOSf9WYXjMuejh9pk9PZXt371goZCCCHebirlZRb4LQZiY2MxNzcnJiYmz4fCCSEKRlJSEjdv3tSsy/ymmzZtGosXL2bv3r24uroWdjginzzv5/ZN/m7aunUrAwcOZPXq1bi6uvLtt9+ydu1aLl68iL29PZMmTeLOnTt8//33APj5+fHRRx+xbNky2rdvT3h4OF5eXmhpaXHq1CkAjh8/TvPmzZkzZw7dunXj119/5csvv3ypJdjy+jNdeegaC/64Qm+Xsnzdu062/ffmL+Chry+7XVT4tdPhZP+TGOm+eGlGIYQQb4eX+V6S4e5CCFHIZsyYgYODA6dOnaJhw4ZvTS+sKB7c3NyIjIxk5syZhIeHU7NmTXbt2oW9vT0A4eHhhISEaI53d3fn0aNHrFixgrFjx2JhYUGrVq2YP3++5pjGjRuzZcsWvvzyS6ZMmULFihXZunVroa2RDrkY7v64Jz3EVoWDuYMk6EIIIV6ZJOlCCFEEDB48uLBDEOKVjRgxghEjRuS4z8/PL9u2kSNHMnLkyOe2+f777xepSvnxyY8Lx71guHtICRXVrKrleIwQQgiRG9JdI4QQQgjxAompmXPSs/ekq2NiSLt3D4DbNjIfXQghxOuRJF0I8cZ6y0pqiDec/Ly+2RKeM9w9sxf9oYU2ifrSky6EEOL1SJIuhHjj6OrqAhlVooV4U2T+vGb+/Io3i2ad9Ock6Tet0wGoYlWl4AITQghR7MicdCHEG0dbWxsLCwsiIiIAMDIyQqVSFXJUQuRMURQSEhKIiIjAwsJCs9yYeLM8r3Dcf8uvQRmTMpjrmxdobEIIIYoXSdJfw/7g/Wy7so13Sr3DR7U/KuxwhHirlCpVCkCTqAtR1FlYWGh+bsWbJ3OddEPd7L86JQUFARBio5L56EIIIV6bJOmv4UHiA06En8BY17iwQxHiraNSqbCzs8PW1pbU1NTCDkeI59LV1ZUe9Dfcs+akK4pC8tVrAITaqOhqVbXAYxNCCFG8FHqSvnLlSr7++mvCw8OpUaMG3t7eNGvW7JnHb9y4kQULFnD16lXMzc3p0KEDCxcuxNraugCjzmBpYAnAw6SHBX5tIUQGbW1tSX6EEPkuMTXnJD0t4j7pMTGoteCONVI0TgghxGsr1MJxW7duxcvLiy+++ILz58/TrFkzOnbsSEhISI7HHzt2jEGDBuHp6cnFixfZvn07p0+fZsiQIQUceQZL/YwkPSo5qlCuL4QQQoiC8V9Pug6KWk1SUBDRO34iYv48AMItIU1HRTVrSdKFEEK8nkLtSV+8eDGenp6aJNvb25s9e/awatUq5s6dm+34kydP4uDgwKhRowBwdHTk448/ZsGCBQUad6bMnvTopOhCub4QQgghCkZm4TjD2CiuuXUm7f79LPuvlFVhY2hDCcMShRGeEEKIYqTQetJTUlI4e/Ys7dq1y7K9Xbt2HD9+PMdzGjduzO3bt9m1axeKonDv3j1+/PFH3nvvvWdeJzk5mdjY2CyvvKJJ0pOjUaer86xdIYQQQhQdiqIQ/7hwnM6lv0m7fx+Vnh5G77yDlYcHN8f1xKetlvSiCyGEyBOFlqQ/ePAAtVpNyZIls2wvWbIkd+/ezfGcxo0bs3HjRtzc3NDT06NUqVJYWFiwfPnyZ15n7ty5mJuba17lypXLs3uw0LcAQEEhJiUmz9oVQgghRNGRnJaOomT8WTsqEgCTli2x/+F7So7/nBPVtEjVVcl8dCGEEHmiUOekA9nWNlYU5ZnrHQcGBjJq1CimTp3K2bNn+eOPP7h58ybDhg17ZvuTJk0iJiZG8woNDc2z2HW0dDDTMwMgKknmpQshhBDFUeZ8dACtyAcA6NjaarZdfngZkKJxQggh8kahzUkvUaIE2tra2XrNIyIisvWuZ5o7dy5NmjTh888/B6B27doYGxvTrFkzZs+ejZ2dXbZz9PX10dfXz/sbeMzKwIrYlFhJ0oUQQohiKnONdD0dLdT3IwDQLZmRpKeoU7gWlbEEmwx3F0IIkRcKrSddT08PFxcX9u3bl2X7vn37aNy4cY7nJCQkoKWVNeTMpZeUzHFoBSxzXrpUeBdCCCGKp8Qn1khPu5eRpOs87lC4Gn2VNCUNc31z7IyzdxYIIYQQL6tQh7uPGTOG7777Dh8fHy5dusTo0aMJCQnRDF+fNGkSgwYN0hzfpUsXfvrpJ1atWsWNGzfw9/dn1KhRNGjQgNKlSxfKPWTOS5eedCGEEKJ40iy/pqtN2r17AOjYZPSkBz0MAqCqVdVnTtcTQgghXkahLsHm5uZGZGQkM2fOJDw8nJo1a7Jr1y7s7e0BCA8Pz7Jmuru7O48ePWLFihWMHTsWCwsLWrVqxfz58wvrFrAysALgYdLDQotBCCGEEPknM0k31NMmLSKzJz0jSb8ZexOAiuYVCyc4IYQQxU6hJukAI0aMYMSIETnu8/Pzy7Zt5MiRjBw5Mp+jyr0nl2ETQgghRPGTmJoxJ92SVNITEgDQfVw4LjgmGIDyZuULJzghhBDFTqFXd3/TWepnJOnSky6EEEIUT5k96bYpjwDQMjFBy9gYgODYjCTdwcyhUGITQghR/EiS/po0heNkTroQQghRLGUm6TZJMcB/y6+p09WEPspY2tXezL5wghNCCFHsSJL+mmS4uxBCCFG8ZVZ3t06KBf6bj3434S4p6SnoaulKZXchhBB5RpL015SZpMtwdyGEEKJ4in+8TrpFQkZPuq5txvJrmfPRy5mWQ1tLu3CCE0IIUexIkv6arPQzqrtHJUUV2lrtQgghhMg/mT3p5vEZU9syh7sHP8pI0mWouxBCiLwkSfprsjCwACA1PZX41PjCDUYIIYQQeS5zTrpJXDQAOiUf96THSpIuhBAi70mS/poMdQwx1DEEICpZiscJIYQQxU1mkm4UmzG1TcfWBoBbsbcASdKFEELkLUnS80DmMmxS4V0IIYQofhIfz0k3iMn4ntd93JMeEhsCSJIuhBAib0mSngdkGTYhhBCi+EpIUaNS0tHX9KTbkqpO5U7cHUCSdCGEEHlLkvQ8kDkvXSq8CyGEEMVPYqoa8+R4VGo1qFTolChBaFwo6Uo6hjqG2BjaFHaIQgghihFJ0vNAZoV3WStdCCGEKH4SUtRYJ2Usv6ZtbY1KV1cz1N3BzAGVSlWY4QkhhChmJEnPAzLcXQghhCi+MpL0WAB0M5dfe1zZvbxZ+UKLSwghRPEkSXoeyEzSZbi7EEIIUfwkpqRhnZjRk565/JpUdhdCCJFfJEnPA5rq7rIEmxBCCFHsxKeoKfF4uLvO4570J4e7CyGEEHlJkvQ8kNmTHp0UXbiBCCGEECLPJaaosXo83F2nZEaSntmTLsPdhRBC5DVJ0vOAlUFG4TgZ7i6EEEIUL4qikPDEcHddW1sSUhOISIgApCddCCFE3pMkPQ9oCsfJcHchhBCiWElOSyddQVM4TqdkSUIfhQJgoW+Bub55YYYnhBCiGJIkPQ9Y6FsAEJ8aT4o6pXCDEUIIIUSeSUxRA2iWYNOxtZWh7kIIIfKVJOl5wEzPDB2VDiDLsAkhhBDFSUKqGl11GuYpCUBGkp65/JoMdRdCCJEfJEnPAyqVCgsDC0CGvAshhBDFSWJKGlaPe9FVenpoW1hoknRZfk0IIUR+kCQ9j8ha6UIIIUTxY22sz4R3rIGMXnSVSqVJ0mW4uxBCiPwgSXoe0ayVLsPdhRBCiGLD0liPNjYZvy7plCwJIMPdhRBC5CtJ0vOIZq305OjCDUQIIYQQeSr13j0AdGxtiEmO0XzXlzeVnnQhhBB5T5L0PJLZky7D3YUQQojiJS3iPgC6tiU1vei2hrYY6RoVZlhCCCGKKUnS84iVgRUgw92FEEKI4iZN05Nuq/meL2FUojBDEkIIUYxJkp5HNNXdJUkXQgghipW0iAggY056ojoRAEMdw8IMSQghRDEmSXoeyZyTLkuwCSGEEMVLasR/c9ITUyVJF0IIkb8kSc8jVvoy3F0IIYQobhRF+W9OesmSJKZJki6EECJ/SZKeR2S4uxBCCFH8pD96hJKYkZjr2NqSpE4CJEkXQgiRfyRJzyOZheOik6NRp6sLORohhBBC5IXM+ehaZmZoGRpKT7oQQoh8p1PYAbzJ4o4cIWrbNoxdXTHv1wcABYXYlFjNHHUhhBBCvLm0zc2x/XwcSlrGA/iktIyedANtg8IMSwghRDEmSfprSLkVTNz+A6THx2P1wQeY6ZkRmxJLVFKUJOlCCCFEMaBjY4O1p6fmvaYnXVd60oUQQuQPGe7+GoybNgEg8cxZ0hMTNYn5w6SHhRmWEEIIIfJJZpIuPelCCCHyiyTpr0HP0RGd0nYoqakknD6Npb4swyaEEEIUZzInXQghRH6TJP01qFQqTJpk9KbH+/v/t1a6VHgXQgghiqXMOemSpAshhMgvkqS/JuPHSXqcv7+mwrsk6UIIIUTxJD3pQggh8psk6a/JuFEj0NIi5dp17JNMALgSdaWQoxJCCCFEfpCedCGEEPmt0JP0lStX4ujoiIGBAS4uLhw9evS5xycnJ/PFF19gb2+Pvr4+FStWxMfHp4CizU7bwgKDWjUBaBCiB8DxsOOkqlMLLSYhhBBC5A9N4TgdKRwnhBAifxRqkr5161a8vLz44osvOH/+PM2aNaNjx46EhIQ885w+ffpw4MAB1q1bx5UrV9i8eTNVq1YtwKizy5yXbn7hFtYG1sSnxnM24myhxiSEEEKIvJeklp50IYQQ+atQk/TFixfj6enJkCFDqFatGt7e3pQrV45Vq1blePwff/zB4cOH2bVrF23atMHBwYEGDRrQuHHjAo48K+OmTQFIOH6cd0s3A+Bw6OHCDEkIIYQQ+UB60oUQQuQ3ncK6cEpKCmfPnmXixIlZtrdr147jx4/neM7OnTupX78+CxYs4IcffsDY2JiuXbsya9YsDA1zfqKdnJxMcnKy5n1sbGze3cRjhrVqoWVigjomhtZJjvwEHL59mPHvjEelUuX59YQQQoiiZOXKlXz99deEh4dTo0YNvL29adasWY7Huru7s379+mzbq1evzsWLFwHw8/Nj8ODB2Y5JTEzEwKBwk2MpHCfeNmq1mtRUmcYpRG7o6emhpfX6/eCFlqQ/ePAAtVpNyZIls2wvWbIkd+/ezfGcGzducOzYMQwMDPj555958OABI0aM4OHDh8+clz537lxmzJiR5/E/SaWri1GjhsTtP0CloDh0zXQJfRTKrdhbOJo75uu1hRBCiMKUOXVt5cqVNGnShDVr1tCxY0cCAwMpX758tuOXLl3KvHnzNO/T0tKoU6cOvXv3znKcmZkZV65kLcRa2Am6oihSOE68NRRF4e7du0RHRxd2KEK8MbS0tHB0dERPT++12im0JD3T0z3NiqI8s/c5PT0dlUrFxo0bMTc3BzKGzL///vt88803OfamT5o0iTFjxmjex8bGUq5cuTy8gwwmTZsSt/8AqSdO886H73A87DhHbh+RJF0IIUSx9uTUNQBvb2/27NnDqlWrmDt3brbjzc3NNd/hAL/88gtRUVHZes5VKhWlSpXK3+BfUpI6CQUFkCRdFH+ZCbqtrS1GRkYyOlSIF0hPTycsLIzw8HDKly//Wv9mCi1JL1GiBNra2tl6zSMiIrL1rmeys7OjTJkyWb7cq1WrhqIo3L59Gycnp2zn6Ovro6+vn7fB5yBzvfSEgABajBnN8bDjHL59mA9rfJjv1xZCCCEKw6tMXXvaunXraNOmDfb29lm2x8XFYW9vj1qtxtnZmVmzZlG3bt1ntlMQ09sye9EBDLRlTroovtRqtSZBt7a2LuxwhHhj2NjYEBYWRlpaGrq6uq/cTqEVjtPT08PFxYV9+/Zl2b5v375nFoJr0qQJYWFhxMXFabYFBQWhpaVF2bJl8zXeF9ErVw7d8uUhLY1GYRnrpZ+7d47YlLz/JUEIIYQoCl5l6tqTwsPD2b17t6YXPlPVqlXx8/Nj586dbN68GQMDA5o0acLVq1ef2dbcuXM1vfTm5ub5Mmoucz66npYe2lraed6+EEVF5hx0IyOjQo5EiDdL5jB3tVr9Wu0UanX3MWPG8N133+Hj48OlS5cYPXo0ISEhDBs2DMgYqj5o0CDN8f3798fa2prBgwcTGBjIkSNH+Pzzz/Hw8Hhm4biCZPK4SI566XfUU8qjVtQcv5O7ngQhhBDiTfUyU9ee5Ofnh4WFBd27d8+yvVGjRgwYMIA6derQrFkztm3bRuXKlVm+fPkz25o0aRIxMTGaV2ho6Cvdy/Nk9qRLZXfxtpAh7kK8nLz6N1Ooc9Ld3NyIjIxk5syZhIeHU7NmTXbt2qUZ8hYeHp5lzXQTExP27dvHyJEjqV+/PtbW1vTp04fZs2cX1i1kYT10KHFHjpAaGsqodaaM661w+PZhOjh2KOzQhBBCiDz3KlPXMimKgo+PDwMHDnxhgR0tLS3eeeed5/akF8T0NqnsLoQQoiAUak86wIgRI7h16xbJycmcPXuW5s2ba/b5+flx6NChLMdXrVqVffv2kZCQQGhoKIsWLSoSvegAuiVtsf9+Pbrly2N0/xHTN6q59O9h1OmvN9xBCCGEKIpeZepapsOHD3Pt2jU8PT1feB1FUQgICMDOzu614n1dkqQLIQpb5ggkUbwVepJe3Oja2WkSddsYGO0Xxd+XDxd2WEIIIUS+eNmpa5nWrVtHw4YNqVmzZrZ9M2bMYM+ePdy4cYOAgAA8PT0JCAjQtFlYJEkXomjr0qULbdq0yXHfiRMnUKlUnDt3TrNtx44dtGrVCktLS4yMjKhSpQoeHh6cP38+y7kpKSl8/fXX1KtXD2NjY8zNzalTpw5ffvklYWFhOV7P3d0dlUr13NercHNzIygo6JXOzUmVKlXQ09Pjzp07edameH2SpOcD3VKlsP/he2JtjbGNgaA1Swo7JCGEECJfuLm54e3tzcyZM3F2dubIkSPPnboGEBMTw44dO57Zix4dHc3QoUOpVq0a7dq1486dOxw5coQGDRrk+/08T5Ja1kgXoijz9PTk4MGDBAcHZ9vn4+ODs7Mz9erVA2DChAm4ubnh7OzMzp07uXjxIt9++y0VK1Zk8uTJmvOSk5Np27YtX331Fe7u7hw5coSzZ8+yYMECIiMjn1krY+nSpYSHh2teAL6+vtm2ZUpJScnVPRoaGmJra5urY1/k2LFjJCUl0bt3b/z8/PKkzdeRWbBQAMpbJiYmRgGUmJiYfL/Wpe0+SmCVqsqhBlWVyw8u5fv1hBBCvJkK8rvpbZEfn+mv135VavrVVIbuHZpnbQpRFCUmJiqBgYFKYmJiYYfyUlJTU5WSJUsq06dPz7I9Pj5eMTU1VZYvX64oiqKcOHFCAZSlS5fm2E56errmz3PnzlW0tLSUc+fOvfDY5wGUn3/+WfP+3XffVT755BNl9OjRirW1tdK8eXNFURRl0aJFSs2aNRUjIyOlbNmyyvDhw5VHjx5pzvP19VXMzc0176dNm6bUqVNH+f777xV7e3vFzMxMcXNzU2JjY18Yk7u7uzJx4kRl9+7dSoUKFbLdS2hoqOLm5qZYWloqRkZGiouLi3Ly5EnN/l9//VVxcXFR9PX1FWtra6VHjx7PvF9FURRzc3PF19dXURRFuXnzpgIoW7duVd59911FX19f8fHxUR48eKD07dtXKVOmjGJoaKjUrFlT2bRpU5Z21Gq1Mm/ePKVixYqKnp6eUq5cOWX27NmKoihKy5YtlU8++STL8Q8ePFD09PSUAwcOvPAzeV3P+7fzMt9L0pOejyp37keKgQ62MfDTz3MLOxwhhBBCvIbM6u7Sky7eRoqikJCSVigvRVFyFaOOjg6DBg3Cz88vyznbt28nJSWFDz74AIDNmzdjYmLCiBEjcmznyaHomzdvpm3bttStW/eFx76s9evXo6Ojg7+/P2vWrAEyCmUuW7aMf//9l/Xr13Pw4EHGjx//3HauX7/OL7/8wm+//cZvv/3G4cOHmTdv3nPPefToEdu3b2fAgAG0bduW+Pj4LLXA4uLiePfddwkLC2Pnzp1cuHCB8ePHk56eDsDvv/9Oz549ee+99zh//jwHDhygfv36L/0ZTJgwgVGjRnHp0iXat29PUlISLi4u/Pbbb/z7778MHTqUgQMHcurUKc05kyZNYv78+UyZMoXAwEA2bdqkKVY6ZMgQNm3aRHJysub4jRs3Urp0aVq2bPnS8RWWQq3uXtxpGRhg0Ko56bsOYnDwNIE9AqluXb2wwxJCCCHEK8icky5LsIm3UWKqmupT9xTKtQNntsdIL3dpi4eHB19//TWHDh3SJGU+Pj707NkTS0tLAIKCgqhQoQI6Ov+1uXjxYqZOnap5f+fOHczNzQkKCqJFixZZrtGjRw9NwczatWtz/PirLblcqVIlFixYkGWbl5eX5s+Ojo7MmjWL4cOHs3Llyme2k56ejp+fH6ampgAMHDiQAwcOMGfOnGees2XLFpycnKhRowYAffv2Zd26dZrPbNOmTdy/f5/Tp09jZWWliTfTnDlz6Nu3LzNmzNBsq1OnTi7vPOv99uzZM8u2cePGaf48cuRI/vjjD7Zv307Dhg159OgRS5cuZcWKFXz44YcAVKxYkaZNmwLQq1cvRo4cya+//kqfPn2AjGkGmTUC3hTSk57PyvToC4DrZYWVZ5+9vqsQQgghijYpHCdE0Ve1alUaN26Mj48PkNHLfPToUTw8PLIc93TC5uHhQUBAAGvWrCE+Pj5LT/zTx65cuZKAgAA8PDxISEh45Vhz6nn+888/adu2LWXKlMHU1JRBgwYRGRlJfHz8M9txcHDQJOgAdnZ2REREPPfa69atY8CAAZr3AwYM4KeffiI6OhqAgIAA6tatq0nQnxYQEEDr1q2fe43cePozUKvVzJkzh9q1a2NtbY2JiQl79+7V1Da5dOkSycnJz7y2vr4+AwYM0Pz9BwQEcOHCBdzd3V871oIkPen5zLhRI7A0xzwqhmj/I1yoe4E6Ni//lEkIIYQQhUvTk64tPeni7WOoq03gzPaFdu2X4enpyaeffso333yDr68v9vb2WZI6Jycnjh07RmpqKrq6ugBYWFhgYWHB7du3s7Tl5OTE5cuXs2zLXA7yWQlsbhkbG2d5HxwcTKdOnRg2bBizZs3CysqKY8eO4enp+dyiapn3kEmlUmmGpeckMDCQU6dOcfr0aSZMmKDZrlar2bx5M8OHD3/hEtcv2q9SqbJNU8jpHp7+DBYtWsSSJUvw9vamVq1aGBsb4+XlpSmsl5ult4cMGYKzszO3b9/Gx8eH1q1ba4qZvimkJz2fqXR1sezYCYAmgQrfnP+mkCMSQgghxKuQnnTxNlOpVBjp6RTK62WHKffp0wdtbW02bdrE+vXrGTx4cJY2+vXrR1xc3HOHkD957L59+7Ity5Yfzpw5Q1paGosWLaJRo0ZUrlz5mUu8vY5169bRvHlzLly4QEBAgOY1fvx41q1bB2QM4w8ICODhw4c5tlG7dm0OHDjwzGvY2NhkqWB/9erVXI06OHr0KN26dWPAgAHUqVOHChUqcPXqVc1+JycnDA0Nn3vtWrVqUb9+fdauXcumTZuyjaJ4E0iSXgDMOncGoMEVhTMhxzly+0ghRySEEEKIl5VZOM5I16iQIxFCPI+JiQlubm5MnjyZsLCwbEOdXV1dGTt2LGPHjmXMmDEcO3aM4OBgTp48ybp161CpVGhpZaRJo0ePxtXVlVatWrF06VLOnTvHzZs32bNnD7t370Zb++V6+Z+nYsWKpKWlsXz5cm7cuMEPP/zA6tWr86x9yOjN/uGHH+jXrx81a9bM8hoyZAhnz57lwoUL9OvXj1KlStG9e3f8/f25ceMGO3bs4MSJEwBMmzaNzZs3M23aNC5dusQ///yTZX59q1atWLFiBefOnePMmTMMGzYsW49/TipVqsS+ffs4fvw4ly5d4uOPP+bu3bua/QYGBkyYMIHx48fz/fffc/36dc3f25OGDBnCvHnzUKvV9OjRI48+vYIjSXoBMHR2Rqe0HUYpUO+6whT/KdxPuE/SlSDi/P1RcrkuohBCCCEKjwx3F+LN4enpSVRUFG3atKF8+fLZ9i9cuJBNmzZx/vx5OnfujJOTE7179yY9PZ0TJ05gZmYGZCSFBw4cYOLEifj6+tK0aVOqVauGl5cXTZo04ZdffsmzmJ2dnVm8eDHz58+nZs2abNy4kblz83aFqJ07dxIZGZlj4urk5EStWrVYt24denp67N27F1tbWzp16kStWrWYN2+e5qFEixYt2L59Ozt37sTZ2ZlWrVplqcC+aNEiypUrR/Pmzenfvz/jxo3DyOjFDzinTJlCvXr1aN++PS1atNA8KHj6mLFjxzJ16lSqVauGm5tbtjn4/fr1Q0dHh/79+2Ng8Ob9P1ul5HZNg2IiNjYWc3NzYmJiNP/4CkLEwoVEfreOizVN2V47HvdzZjhcjgZA28ICsy5dsOjZA4Nq1QosJiGEEEVDYX03FWf58ZmOPDCSQ7cPMd11Or0q98qTNoUoipKSkrh58yaOjo5vZIIjRGhoKA4ODpw+fZp69eoV2HWf92/nZb6XpCe9gGQOea/x7yOmb0rH4XI0ipYKbWtr1NHRRP3wAzd79CR40IeoY2MLOVohhBBCPE2WYBNCiKItNTWVkJAQJkyYQKNGjQo0Qc9LkqQXEP0qVdB3cgIgXU+HPfVUfDZMl4Qfl1Hu2zWYduiASleXhL/+IvSjoajjnr3MghBCCCEKXqJaCscJIURR5u/vj729PWfPns3z+fwFSZL0AqJSqSiz1JuSU6dQ+cBBwj7uzF3zdCb4TyK1QS3Kei/B4cftaJmbk3jhAreHDyc9MbGwwxZCCCHEY9KTLoQQRVuLFi1QFIUrV65Qq1atwg7nlUmSXoD0K1TAqn9/dG1smOI6hbImZQmLD+OzPz8jWZ2MQZUqlP/uO7RMTEg4fZrbn44kXYrKCSGEEEWCprq7jlR3F0IIkX8kSS8kpnqmLG+1HFNdU85HnOeLY1+QrqRjWKsm5b5dg8rQkHh/f8LGjuNVavulx8cTu2cvSlpaPkQvhBBCvH2kJ10IIURBkCS9EFWyrIR3S290tHTYc2sP3ue8ATCqV49yq1ai0tPj0b59RG3Y+NJth0+fwZ3PPuPBmjV5HLUQQgjxdsrsSZc56UIIIfKTJOmFrIFdA2Y2ngmA77++bL28FQDjRo2w/fxzACK+/pqkK0G5bjPl9h1id+0CIHrLVpTU1DyOWgghhHi7KIqi6UmXJF0IIUR+kiS9COhSsQufOH8CwJxTc1h/cT2KomA54AOM322OkpJC2LhxpCcl5aq9h9+vB7UagLT793l08M98i10IIYR4G6Slp6FWMr5bZbi7EEKI/CRJehHxce2PGVBtAAoKC88sZN5f80hX0ik9Zw7a1tYkX71KxMJFAKijo4n+6WfCJkwgzt8/SzvqmBiif9wBgOHjdQGjt24p2JsRQgghipmEtATNn6UnXQghRH6SJL2IUKlUjH9nPOPqjwNg0+VNjD40mlQLY0p/NQeAqA0buDVgAEFNmxE+eTIxv+7k9qcjSbp8WdNO1JatKAkJ6FetSukFC0ClIv74CVJu3cpyvaSgIGL+9z+Uxz3uQgghhHi2zPnoOioddLV0CzkaIcTbws/PDwsLC8376dOn4+zs/Nxz3N3d6d69e77GJfKXJOlFiEql4sMaH7Lw3YXoaenxZ+iffLj7QyKcy2E5YAAAiWfOQloa+lWqYFC9OkpiIrc/+ZS0qCjSU1J4uOEHAKw9BqNXtgwmzZsDELV1m+Y6SVeuENy3H2Gfj+f2qM9kPXYhhBDiBWQ+uhBFX5cuXWjTpk2O+06cOIFKpeLcuXOabTt27KBVq1ZYWlpiZGRElSpV8PDw4Pz581nOTUlJ4euvv6ZevXoYGxtjbm5OnTp1+PLLLwkLC8vxejt27EBbW5uQkJAc91etWpVRo0a99D2OGzeOAwcOvPR5z1KlShX09PS4c+dOnrUpXp8k6UVQe4f2fNf+Oyz0Lbj08BJu/3PjUNfyWH/8MTZjxlDxj91U+PUXyvusQ7dcOVLv3OHOmDHE/PwL6vsP0ClVCrOOHQGw6OsGQMxPP5GelETa/fuEDhtOekLGsL24AwcIdncn7eHDQrtfIYQQoqiT5deEKPo8PT05ePAgwcHB2fb5+Pjg7OxMvcfTQSdMmICbmxvOzs7s3LmTixcv8u2331KxYkUmT56sOS85OZm2bdvy1Vdf4e7uzpEjRzh79iwLFiwgMjKS5cuX5xhL165dsba2Zv369dn2+fv7c+XKFTw9PV/6Hk1MTLC2tn7p83Jy7NgxkpKS6N27N35+fnnS5utIlWLXGpKkF1F1bevyY5cfcbVzJUmdxJzzC5ha8zLqD7qi5+AAgLaFBWW/WYHKyIiEEye5O2sWAFaDBqHSzRiKZ9K8OTql7VDHxBDzy6+EfvIpaeHh6Dk4UHbVSrTNzUm68De3+vbLNiReCCGEEBmS1LL8mnjLKQqkxBfOS1FyFWLnzp2xtbXNlnAmJCSwdetWTVJ88uRJFixYwOLFi1m8eDHNmjXD0dGRd999ly+++IJdj1dJAliyZAnHjh3j4MGDjBo1ChcXFypVqkT79u1ZtWoVX331VY6x6OrqMnDgQPz8/FCeit/HxwcXFxfq1KnD4sWLqVWrFsbGxpQrV44RI0YQFxf3zHt8eri7Wq1mzJgxWFhYYG1tzfjx47Nd71nWrVtH//79GThwID4+PtnOu337Nn379sXKygpjY2Pq16/PqVOnNPt37txJ/fr1MTAwoESJEvTs2VOzT6VS8csvv2Rpz8LCQvN3c+vWLVQqFdu2baNFixYYGBiwYcMGIiMj6devH2XLlsXIyIhatWqxefPmLO2kp6czf/58KlWqhL6+PuXLl2fOnIzpwa1ateLTTz/NcnxkZCT6+vocPHgwV59LUaBT2AGIZytpXJLVbVez+fJmlpxdgn+YP51+6kT3St1xr+lOOdNyGFSuTOl5c7kz6jNIS0PLxASLPr01bai0tbHs04f73ku5O3MmpKejbW5OudWr0HNwQG/zZkKHDiU1JIRbfftR7ts1GNauXYh3LYQQQhQ9ianSky7ecqkJ8FXpwrn25DDQM37hYTo6OgwaNAg/Pz+mTp2KSqUCYPv27aSkpPDBBx8AsHnzZkxMTBgxYkSO7WSel3ls27ZtqVu37guPfZqnpyeLFy/m8OHDtGjRAoD4+Hi2bdvGggULANDS0mLZsmU4ODhw8+ZNRowYwfjx41m5cuUL7xdg0aJF+Pj4sG7dOqpXr86iRYv4+eefadWq1XPPe/ToEdu3b+fUqVNUrVqV+Ph4Dh06RMuWLQGIi4vj3XffpUyZMuzcuZNSpUpx7tw50tPTAfj999/p2bMnX3zxBT/88AMpKSn8/vvvuYr5SRMmTGDRokX4+vqir69PUlISLi4uTJgwATMzM37//XcGDhxIhQoVaNiwIQCTJk1i7dq1LFmyhKZNmxIeHs7lxzW6hgwZwqeffsqiRYvQ19cHYOPGjZQuXVpzb28C6Ukv4rRUWnxQ7QO2dt5KXdu6pKSnsC1oG11+7sLEoxO5FXMLs3btKPFJxhJuVh9+iLaJSZY2LHr1Ah0dSE8HXV3KLF+m6Y3Xr+CIw5bNGNSsiTo6mmD3wdkqxgshhBBvu0S1zEkX4k3g4eHBrVu3OHTokGabj48PPXv2xNLSEoCgoCAqVKiAjs5//ZWLFy/GxMRE84qJidEcW6VKlSzX6NGjh+a4xo0bPzOW6tWr07BhQ3x9fTXbtm3bhlqtpl+/fgB4eXnRsmVLHB0dadWqFbNmzWLbtm3PajIbb29vJk2aRK9evahWrRqrV6/G3Nz8hedt2bIFJycnatSogba2Nn379mXdunWa/Zs2beL+/fv88ssvNG3alEqVKtGnTx9cXV0BmDNnDn379mXGjBlUq1aNOnXqZJkmkFteXl707NkTR0dHSpcuTZkyZRg3bhzOzs5UqFCBkSNH0r59e7Zv3w5kPFxYunQpCxYs4MMPP6RixYo0bdqUIUOGANCrVy9UKhW//vqr5hq+vr64u7s/94FKUSM96W+IihYVWd9hPWfunWHdP+vwD/Pn9xu/s+fWHgbXGMxHwz/ColdPdEqVynaujo0N5t27EfPTz9jNmIFxgwZZ95cogf16P26PHEX88eOEDhtOmfnzMOvUqaBuTwghhCjSZE66eOvpGmX0aBfWtXOpatWqNG7cGB8fH1q2bMn169c5evQoe/fuzXLc0wmbh4cHXbt25dSpUwwYMCDL0O+nj125ciXx8fEsW7aMI0eOPDceT09PvLy8WLFiBaamppoHBpkV2//880+++uorAgMDiY2NJS0tjaSkJOLj4zE2fv7ogZiYGMLDwzWJM2SMJqhfv/4Lh7yvW7eOAY8LUwMMGDCA5s2bEx0djYWFBQEBAdStWxcrK6sczw8ICOCjjz567jVyo379+lneq9Vq5s2bx9atW7lz5w7JyckkJydrPotLly6RnJxM69atc2xPX1+fAQMG4OPjQ58+fQgICODChQvZht4XdZKkv0FUKhXvlHqHd0q9w8XIiyw/vxz/O/6s/Wctv934jQkNJtBKZZfjuXbTp2M7ejQ6zyg0oWVsTNnVqwibMIFHu//gzthxJJw5i45dKbSMjdE2NkYdH09aRARp9++jjo7B5N3mWPTu/UY9lRJCCCFeReYSbNKTLt5aKlWuhpwXBZ6ennz66ad88803+Pr6Ym9vnyWpc3Jy4tixY6SmpqL7uI6ThYUFFhYW3L59O0tbTk5OmqHUmezsMn7fflYC+6S+ffsyevRotm7dSosWLTh27BgzZ84EIDg4mE6dOjFs2DBmzZqFlZUVx44dw9PTM1+LqAUGBnLq1ClOnz7NhAkTNNvVajWbN29m+PDhGBo+//91L9qvUqmyPSjI6Z6efhCxaNEilixZgre3t2auvpeXFykpKbm6LmQMeXd2dub27dv4+PjQunVr7O3tX3heUSLD3d9QNaxrsKr1KrxbemNnbEd4fDhef3rRbGszhu0fxjcB33DszjHU6RnroKt0dJ6ZoGfS0tOjzMKFWPbvB4pC1KZN3F+0mHszZxE2YSL3Zs4icvUaYnb8RNyBA9ydOo3QYcNIe/BA04aSkkLMr78S9sUXGeuwS5VGIYQQxYAswSbEm6NPnz5oa2uzadMm1q9fz+DBg7N0KvXr14+4uLhczfvu168f+/bty7YsW26ZmprSu3dvfH198fHxoUKFCpr56WfOnCEtLY1FixbRqFEjKleu/Mwl3XJibm6OnZ0dJ0+e1GxLS0vj7Nmzzz1v3bp1NG/enAsXLhAQEKB5jR8/XjPkvXbt2gQEBPDwGStA1a5d+7lLwdnY2BAeHq55f/XqVRIery71PEePHqVbt24MGDCAOnXqUKFCBa5evarZ7+TkhKGh4XOvXatWLerXr8/atWvZtGkTHh4eL7xuUSM96W8wlUpF6/KtcbVz5bt/vuOHwB+ISY7B/44//ncy5pU72zgzu+ls7M1y9/RIpa1NySlTMHR2JjHgAukJCaTHx5MeH4+WsRE6Njbo2NqipKQQ+d064g8f4UbXbpSa8iUpwSFEbdxI2v37AMTs+ImIhYuwHPABln36oP2C+TFKSgopwcEkX79OWkQE6thHpD96hDo+DuOGjTDv0vn1PjAhhBDiFUlPuhBvDhMTE9zc3Jg8eTIxMTG4u7tn2e/q6srYsWMZO3YswcHB9OzZk3LlyhEeHs66detQqVRoaWX0ZY4ePZrff/+dVq1aMX36dJo1a4alpSVBQUHs3r0bbW3tF8bj6elJs2bNCAwMZNy4cZoHBhUrViQtLY3ly5fTpUsX/P39Wb169Uvd62effca8efNwcnKiWrVqLF68mOjo6Gcen5qayg8//MDMmTOpWbNmln1DhgxhwYIFXLhwgX79+vHVV1/RvXt35s6di52dHefPn6d06dK4uroybdo0WrduTcWKFenbty9paWns3r2b8ePHAxlV1lesWEGjRo1IT09nwoQJmlELz1OpUiV27NjB8ePHsbS0ZPHixdy9e5dq1aoBYGBgwIQJExg/fjx6eno0adKE+/fvc/HixSxL2mUWkDMyMqJHjx4v9ZkWBZKkFwNGukaMqjeK4XWGExQVxD8P/uGfB/9wIOQAAfcDeH/n+3i5eNGvaj+0VC8ePKFSqTDv2hXzrl2fe5xphw6Ejfuc5KAg7oweo9muY2ODSatWPDpwgLR797i/aDEPVnyDYe3aGDo7Y1i3LrplypBy8wbJQVdJvnqV5GvXSAkJAbU6x2vF/LiD5GvXsPH6TIbXCyGEKHDSky7Em8XT05N169bRrl07ypcvn23/woULadCgAatWrcLHx4eEhARKlixJ8+bNOXHiBGZmZkBGUnjgwAG8vb3x9fVl0qRJpKen4+joSMeOHRk9evQLY2natClVqlTh6tWrfPjhh5rtzs7OLF68mPnz5zNp0iSaN2/O3LlzGTRoUK7vc+zYsYSHh+Pu7o6WlhYeHh706NFDU/juaTt37iQyMjLHxNXJyYlatWqxbt06li1bxt69exk7diydOnUiLS2N6tWr88033wDQokULtm/fzqxZs5g3bx5mZmY0b95c09aiRYsYPHgwzZs3p3Tp0ixduvSFPfwAU6ZM4ebNm7Rv3x4jIyOGDh1K9+7ds9zPlClT0NHRYerUqYSFhWFnZ8ewYcOytNOvXz+8vLzo378/BgZvXi0RlZLbhfSKidjYWMzNzYmJidH84yuuwuLCmHp8KqfCM9YzfKfUO0xsMJHKlpXz7BrpycncX7yEh99/j37Vqli7f4hZx46o9PRIT0kh9rffebh+PclXruSqPS0TE/QrVkS3TGm0TM3QNjVBHfuI6MdVLi3796Pkl1+i0iqYmRrJN26i0tNDr2yZArmeEOLt9DZ9NxWUvP5M5/81nw2XNuBZ0xMvF6/XD1CIIiwpKYmbN2/i6Oj4RiY4QoSGhuLg4MDp06epV69egV33ef92XuZ7SXrSi7HSJqX5tu23bLuyjcVnF3P67mne3/k+nSt0ZoTzCMqaln3ta2jp61Ny0kRsPhuFytAwSy+3lp4eFj17YN6jOynXr5MYEEDC+fMkBlwgLSICvQqO6Ds5YeDkhF6lSuhXqoSOrW2OPeUG1atxd8ZMojZtRv0oDqtBA0k4c5aEM2dIungR3ZIlMahTG8M6dTCsXRvd0qVRPTH8SB0XR7z/ceIOHSItIgLz7t0yHibo5PxPQElN5f7KlUSu+Ra0tbEZORJrT48sbQohhHh7SE+6EEIUfampqYSHhzNx4kQaNWpUoAl6XpIkvZjTUmnRt2pfmpRugvc5b/YG7+V/N/7H7lu76VaxG66lXaljU4dSxtmXbnup6xg9e2kMlUqF/uMk3OL991+pfcu+fdEyMSVs4kRi//c/Yv/3vyz70+7eJfHCBaL4IWODtjY6trboliyJSkeHhAsX4IkidvH+/txfthzrj4Zg3r07Wnp6mn0poaHcGTeOpAt/Z2xIT+f+4sXE/fknpefPQ7dcOZKDrhJ3+DAJZ06jW7o0Ji1aYNyoEVrytFkIIYqlJHXGnHRZgk0IIYouf39/WrZsSeXKlfnxxx8LO5xXJkn6W6KcWTkWtVjExQcXWXpuKSfCT7Dj6g52XN0BgK2hLY1KN8K9hjtOlk6FHG3OzDu/h5aJMXfGjEWlrY2RiwtG79THsHZtUu/eJfHC3yReuEDS5cuQmkpaeDhpT1SV1HNwwKRFC7RMTIjauJHU0FDuTp1GxIKv0S1TBp1SJdGxLsGjPXsyCuWZmmI3YzrpySncmz2bxPPnudG9B9oW5qSFhWeJLXrLVlQGBhi98w5ahoYZBfcSElDp6mLz6ScYvfNOQX9cQggh8lBiqvSkCyFEUdeiRYsXrhH/JpAk/S1To0QNvm33LX+F/8Xe4L38ff9vgqKCiEiMYOf1ney8vpM25dswtPZQqllXK+xwszFt0YLKJ46j0tHJNvTcvEsXABS1mrQHkaTdu0tq+F3S4+IwcqmHnoOD5lhrj8FEb99O5Dof0iIiSL5yJcu8eUMXF8osmI9umYy56MYN3iFs8hcknDpFWkICKn19jBo1xKRxY1KCg3n05yHSwsOJP3o0W8wh//xD+e++w6he3Xz4RDIoaWkk/PUXhnXqoGX8ZqxhKoQQb5LMnnRJ0oUQQuS3Qi8ct3LlSr7++mvCw8OpUaMG3t7eNGvW7IXn+fv78+6771KzZk0CAgJyfT0pzpNdQmoC/z74ly1XtrAveJ9meznTciiKQrqSTjrplDctT5MyTWhSugmVLSsXiyrrSmoqyTdukhZxj7R790i9ew9du1KYd++e7SGAkp5O/LFjoCgYNWiAluF/v6gpikJyUBAJp8+AlgotIyO0jIyI3rqNeH9/tExMKO/nh2HNGppzkq4EkRJ8C6N33kHH0vKV7yHtwQPujBmbkaTXq4f9hh8KrLCeECJvyHdT3svrz3TQ7kGcjzjPkhZLaGPfJg8iFKLoksJxQryaYlE4buvWrXh5ebFy5UqaNGnCmjVr6NixI4GBgTkulZApJiaGQYMG0bp1a+7du1eAERdPRrpGNLBrQAO7BlyLusa3/3zLnlt7CH0UmuW4u/F3+evuXyw5uwRbQ1ta27emW8VuVLeu/sYm7CpdXQyqVIYqL654r9LSwuSJpSWy7FOpMKhSBYMqVbJsN2nWjNCPhpJw5gyhnp6U9/MlJTiYhxs2kHjm8TIU2toYN2yAabv26FepTGpoKCm3bpFyKxglNQVtK2u0rSzRsbJGv1JFDOvW1cx9Tzh3jjufeWnWpk88d47obdux7Ov2Gp+KEEKIp0nhOCGEEAWlUHvSGzZsSL169Vi1apVmW7Vq1ejevTtz58595nl9+/bFyckJbW1tfvnlF+lJzwfhceGEx4ejpdLSrK3+74N/8Q/z56/wvzTD/gAqmlekS8Uu1CtZD0czRywMLAop6qJJHRdPiIcHSX//nXWHjg56ZcuScuvWS7Wn0tPDsF499BwdiN7+I6SloVexIibNm/PQ1xctU1Mq7vodHRubHM9X0tNJ+Osvkq9dx7BWTQxq1HhmlXsAdUwMD1avIS0iAv1KFdGrWBH9SpXQc3DIVY99enIySX//jUHt2mjp67/Uvarj4on59Re0jI0x79btjX0YJMSLyHdT3svrz7TLz124FXsLvw5+uJR0yYMIhSi6pCddiFfzxvekp6SkcPbsWSZOnJhle7t27Th+/Pgzz/P19eX69ets2LCB2bNnv/A6ycnJJCcna97Hxsa+etBvETsTO+xM7LJsq21Tm/7V+pOsTuav8L/4343/cTDkINdjruN9zltznKW+JY7mjlS3rk6tErWoVaIWZU3LvrUJlraJMeXXfkvwh+4kX76MtrU1lm59sHBzQ7dkSVKCg4ndu5dHe/dlLE1nb5/xcnBAZWiA+mEU6oeRpN2/T+KFv0mLiCDh5EkSTp4EwKxTR+xmzUJlYEDCX3+RdPEi9+bOo8ziRVniSLl9h5hffiHm559JvXNHs13L2BjD+i6YNG6MaceO6NraavbFHTlC+JdTSIuIyHZfBjVqUHbZUs28/ZwkXrhA2KTJpNy4gY6NDVYeHli69XnuagAA6uhoHm7YyMMffiA9JiYj/pu3sPH67IU/R2mRkWhbWb21P29CiPyRkJYASE+6EEKI/FdoSfqDBw9Qq9WULFkyy/aSJUty9+7dHM+5evUqEydO5OjRo+g8p+BKe/gAADj9SURBVOfvSXPnzmXGjBmvHa/4j762Ps3KNqNZ2WY8SnnEvuB97Avex/Xo64THhxOVHEVURBTnIs5pzrHUt6ReyXrUL1mf+qXqU9mysqaH/m2gbW6Ow8YNJP7zD4b16mVZ8k3P3p4SH31EiY8+emE7iqKQcvMm8SdOkPTPvxi9Ux/znj01CWmpmTO41bsPsbt2Yd6jOybNmpF44QIPvl1L3IEDmna0TEwwrF2bxIsXSY+JIf7wEeIPH+He/AUYu7pi3rULCWfOEr19e0aMjo6Yd+tKys1bJN+4QfLVqyRdvMjN93tTxtsb44YNssSZnpzMgxUriFznA+npoFKRdv8+EfPnE7lmDZYDB2Dapg36Tk6a3nhFrSbx3Dli/9hDzM8/k56Q8Quxjp0daeHhRK5Zg5KWiu24cc9MwOOOHiXs8/FYD/sYa3f33P8FCSHECySlyRJsQgghCkahV3d/+pdtRVFy/AVcrVbTv39/ZsyYQeXKL54/nGnSpEmMGTNG8z42NpZy5cq9esAiC1M9U3o69aSnU08gowhdcGww16Kv8e+Df/nnwT9cfniZqOQoDoQc4EBIRqJoqGNISaOS2BrZYmtkS3nT8tQvVZ/aNrXR1365IdFvCi1jY4wbNXqtNlQqFfoVKqBfoUKO+w1r1MBq4EAerl/P3WnT0S1XjoRTpzT7jVwbYdGzJ6Zt2qBlaIiiVpN0+TIJp/7i0d69JAYEEO/vT7y/f+YFsRo0CJvRXlnWgE8NCyP0009JDrxEiIcHJSdNwuTd5iQFXiLpUiCP9u4j5cYNAMy6dqHk55/z6NAhItd+R2pICA+Wr+DB8hVoW1pi1KAB2mZmPDp4EHVkpOYa+lWrUuLjoZi2a0fU5i3cmz2bh+t8UFJTKTlpUpb/TyhpadxfsYLI1WsAeLT7D6wGDsxW/E8IIV5V5px0I53njwQSQog3yaFDh2jZsiVRUVFYWFgUdjjisUKbk56SkoKRkRHbt2+nR48emu2fffYZAQEBHD58OMvx0dHRWFpaov3EL93p6ekoioK2tjZ79+6lVatWL7yuzPsreCnqFAIjAzlz7wxn7p3h/L3zmmGDT9PT0qOObR2alG5CJ8dO2YbcixdLj4/neucu/60Rr6ODedeuWA/xfGZynyklOJiYnf8j9rffQKWi1MwZGDdokOOx6YmJhE+ZmnFsDrRLlMBuxnRMW7fWbFPS0ojd/Qcxv/5KwrlzKAlZfw60zMwwbd0as06dMG7aJEsiHrVlK3enTwfAuGlTjBs3xtC5Djq2toRP/oKEv/4CwLJ/P2wnTHjp+e9CFCb5bsp7efmZpqWnUfeHjGU0j7odldoroth70+ekh4aGMn36dHbv3s2DBw+ws7Oje/fuTJ06FWtr6yzHXrt2jTlz5rBv3z7u379P6dKladSoEWPHjqV+/fqa4/7880++/vprTp06RWJiIg4ODnTs2JExY8ZQpkwZTbJbo0YNLly4kCVnsbCwwNvbG/fHo/wcHBwIDg7mxIkTNHqiA8fLy4uAgAAOHTqU7Z6mT5/+wtHBN2/exOGJJYdzIyUlhYcPH1KyZMk8mSrYrl07Dhw4gL+/f5Z7e1vk1Zz0Qi8c5+LiwsqVKzXbqlevTrdu3bIVjktPTycwMDDLtpUrV3Lw4EF+/PFHHB0dMc7F+tDyi1DhS0tP4/aj20QkRHAv4R4RCRFcibrC6buneZD4QHOcChX1S9WnS4UumOmZ8c+Df/j3wb9cibpCBfMK9K/Wn9blW6OjVegDQoqc+JOnuDdnNkaNXLEe7I5u6dL5ch1FUXjo68f9JUsA0K9cGYPq1TCoXh2zjh3Rfs4TWSU1lcR//iXh1EnU0dEYN22GccMGqJ6YCvC06B9/JHzKVMjhf1taRkaUmjUT8/fee+37EqKgyXdT3svLzzQuJQ7Xza4AnBlwptiO+BIi05ucpN+4cQNXV1cqV67M7NmzcXR05OLFi3z++eekpKRw8uRJrKysADhz5gytW7emZs2aTJ48mapVq/Lo0SN+/fVXDh48qOk0XLNmDSNGjODDDz9k0KBBODg4EBISwvfff4+ZmRmLFy/WJOkGBgasXLmSwYMHa2LKKUm/d+8eDRo0yNIx+bwkPS4ujri4OM37d955h6FDh/LRE9MlbWxsNA8HUlJS0HvO71T5ISQkhBo1auDh4UFCQgJr164t0Os/LTU1FV1d3QK9Zl4l6SiFaMuWLYqurq6ybt06JTAwUPHy8lKMjY2VW7duKYqiKBMnTlQGDhz4zPOnTZum1KlT56WuGRMTowBKTEzM64Qu8kF6erpyPfq6svnSZmXwH4OVmn41X/hqu72t4vuPrxIRH1HY4b/V1ImJSnpKSoFcK/HiReX+mm+VkBGfKFeaNlUCq1RVrnftpiRdv1Eg1xciP8h3U97Ly8/0fsJ9paZfTaWWXy0lPT09D6ITomhLTEz8f3t3HlZV1bYB/D4M5zAPAjKEAg6IQGrhEI6lBpKzqYQT5vQiaio5lgk5JFYq+SWUrwifSip8mlZqIr5qzgOKkaI4JYoQIsggyHTW94evu47MiIBy/67rXHH2XnvvtZ88PKyz9lpLXL58WeTn50vblEqleFT4qF5e1fnc9evXT1hbW4u8vDyV7SkpKUJHR0f4+PhI9+Pk5CRcXFxESUlJqfNkZmYKIYS4c+eOkMvlYtasWWVe72m5Q4cOCQBi7ty5olmzZiqxMzQ0FGFhYdJ7GxsbMXPmTCGXy8WePXuk7TNnzhS9evWq0n3a2NiINWvWSO+9vb3F4MGDxRdffCEsLS2FjY2NEEKIzZs3CxcXF6GnpyfMzc2Fl5eX+Ouvv6Tjntb76X2EhYUJQ0ND8euvvwoHBwehq6sr3N3dxb179yqtU0BAgPjggw9EQkKC0NfXF7m5uaViNXnyZNG0aVOhUCiEk5OT+Pnnn6X9x44dEz179hTa2trCyMhIuLm5iYyMjDLvVwgh2rdvL/z9/aX3AERISIgYNGiQ0NHREYsXLxbFxcViwoQJwtbWVmhpaQl7e3sRFBRUqu6hoaHC0dFRyOVyYWFhIaZNmyaEEOLDDz8U/fv3VylbVFQkzM3NRWhoaKnzlPXZeao6ealeuyA9PT3x4MEDLFmyBCkpKXB2dsbevXthY2MDAEhJSUFSUlJ9VpHqkEwmQwvDFmhh2AIfOHyAe7n3sOfmHkTfjoZSKKWZ4lsZt8LRu0cReTUSKY9SsCp2FVbFroK9sT26vdYNXSy6QE2mhpzCHOQW5eJx8WMYaxnDVNsUJtomsNCxgI4mxxTWJrU6/JZdy9ERWo6OAJ705Jc8fAh1IyPO5k5EL0x+0d9rpPN3DTVW+cX56PJDl3q59ulRp6v0t1tGRgb279+P5cuXQ1tbdSUGCwsLjB49Gtu3b0dwcDDi4uJw6dIl/PDDD1ArY0nZp+Ozo6KiUFhYiHnz5pV5zWfHcc+aNQtbtmzBt99+izlz5pRbV1tbW/j4+GDhwoXo169fmXWoroMHD8LAwAAHDhyA+O9Th4WFhVi6dCnatGmDtLQ0zJ49G+PHj8fevXvLPU9eXh6+/vprbN68GWpqahgzZgzmzJmDiIiIco8RQiAsLAzr1q2Dg4MD7O3tERkZKT1RoFQq4eHhgZycHGzZsgUtW7bE5cuXpZ7/uLg49OnTBxMmTMDatWuhoaGBQ4cOoaSkpFox8Pf3x4oVK7BmzRqoq6tDqVTC2toakZGRMDU1xYkTJzBlyhRYWlpi5MiRAICQkBD4+fkhMDAQHh4eyMrKwvH/zs80adIk9OzZEykpKbC0fDIMd+/evcjNzZWOfxHq/TlhX19f+Pr6lrkvPDy8wmMDAgIQ8N8xqvTqsdKzwuR2kzG5XelZz9ubtcfkdpOx9+ZeRF6NxB8P/kBiZiISMxMR9kdYhefVUNPA8NbD8a/2/4KptumLqj7VAZlMBg1j4/quBhG94vJL/m6kE1HDde3aNQgh0LZt2zL3t23bFpmZmbh//z6uXbsGAHBwcKj0nAYGBlIDrTI6Ojrw9/fHJ598gsmTJ8PQ0LDcsosWLUJYWBgiIiIwduzYKp2/Irq6utiwYYPKY+4TJkyQfm7RogXWrl2Lzp07Izc3F3p6emWep6ioCN999x1atmwJAJg+fTqWLFlS4bVjYmKQl5cHd3d3AMCYMWMQGhoqNdJjYmJw5swZJCQkSJOAt/jHXElffvklOnbsqDIM2snJqTq3DwAYNWqUyj0DUBnLb2dnhxMnTiAyMlJqZC9btgwff/wxZs6cKZXr1KkTAKBr165o06YNNm/eLH1RExYWhhEjRpQbv9pQ7410oppSqCswtPVQDG09FBmPM3Dy3kmcuHcCF+9fhKaaJgzkBtCT60GhrkDm40yk56fjQf4D5BTlYNvVbdh9YzfGOo7FeKfx0Jfr1/ftEBFRA/V0Zncuv0aNmbaGNk6POl15wRd07drwtHdZLpdLP1f2dIwoZ+WpikycOBGrV6/GypUr8cUXX5RbzszMDHPmzMHixYvh6elZrWuU5fXXXy81Dv3ChQsICAhAXFwcMjIyoFQqATwZP+743ycTn6WjoyM10AHA0tISaWlpFV47NDQUnp6e0jLZXl5emDt3Lq5evYo2bdogLi4O1tbW5a7SFRcXhxEjRlT5Xsvzz8n+nvruu++wYcMG3L59G/n5+SgsLESHDh0AAGlpabh37x76/GOi42dNmjQJ69evx7x585CWloY9e/bg4D+WNn4R2EinV0ITrSbo36I/+reofNKwMylnEHQ+CPHp8Vj/+3psiN8APU096Mv1YSA3gKHCEE11msJU2xRNdZrC3tgebzZ9E+pqXM6LiKgxerpGOnvSqTGTyWQNfrhgq1atIJPJcPnyZQwZMqTU/itXrsDMzAxGRkZSYzEhIUFqsJXF3t4eWVlZKo87V0ZDQwPLli3D+PHjMX369ArL+vn5ITg4WKUHuaaenUT70aNHcHNzg5ubG7Zs2QIzMzMkJSXB3d0dhYWF5Z7n2cnWZDKZ9KVGWTIyMrBr1y4UFRUhJCRE2l5SUoKNGzdi5cqVpYYfPKuy/WpqaqXqUFRUVKrcszGIjIzE7NmzsWrVKri6ukJfX1+apb8q1wWAcePGYcGCBTh58iROnjwJW1tb9OjRo9LjnsfzD34gesl0tuyMiPciEPR2EFoYtoBSKJFdmI3k3GQkZCTgVMop/HTjJ2z8YyMCzwRiwv4J6B3VG0tOLsGplFNIeJCAmNsxCP8jHCvPrEREQgRuPLxR4S8vIiJ6eT3tSWcjnahhMzExwbvvvovg4GDk5+er7EtNTUVERIQ0w3qHDh3g6OiIVatWSb3L//Tw4UMAwPDhwyGXy/Hll1+Wec2n5Z41YsQIODk5Vbpsmp6eHj777DMsX74c2dnZFd9gNV25cgXp6ekIDAxEjx494ODgUGmPeE1ERETA2toaFy9eRFxcnPQKCgrC//7v/6K4uBjt2rXD3bt3kZiYWOY52rVrV2HvtJmZGVKeLi+MJzOl37p1q9K6HT16FF27doWvry/eeOMNtGrVCjdu3JD26+vrw9bWtsJrm5iYYMiQIQgLC0NYWJjKzP0vCnvSqVGSyWToY9MHvZv3xv38+8gtzEV2YTayC7PxsOAh7ufdx/38+/jr0V84+9dZZDzOQFRiFKISo8o9p5m2GTpZdIKJtgnkanLI1eUw0TLBwJYDG/w3z0REVD420oleHt9++y26du0Kd3f3Ukuw2dvbY/HixQCe/C0YFhaGvn37omfPntISbLm5ufj5558RHR2NI0eOoFmzZlizZg2mT5+O7OxsaQm2u3fvYtOmTdDT08OqVavKrEtgYKA0RrsiU6ZMwZo1a7B161Z06VJ7k/M1b94ccrkc//M//wMfHx/88ccfWLp0aa2d/6nQ0FAMHz4czs7OKtttbGwwf/587NmzB4MHD0bPnj3x/vvvY/Xq1WjVqhWuXLkCmUyGfv36YeHChXj99dfh6+sLHx8fyOVyHDp0CCNGjICpqSl69+6N8PBwDBw4EMbGxvjss89U1qIvT6tWrbBp0ybs378fdnZ22Lx5M86ePQs7OzupTEBAAHx8fNC0aVNpcrvjx49jxowZUplJkyZhwIABKCkpgbe3d+0FrxxspFOjJpPJ0FSnKZrqNC23TJGyCGdTziL6djQO3zkMAQFrPWu8pvcamuo0RWJmIs6nncf9/PvYe6v0TJmH7x7Guj7roCbjgytERC+jp4+7c0w6UcPXunVrnD17FgEBARg5ciTS0tIghMCwYcOwefNm6Oj83XHSuXNnnDt3DsuXL8fkyZORnp4OS0tLdO3aFUFBQVI5X19f2Nvb4+uvv8bQoUORn58PW1tbDBgwAH5+fuXWpXfv3ujduzeio6MrrLOmpiaWLl2KUaNGPff9/5OZmRnCw8PxySefYO3atXjzzTfx9ddfY9CgQbV2jdjYWFy8eLHMNdH19fXh5uaG0NBQDB48GDt27MCcOXPg5eWFR48eoVWrVggMDATwZFhBdHQ0PvnkE3Tu3Bna2tro0qULvLy8AAALFy7EzZs3MWDAABgaGmLp0qVV6kn38fFBXFwcPD09IZPJ4OXlBV9fX+zbt08q4+3tjcePH2PNmjWYM2cOTE1NMXz4cJXz9O3bF5aWlnBycoKVldXzhKxKZKKRPaNbrUXkiaqooKQAF9MuIu5+HB4VPUJhSSEKSwqx+8ZuFJQUYF6neRjr+PyzdhLRq4m5qfbVZkwjEiIQeCYQ7rbu+LrX17VUQ6KG6/Hjx7h16xbs7OygVYfLrL4o/v7+WL16NaKjo+Hq6lrf1aGXUF5eHqysrLBx40YMGzas3HIVfXaqk5fYk05UCxTqCnS27IzOlp1Vtrc2bo3lp5djTewadDTviLYmZS8JQkREDZfUk67+8jdWiBqjzz//HLa2tjh9+jS6dOlSK2uSU+OgVCqRmpqKVatWwdDQsFafQqgIG+lEL5BnG0+cuHcCh+4cwrzf5mH7gO0cn05E9JLhmHSil19dTPZFr56kpCTY2dnB2toa4eHh0hJzLxq/RiJ6gWQyGZZ0XYKmOk3xZ/afWHl2JYqVxfVdLSIiqgYuwUZE1DjZ2tpCCIE7d+5UuJZ6bWNPOtELZqRlhBXdV2BS9CTsvLYTu67vQhOtJjDTNoOlriWcTZ3xutnrcDZxhp5cr76rS0REz2BPOhER1SU20onqQGfLzpjbaS7WxK5BkbII6fnpSM9PR0JGAv5z5z8AABlkcGjiAA87D3jYecBC16Kea01ERADwuIQ96UREVHfYSCeqI2Mdx2KUwyhkFmQiLS8N6fnp+DPrT8Snx+P3+7/j3qN7SMhIQEJGwpOJ5iw6oqtVVxgpjGCkMIKhwhD6cn1oqWtBS0ML2hraMJAbQCaT1fetERG90p72pHMJNiIiqgtspBPVIXU1dZhqm8JU2xQA0NO6p7QvPT8dh+4cwp6bexD7VyzOpp7F2dSzFZ7PQG6A101fRzuzdmhn1g5dLLpAU13zhd4DEVFjw8fdiYioLnHiOKIGwlTbFCPsRyC8Xzj2v78fs11mY1DLQehl3QvtzdrDxsAGZtpm0NfUh4bak+/XsguzcfzecYRcDMHUmKkYtXcU7uTcqec7IaLGJjg4WFoT1sXFBUePHi237Pjx4yGTyUq9nJycVMrt2LEDjo6OUCgUcHR0xI8//viib6Nc7EknIqK6xJ50ogbISs8KE5wnVFimsKQQ1zKv4ff03xF/Px5Hk4/iSsYVeP7iicAegSq99EREL8r27dsxa9YsBAcHo1u3bvj+++/h4eGBy5cvo3nz5qXKf/PNNwgMDJTeFxcXo3379hgxYoS07eTJk/D09MTSpUsxdOhQ/Pjjjxg5ciSOHTuGLl261Ml9/RNndyciorrEnnSil5RcXQ4nUyd4OXjhix5fIGpgFNqZtUNOYQ6mH5yO4LhgFCmL6ruaRPSKW716NSZOnIhJkyahbdu2CAoKQrNmzRASElJmeUNDQ1hYWEivc+fOITMzU2UN46CgILz77rtYuHAhHBwcsHDhQvTp0wdBQUF1dFeq+Lg7Eb0qDh8+DJlMhocPHwIAwsPDYWRkVOExAQEB6NChwwuvG/2NjXSiV4SFrgXC3cPxQZsPICAQcjEEvbb1wseHP8bu67uRnJuMKxlXcCjpELZe2YoN8Ruw/8/9SMxMREFJgXQeIQSKSorYwCeiShUWFiI2NhZubm4q293c3HDixIkqnSM0NBR9+/aFjY2NtO3kyZOlzunu7l7hOQsKCpCdna3yqi3sSSd6udy5cwcTJ06ElZUV5HI5bGxsMHPmTDx48KBU2evXr+PDDz+EtbU1FAoF7Ozs4OXlhXPnzqmUO3ToEN577z2YmJhAR0cHjo6O+Pjjj5GcnAzg78avs7MzSkpKVI41MjJCeHi49N7W1hYymQynTp1SKTdr1iy8/fbbZd5TbGwsZDIZjh07VuZ+d3d3DBo0qLLQlOLp6YnExMRqH1ceNzc3qKurl7o3qh4+7k70CtFU18Snb32Kdmbt8PW5r5HxOAPRt6MRfTu6wuPUZGowlBuioKQABSUFKBFPkouWuhb05frQk+vB3tgew1oNw1tWb0FNxu/3iAhIT09HSUkJzM3NVbabm5sjNTW10uNTUlKwb98+/PDDDyrbU1NTq33OFStW4PPPP69G7atOGpOuzjHpRA3dzZs34erqCnt7e2zduhV2dna4dOkS5s6di3379uHUqVNo0qQJAODcuXPo06cPnJ2d8f3338PBwQE5OTnYvXs3Pv74Yxw5cgQA8P3338PX1xfe3t7YsWMHbG1tkZSUhE2bNmHVqlVYvXq1dP0bN25g06ZNKk8HlUVLSwvz58+XrlEZFxcXtG/fHmFhYejevbvKvjt37iAmJgY7d+6sTqgAANra2tDWrp0vIJOSknDy5ElMnz4doaGheOutt2rlvDVVVFQETc2Xc0Jl/qVN9Aoa2HIg/jPiP9jy3hZMaTcFbZu0hQwyNNFqAicTJ/Rt3hcDWwxEO7N20NfUh1IokVmQibziPKmBDjxZG/h+/n3cyrqF/X/ux79i/oX3dr6H7y9+zwnqiEjy7FKQQogqLQ/59DHLIUOGPPc5Fy5ciKysLOl1507t/Y6SHnfXZE86NV5CCCjz8urlJYSocj2nTZsGuVyO6Oho9OrVC82bN4eHhwdiYmKQnJyMTz/9VLqf8ePHo3Xr1jh69Cj69++Pli1bokOHDvD398fu3bsBAHfv3sVHH32Ejz76CBs3bsTbb78NW1tb9OzZExs2bMDixYtVrj9jxgz4+/vj8ePHFdbzX//6F06dOoW9e/dW+d4mTpyIyMhIPHr0SGV7eHg4zMzM0L9/f2zZsgUdO3aEvr4+LCwsMGrUKKSlpZV7zrIedw8MDIS5uTn09fUxceLESu/lqbCwMAwYMABTp07F9u3bS9Xz4cOHmDJlCszNzaGlpQVnZ2f88ssv0v7jx4+jV69e0NHRgbGxMdzd3ZGZmQngydMHzw556tChAwICAqT3MpkM3333HQYPHgxdXV0sW7YMJSUlmDhxIuzs7KCtrY02bdrgm2++KVX3jRs3wsnJCQqFApaWlpg+fToAYMKECRgwYIBK2eLiYlhYWGDjxo1ViktNsCed6BWlrqaO9mbt0d6sPWa8MQNKoSyzB1wIgQePHyDjcYa0BrtCXQEhBHKKcpBbmIvMgkwcvnMYv9z8Bcm5yfg27lt8G/ctrPWs8ZbVW9LSb5mPM5HxOANZBVnQ1dSFocIQxgpj6MuffBFQpHzyGL2aTA2tjFrB1sAW6mrqdR8cIqoVpqamUFdXL9XDnZaWVqon/FlCCGzcuBFjx46FXC5X2WdhYVHtcyoUCigUimreQeWUQonHJXzcnUjk5+Pqmy71cu0252Mh09GptFxGRgb279+P5cuXl+odtrCwwOjRo7F9+3YEBwcjLi4Oly5dwg8//AA1tdJ/Hz1tuEZFRaGwsBDz5s0r85rPNnBnzZqFLVu24Ntvv8WcOXPKrautrS18fHywcOFC9OvXr8w6PGv06NGYO3cuoqKiMH78eABPfpeGh4fD29sbGhoaKCwsxNKlS9GmTRukpaVh9uzZGD9+fJW/DIiMjIS/vz/WrVuHHj16YPPmzVi7di1atGhR4XFCCISFhWHdunVwcHCAvb09IiMjpScKlEolPDw8kJOTgy1btqBly5a4fPky1NWf/B0YFxeHPn36YMKECVi7di00NDRw6NChUkMHKuPv748VK1ZgzZo1UFdXh1KphLW1NSIjI2FqaooTJ05gypQpsLS0xMiRIwEAISEh8PPzQ2BgIDw8PJCVlYXjx48DACZNmoSePXsiJSUFlpaWAIC9e/ciNzdXOv5FYCOdqJEo7xF1mUymsnb7PxlpGUk/d7XqCj8XPxy4fQC7ru/C+b/O427uXfxf4v/h/xL/r0Z10lLXQmvj1rA3toetgS1sDW1ha2ALXU1dpD5KRcqjFKQ8SoGptin62vSFQr32/wAnopqTy+VwcXHBgQMHMHToUGn7gQMHMHjw4AqPPXLkCK5fv46JEyeW2ufq6ooDBw5g9uzZ0rbo6Gh07dq19ipfRU/HowN83J2oobt27RqEEGjbtm2Z+9u2bYvMzEzcv38f165dAwA4ODhUek4DAwOpgVYZHR0d+Pv745NPPsHkyZNhaGhYbtlFixYhLCwMERERGDt2bKXnbtKkCYYMGYKwsDCpkX748GHcvHkTEyY8WRXo6X8BoEWLFli7di06d+6M3Nxc6OnpVXqNoKAgTJgwAZMmTQIALFu2DDExMZX2psfExCAvLw/u7u4AgDFjxiA0NFRqpMfExODMmTNISEiAvb29VL+nvvzyS3Ts2BHBwcHStmeX5qyKUaNGqcQAgMpQKDs7O5w4cQKRkZFSI3vZsmX4+OOPMXPmTKlcp06dAABdu3ZFmzZtsHnzZumLmrCwMIwYMaJK8awpNtKJqMq0NLQwsOVADGw5EHlFeTj31zmcvHcSF9IuQF1NHU0UTdBEuwkM5AbIK8rDw4KHeFjwEDmFOVCXqUNTXROaapp4XPIY1zKvIb84H/Hp8YhPj6/02k3ONsFw++HwbOMJEy0T3Mq6hcsZl5GYkQh9uT7sje3R2rg1rPSsOGaeqA75+flh7Nix6NixI1xdXbF+/XokJSXBx8cHwJPH0JOTk7Fp0yaV40JDQ9GlSxc4OzuXOufMmTPRs2dPrFy5EoMHD8bu3bsRExNT7oRJL9LTXnSA66RT4ybT1kab87H1du3a8PSxeblcLv1c2dCcqg7f+aeJEydi9erVWLlyJb744otyy5mZmWHOnDlYvHgxPD09q3xuNzc3XL9+Ha1atcLGjRvRrVs3tGnTBgBw4cIFBAQEIC4uDhkZGVAqlQCejBd3dHSs9PwJCQnS7++nXF1dcejQoQqPCw0NhaenJzQ0njQvvby8MHfuXFy9ehVt2rRBXFwcrK2tpQb6s+Li4lSW4qypjh07ltr23XffYcOGDbh9+zby8/NRWFgozVaflpaGe/fuoU+fPuWec9KkSVi/fj3mzZuHtLQ07NmzBwcPHnzuulaEjXQiqhEdTR30tO5Z4/XYlUKJOzl3kJCRgOuZ1/Fn9p+4nX0bt7Nvo6CkAGbaZrDUtYS5rjku3r+I1EepWP/7emyM3whNdU1pjGipemnooK9NXyzvvvx5bo+IqsjT0xMPHjzAkiVLkJKSAmdnZ+zdu1earT0lJQVJSUkqx2RlZWHHjh1ljgsEnvRcbNu2DYsWLcJnn32Gli1bYvv27fWyRvo/J43jF4DUmMlksio9cl6fWrVqBZlMhsuXL5c518WVK1dgZmYGIyMjqbGYkJBQ4fJi9vb2yMrKUnncuTIaGhpYtmwZxo8fL41tLo+fnx+Cg4NVepAr8nQ1jPDwcMybNw87d+7Et99+CwB49OgR3Nzc4Obmhi1btsDMzAxJSUlwd3dHYWFhlc5fExkZGdi1axeKiopUlt8sKSnBxo0bsXLlykonp6tsv5qaWqm5CYqKSq9EpKurq/I+MjISs2fPxqpVq+Dq6gp9fX189dVXOH36dJWuCwDjxo3DggULcPLkSZw8eRK2trbo0aNHpcc9DzbSiaheqMnUYGNgAxsDG8D27+1KoYRSKKGh9vevp2JlMQ4mHcQPCT/gfNp5FBcXQ1tDG22btEWbJm2QU5iDa5nXcDPrJvKK86AUyrq/IaJGzNfXF76+vmXu++eyQ08ZGhoiLy+vwnMOHz4cw4cPr43qPRcuv0b08jAxMcG7776L4OBgzJ49W6UBlpqaioiICEybNg3Ak0nHHB0dsWrVKnh6epYaE/7w4UMYGRlh+PDhWLBgAb788kusWbOm1DWflnvWiBEj8NVXX1W66oSenh4+++wzBAQEYODAgZXeo0wmw4cffogNGzbA2toaampq0mPbV65cQXp6OgIDA9GsWTMAKLWUXGXatm2LU6dOYdy4cdK2ypZTi4iIgLW1NXbt2qWy/eDBg1ixYgWWL1+Odu3a4e7du0hMTCyzN71du3Y4ePBgufEyMzNDSkqK9D47Oxu3bt2q9H6OHj2Krl27quSoGzduSD/r6+vD1tYWBw8exDvvvFPmOUxMTKRhBidPnqx05v5aIRqZrKwsAUBkZWXVd1WIqAZuPbwlbmTeEMUlxaX2FZYUiuuZ18WNhzfqoWZENcfcVPtqK6aPCh+JMylnxKl7p2qpZkQNX35+vrh8+bLIz8+v76pUW2JiojA1NRU9evQQR44cEUlJSWLfvn3C2dlZdOjQQeTk5EhlT58+LfT19UW3bt3Enj17xI0bN8TFixfFsmXLRM+ePaVy69atEzKZTEyYMEEcPnxY/Pnnn+LYsWNiypQpws/PTwghxKFDhwQAkZmZKR138OBBoaGhITQ0NERYWJi03cbGRqxZs0Z6X1hYKFq2bCm0tLREr169Kr3H27dvCzU1NWFsbCwmTZokbU9LSxNyuVzMnTtX3LhxQ+zevVvY29sLAOLChQtl1jMsLEwYGhpK59i2bZtQKBQiNDRUXL16VSxevFjo6+uL9u3bl1uf9u3bi/nz55fanp2dLRQKhdi1a5cQQoi3335bODs7i+joaHHz5k2xd+9esW/fPiGEEFevXhVyuVxMnTpVXLx4USQkJIjg4GBx//59IYQQCxYsEBYWFuK3334T8fHxYsiQIUJPT0/4+/tL1wMgfvzxR5U6BAUFCQMDA/Hrr7+Kq1evikWLFgkDAwOV+wkPDxdaWlrim2++EYmJiSI2NlasXbtW5TzR0dFCLpcLdXV1kZycXG4sKvrsVCcv8bktInqp2BraooVRizJnhddU00RLo5ZoYVjxDKRERFWlo6mDThad0MWy7h+1J6Lqa926Nc6ePYsWLVpg5MiRsLGxgYeHB+zt7XH8+HGVyb46d+6Mc+fOoWXLlpg8eTLatm2LQYMG4dKlSyrLffn6+iI6OhrJyckYOnQoHBwcMGnSJBgYGFQ4g3vv3r3Ru3dvFBcXV1hnTU1NLF26tMpLnTVv3hx9+/ZFZmamyiRpZmZmCA8PR1RUFBwdHREYGIivv/66Sud8ytPTE4sXL8b8+fPh4uKC27dvY+rUqeWWj42NxcWLF/H++++X2qevrw83NzeEhoYCAHbs2IFOnTrBy8sLjo6OmDdvnjR7u729PaKjo3Hx4kV07twZrq6u2L17tzTGfeHChejZsycGDBiA9957D0OGDEHLli0rvR8fHx8MGzYMnp6e6NKlCx48eFDqyS9vb28EBQUhODgYTk5OGDBggDSx4FN9+/aFpaUl3N3dYWVlVel1n5fsv986NBrZ2dkwNDREVlYWDAwM6rs6REREzE0vAGNKVHOPHz/GrVu3YGdnBy2tl3/CRH9/f6xevRrR0dFwdXWt7+rQSygvLw9WVlbYuHEjhg0bVm65ij471clLHJNORERERESvrM8//xy2trY4ffo0unTpUqU1yYmAJ+u7p6amYtWqVTA0NMSgQYPq5LpspBMRERER0SutTib7oldOUlIS7OzsYG1tjfDwcOnx+xeNjXQiIiIiIiKiZ9ja2pZa+q0u8FkPIiIiIiIiogaCjXQiIiIiIiqlkc0vTfTcauszw0Y6ERERERFJNDU1ATyZ0ZqIqq6wsBAAoK5eeqng6uCYdCIiIiIikqirq8PIyAhpaWkAAB0dHchksnquFVHDplQqcf/+fejo6Dz3BHNspBMRERERkQoLCwsAkBrqRFQ5NTU1NG/e/Lm/1GIjnYiIiIiIVMhkMlhaWqJp06YoKiqq7+oQvRTkcjnU1J5/RHm9N9KDg4Px1VdfISUlBU5OTggKCkKPHj3KLLtz506EhIQgLi4OBQUFcHJyQkBAANzd3eu41kRERERErz51dfXnHl9LRNVTrxPHbd++HbNmzcKnn36KCxcuoEePHvDw8EBSUlKZ5X/77Te8++672Lt3L2JjY/HOO+9g4MCBuHDhQh3XnIiIiIiIiKj2yUQ9rq3QpUsXvPnmmwgJCZG2tW3bFkOGDMGKFSuqdA4nJyd4enpi8eLFVSqfnZ0NQ0NDZGVlwcDAoEb1JiIiqk3MTbWPMSUiooakOnmp3nrSCwsLERsbCzc3N5Xtbm5uOHHiRJXOoVQqkZOTgyZNmpRbpqCgANnZ2SovIiIiIiIiooao3sakp6eno6SkBObm5irbzc3NkZqaWqVzrFq1Co8ePcLIkSPLLbNixQp8/vnnpbazsU5ERA3F05xUjw+3vXKexpL5noiIGoLq5Pp6nzju2enphRBVmrJ+69atCAgIwO7du9G0adNyyy1cuBB+fn7S++TkZDg6OqJZs2Y1rzQREdELkJOTA0NDw/quxishJycHAJjviYioQalKrq+3RrqpqSnU1dVL9ZqnpaWV6l1/1vbt2zFx4kRERUWhb9++FZZVKBRQKBTSez09Pdy5cwf6+vo1Wr8uOzsbzZo1w507dzjGrYoYs5ph3GqGcas+xqxmajNuQgjk5OTAysqqlmpHVlZWNc73/EzUDONWfYxZzTBuNcO4VV995fp6a6TL5XK4uLjgwIEDGDp0qLT9wIEDGDx4cLnHbd26FRMmTMDWrVvRv3//al9XTU0N1tbWNarzPxkYGPAfdzUxZjXDuNUM41Z9jFnN1Fbc2INeu2oj3/MzUTOMW/UxZjXDuNUM41Z9dZ3r6/Vxdz8/P4wdOxYdO3aEq6sr1q9fj6SkJPj4+AB48qh6cnIyNm3aBOBJA33cuHH45ptv8NZbb0m98Nra2vzjhoiIiIiIiF569dpI9/T0xIMHD7BkyRKkpKTA2dkZe/fuhY2NDQAgJSVFZc3077//HsXFxZg2bRqmTZsmbff29kZ4eHhdV5+IiIiIiIioVtX7xHG+vr7w9fUtc9+zDe/Dhw+/+ApVQqFQwN/fX2WcO1WMMasZxq1mGLfqY8xqhnF7dfH/bc0wbtXHmNUM41YzjFv11VfMZILrvRARERERERE1CGr1XQEiIiIiIiIieoKNdCIiIiIiIqIGgo10IiIiIiIiogaCjXQiIiIiIiKiBoKN9GoIDg6GnZ0dtLS04OLigqNHj9Z3lRqMFStWoFOnTtDX10fTpk0xZMgQXL16VaWMEAIBAQGwsrKCtrY23n77bVy6dKmeatwwrVixAjKZDLNmzZK2MW5lS05OxpgxY2BiYgIdHR106NABsbGx0n7GTVVxcTEWLVoEOzs7aGtro0WLFliyZAmUSqVUhjEDfvvtNwwcOBBWVlaQyWTYtWuXyv6qxKigoAAzZsyAqakpdHV1MWjQINy9e7cO74KeB3N9xZjvnx9zfdUx11cf833lXopcL6hKtm3bJjQ1NcW///1vcfnyZTFz5kyhq6srbt++Xd9VaxDc3d1FWFiY+OOPP0RcXJzo37+/aN68ucjNzZXKBAYGCn19fbFjxw4RHx8vPD09haWlpcjOzq7HmjccZ86cEba2tqJdu3Zi5syZ0nbGrbSMjAxhY2Mjxo8fL06fPi1u3bolYmJixPXr16UyjJuqZcuWCRMTE/HLL7+IW7duiaioKKGnpyeCgoKkMoyZEHv37hWffvqp2LFjhwAgfvzxR5X9VYmRj4+PeO2118SBAwfE+fPnxTvvvCPat28viouL6/huqLqY6yvHfP98mOurjrm+ZpjvK/cy5Ho20quoc+fOwsfHR2Wbg4ODWLBgQT3VqGFLS0sTAMSRI0eEEEIolUphYWEhAgMDpTKPHz8WhoaG4rvvvquvajYYOTk5onXr1uLAgQOiV69eUuJm3Mo2f/580b1793L3M26l9e/fX0yYMEFl27Bhw8SYMWOEEIxZWZ5N3FWJ0cOHD4WmpqbYtm2bVCY5OVmoqamJX3/9tc7qTjXDXF99zPdVx1xfPcz1NcN8Xz0NNdfzcfcqKCwsRGxsLNzc3FS2u7m54cSJE/VUq4YtKysLANCkSRMAwK1bt5CamqoSQ4VCgV69ejGGAKZNm4b+/fujb9++KtsZt7L99NNP6NixI0aMGIGmTZvijTfewL///W9pP+NWWvfu3XHw4EEkJiYCAC5evIhjx47hvffeA8CYVUVVYhQbG4uioiKVMlZWVnB2dmYcGzjm+pphvq865vrqYa6vGeb759NQcr1GrZzlFZeeno6SkhKYm5urbDc3N0dqamo91arhEkLAz88P3bt3h7OzMwBIcSorhrdv367zOjYk27Ztw/nz53H27NlS+xi3st28eRMhISHw8/PDJ598gjNnzuCjjz6CQqHAuHHjGLcyzJ8/H1lZWXBwcIC6ujpKSkqwfPlyeHl5AeC/taqoSoxSU1Mhl8thbGxcqgzzRcPGXF99zPdVx1xffcz1NcN8/3waSq5nI70aZDKZynshRKltBEyfPh2///47jh07VmofY6jqzp07mDlzJqKjo6GlpVVuOcZNlVKpRMeOHfHFF18AAN544w1cunQJISEhGDdunFSOcfvb9u3bsWXLFvzwww9wcnJCXFwcZs2aBSsrK3h7e0vlGLPK1SRGjOPLg5+BqmO+rxrm+pphrq8Z5vvaUd+5no+7V4GpqSnU1dVLfTOSlpZW6luWxm7GjBn46aefcOjQIVhbW0vbLSwsAIAxfEZsbCzS0tLg4uICDQ0NaGho4MiRI1i7di00NDSk2DBuqiwtLeHo6KiyrW3btkhKSgLAf29lmTt3LhYsWIAPPvgAr7/+OsaOHYvZs2djxYoVABizqqhKjCwsLFBYWIjMzMxyy1DDxFxfPcz3VcdcXzPM9TXDfP98GkquZyO9CuRyOVxcXHDgwAGV7QcOHEDXrl3rqVYNixAC06dPx86dO/Gf//wHdnZ2Kvvt7OxgYWGhEsPCwkIcOXKkUcewT58+iI+PR1xcnPTq2LEjRo8ejbi4OLRo0YJxK0O3bt1KLfmTmJgIGxsbAPz3Vpa8vDyoqan+yldXV5eWZGHMKleVGLm4uEBTU1OlTEpKCv744w/GsYFjrq8a5vvqY66vGeb6mmG+fz4NJtfXyvRzjcDTZVlCQ0PF5cuXxaxZs4Surq74888/67tqDcLUqVOFoaGhOHz4sEhJSZFeeXl5UpnAwEBhaGgodu7cKeLj44WXl1ejWu6hqv4546sQjFtZzpw5IzQ0NMTy5cvFtWvXREREhNDR0RFbtmyRyjBuqry9vcVrr70mLcmyc+dOYWpqKubNmyeVYcyezL584cIFceHCBQFArF69Wly4cEFagqsqMfLx8RHW1tYiJiZGnD9/XvTu3ZtLsL0kmOsrx3xfO5jrK8dcXzPM95V7GXI9G+nVsG7dOmFjYyPkcrl48803peVG6MnyBWW9wsLCpDJKpVL4+/sLCwsLoVAoRM+ePUV8fHz9VbqBejZxM25l+/nnn4Wzs7NQKBTCwcFBrF+/XmU/46YqOztbzJw5UzRv3lxoaWmJFi1aiE8//VQUFBRIZRgzIQ4dOlTm7zJvb28hRNVilJ+fL6ZPny6aNGkitLW1xYABA0RSUlI93A3VBHN9xZjvawdzfdUw11cf833lXoZcLxNCiNrpkyciIiIiIiKi58Ex6UREREREREQNBBvpRERERERERA0EG+lEREREREREDQQb6UREREREREQNBBvpRERERERERA0EG+lEREREREREDQQb6UREREREREQNBBvpRERERERERA0EG+lEVOdkMhl27dpV39UgIiKiF4S5nqjm2EgnamTGjx8PmUxW6tWvX7/6rhoRERHVAuZ6opebRn1XgIjqXr9+/RAWFqayTaFQ1FNtiIiIqLYx1xO9vNiTTtQIKRQKWFhYqLyMjY0BPHk8LSQkBB4eHtDW1oadnR2ioqJUjo+Pj0fv3r2hra0NExMTTJkyBbm5uSplNm7cCCcnJygUClhaWmL69Okq+9PT0zF06FDo6OigdevW+Omnn17sTRMRETUizPVELy820omolM8++wzvv/8+Ll68iDFjxsDLywsJCQkAgLy8PPTr1w/GxsY4e/YsoqKiEBMTo5KYQ0JCMG3aNEyZMgXx8fH46aef0KpVK5VrfP755xg5ciR+//13vPfeexg9ejQyMjLq9D6JiIgaK+Z6ogZMEFGj4u3tLdTV1YWurq7Ka8mSJUIIIQAIHx8flWO6dOkipk6dKoQQYv369cLY2Fjk5uZK+/fs2SPU1NREamqqEEIIKysr8emnn5ZbBwBi0aJF0vvc3Fwhk8nEvn37au0+iYiIGivmeqKXG8ekEzVC77zzDkJCQlS2NWnSRPrZ1dVVZZ+rqyvi4uIAAAkJCWjfvj10dXWl/d26dYNSqcTVq1chk8lw79499OnTp8I6tGvXTvpZV1cX+vr6SEtLq+ktERER0T8w1xO9vNhIJ2qEdHV1Sz2SVhmZTAYAEEJIP5dVRltbu0rn09TULHWsUqmsVp2IiIiobMz1RC8vjkknolJOnTpV6r2DgwMAwNHREXFxcXj06JG0//jx41BTU4O9vT309fVha2uLgwcP1mmdiYiIqOqY64kaLvakEzVCBQUFSE1NVdmmoaEBU1NTAEBUVBQ6duyI7t27IyIiAmfOnEFoaCgAYPTo0fD394e3tzcCAgJw//59zJgxA2PHjoW5uTkAICAgAD4+PmjatCk8PDyQk5OD48ePY8aMGXV7o0RERI0Ucz3Ry4uNdKJG6Ndff4WlpaXKtjZt2uDKlSsAnszGum3bNvj6+sLCwgIRERFwdHQEAOjo6GD//v2YOXMmOnXqBB0dHbz//vtYvXq1dC5vb288fvwYa9aswZw5c2Bqaorhw4fX3Q0SERE1csz1RC8vmRBC1HcliKjhkMlk+PHHHzFkyJD6rgoRERG9AMz1RA0bx6QTERERERERNRBspBMRERERERE1EHzcnYiIiIiIiKiBYE86ERERERERUQPBRjoRERERERFRA8FGOhEREREREVEDwUY6ERERERERUQPBRjoRERERERFRA8FGOhEREREREVEDwUY6ERERERERUQPBRjoRERERERFRA/H/l6ruwQWmXjQAAAAASUVORK5CYII="
      },
      "metadata": {},
      "output_type": "display_data"
     }
    ],
-   "execution_count": 19
+   "execution_count": 36
   },
   {
    "metadata": {
     "ExecuteTime": {
-     "end_time": "2025-06-25T07:05:55.477381Z",
-     "start_time": "2025-06-25T07:05:55.462043Z"
+     "end_time": "2025-06-26T04:01:12.832423Z",
+     "start_time": "2025-06-26T04:01:12.811096Z"
     }
    },
    "cell_type": "code",
@@ -913,11 +394644,11 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "VGG 模型可训练参数量：362346\t QCCNN模型可训练参数量：26304\n"
+      "VGG 模型可训练参数量：903018\t QCCNN模型可训练参数量：1031792\n"
      ]
     }
    ],
-   "execution_count": 20
+   "execution_count": 37
   }
  ],
  "metadata": {},
diff --git a/Modify.py b/Modify.py
index 2b2c7d6..8e8bd56 100644
--- a/Modify.py
+++ b/Modify.py
@@ -71,10 +71,10 @@ def train_model(model, criterion, optimizer, train_loader, valid_loader, num_epo
     valid_acc_list = []
 
     best_valid_acc = 0.0
-    patience = 10  # 早停耐心值
+    patience = 50  # 早停耐心值
     counter = 0   # 计数器
 
-    scheduler = optim.lr_scheduler.ReduceLROnPlateau(optimizer, mode='max', factor=0.5, patience=10)
+    scheduler = optim.lr_scheduler.ReduceLROnPlateau(optimizer, mode='max', factor=0.5, patience=25)
 
     with tqdm(total=num_epochs) as pbar:
         for epoch in range(num_epochs):
@@ -170,7 +170,6 @@ trans1 = transforms.Compose([
     transforms.RandomHorizontalFlip(),  # 随机水平翻转
     transforms.RandomRotation(10),     # 随机旋转±10度
     transforms.ColorJitter(brightness=0.2, contrast=0.2),  # 颜色调整
-    transforms.Resize((18, 18)),       # 调整大小为18x18
     transforms.ToTensor(),           # 转换为张量
     transforms.Normalize((0.5,), (0.5,))  # 归一化到[-1, 1]
 ])
@@ -179,11 +178,10 @@ trans2 = transforms.Compose([
     transforms.RandomHorizontalFlip(),  # 随机水平翻转
     transforms.RandomRotation(10),     # 随机旋转±10度
     transforms.ColorJitter(brightness=0.2, contrast=0.2),  # 颜色调整
-    transforms.Resize((16, 16)),       # 调整大小为16x16
     transforms.ToTensor(),           # 转换为张量
     transforms.Normalize((0.5,), (0.5,))  # 归一化到[-1, 1]
 ])
-train_dataset = FashionMNIST(root='./data/notebook2', train=False, transform=trans1,download=True)
+train_dataset = FashionMNIST(root='./data/notebook2', train=True, transform=trans1,download=True)
 test_dataset = FashionMNIST(root='./data/notebook2', train=False, transform=trans1,download=True)
 
 # 定义训练集和测试集的比例
@@ -256,26 +254,36 @@ class RandomQCCNN(nn.Module):
     def __init__(self):
         super(RandomQCCNN, self).__init__()
         self.conv = nn.Sequential(
-            RandomQuantumConvolutionalLayer(nqubit=4, num_circuits=3, seed=1024),   # num_circuits=3代表我们在quanv1层只用了3个量子卷积核
+            RandomQuantumConvolutionalLayer(nqubit=4, num_circuits=3),   # num_circuits=3代表我们在quanv1层只用了3个量子卷积核
             nn.BatchNorm2d(3),  # 添加批量归一化
             nn.ReLU(),
             nn.MaxPool2d(kernel_size=2, stride=1),
+            # 添加形状检查层以确保尺寸正确
             nn.Conv2d(3, 6, kernel_size=2, stride=1),
             nn.BatchNorm2d(6),  # 添加批量归一化
             nn.ReLU(),
-            nn.MaxPool2d(kernel_size=2, stride=1)
+            # 添加自适应池化层确保固定输出尺寸
+            nn.AdaptiveMaxPool2d((9, 9))  # 确保输出固定尺寸
         )
         self.fc = nn.Sequential(
-            nn.Linear(6 * 6 * 6, 1024),
-            nn.BatchNorm1d(1024),  # 添加批量归一化
-            nn.Dropout(0.5),  # 增加dropout比例
+            # 根据自适应池化后的固定尺寸计算输入维度
+            nn.Linear(6 * 9 * 9, 1024),  # 确保与自适应池化输出匹配
+            nn.BatchNorm1d(1024),
+            nn.Dropout(0.5),
             nn.ReLU(),
-            nn.Linear(1024, 10)
+            nn.Linear(1024, 512),
+            nn.BatchNorm1d(512),
+            nn.ReLU(),
+            nn.Linear(512, 10)
         )
 
     def forward(self, x):
+        # 添加详细的形状检查输出
+        print(f"Input shape: {x.shape}")
         x = self.conv(x)
+        print(f"After conv layers: {x.shape}")  # 添加中间形状检查
         x = x.reshape(x.size(0), -1)
+        print(f"After flatten: {x.shape}")  # 添加展平后形状检查
         x = self.fc(x)
         return x
 #%%
@@ -335,9 +343,9 @@ class ParameterizedQuantumConvolutionalLayer(nn.Module):
 
     def circuit(self, nqubit):
         cir = dq.QubitCircuit(nqubit)
-        cir.rxlayer(encode=True)  #对原论文的量子线路结构并无影响，只是做了一个数据编码的操作
+        cir.rxlayer(encode=True)  # 数据编码
         cir.barrier()
-        for iter in range(4): #对应原论文中一个量子卷积线路上的深度为4，可控参数一共16个
+        for iter in range(5):  # 将线路深度从4增加到5
             cir.rylayer()
             cir.cnot_ring()
             cir.barrier()
@@ -346,20 +354,30 @@ class ParameterizedQuantumConvolutionalLayer(nn.Module):
         return cir
 
     def forward(self, x):
-        kernel_size, stride = 2, 2
-        # [64, 1, 18, 18] -> [64, 1, 9, 18, 2] -> [64, 1, 9, 9, 2, 2]
+        kernel_size, stride = 3, 3  # 使用3x3数据块
         x_unflod = x.unfold(2, kernel_size, stride).unfold(3, kernel_size, stride)
-        w = int((x.shape[-1] - kernel_size) / stride + 1)
-        x_reshape = x_unflod.reshape(-1, self.nqubit)
+        print(f"Input shape: {x.shape}")  # 添加输入形状检查
+        print(f"Unfolded shape: {x_unflod.shape}")  # 添加展开后形状检查
+        
+        # 动态计算w值并验证特征图尺寸
+        w = x_unflod.shape[2]  # 使用实际展开后的尺寸
+        # 确保展平后的总元素数与量子线路输入匹配
+        x_reshape = x_unflod.reshape(-1, kernel_size * kernel_size)  # 将每个3x3块展平为9维
 
         exps = []
-        for cir in self.cirs: # out_channels
+        for cir in self.cirs:
             cir(x_reshape)
             exp = cir.expectation()
             exps.append(exp)
 
         exps = torch.stack(exps, dim=1)
-        exps = exps.reshape(x.shape[0], 3, w, w)
+        out_channels = len(self.cirs)  # 使用动态计算而非硬编码值
+        # 验证总元素数一致性
+        assert exps.numel() == x.shape[0] * out_channels * w * w, \
+            f"Element count mismatch: {exps.numel()} vs {x.shape[0] * out_channels * w * w}"
+        # 确保展平后的总元素数与量子线路输出匹配
+        exps = exps.reshape(x.shape[0], out_channels, w, w)
+        print(f"Reshaped shape: {exps.shape}")  # 添加最终形状检查
         return exps
 #%%
 # 此处我们可视化其中一个量子卷积核的线路结构：
@@ -371,18 +389,26 @@ class QCCNN(nn.Module):
     def __init__(self):
         super(QCCNN, self).__init__()
         self.conv = nn.Sequential(
-            ParameterizedQuantumConvolutionalLayer(nqubit=4, num_circuits=3),
-            nn.BatchNorm2d(3),  # 添加批量归一化
+            ParameterizedQuantumConvolutionalLayer(nqubit=4, num_circuits=3),  # 恢复为4量子比特
+            nn.BatchNorm2d(3),  # 恢复原始通道数
             nn.ReLU(),
-            nn.MaxPool2d(kernel_size=2, stride=1)
+            nn.MaxPool2d(kernel_size=2, stride=1),
+            nn.Conv2d(3, 6, kernel_size=1),  # 添加1x1卷积层增强通道间信息交互
+            nn.BatchNorm2d(6),
+            nn.ReLU(),
+            nn.AdaptiveMaxPool2d((9, 9))  # 确保输出固定尺寸
         )
 
         self.fc = nn.Sequential(
-            nn.Linear(8 * 8 * 3, 128),
-            nn.BatchNorm1d(128),  # 添加批量归一化
-            nn.Dropout(0.5),  # 增加dropout比例
+            # 根据新的特征图大小调整输入维度：6通道、9x9特征图 => 6*9*9=486
+            nn.Linear(6 * 9 * 9, 1024),  # 修改为正确的输入维度
+            nn.BatchNorm1d(1024),
+            nn.Dropout(0.5),
             nn.ReLU(),
-            nn.Linear(128, 10)
+            nn.Linear(1024, 512),
+            nn.BatchNorm1d(512),
+            nn.ReLU(),
+            nn.Linear(512, 10)
         )
 
     def forward(self, x):
@@ -398,7 +424,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model = QCCNN()
 model.to(device)
 criterion = nn.CrossEntropyLoss()
-optimizer = optim.AdamW(model.parameters(), lr=3e-4, weight_decay=1e-5)  # 使用AdamW优化器和适当的权重衰减
+optimizer = optim.AdamW(model.parameters(), lr=5e-4, weight_decay=5e-5, amsgrad=True)  # 优化学习率和weight_decay参数
 optim_model, metrics = train_model(model, criterion, optimizer, train_loader, valid_loader, num_epochs, device)
 torch.save(optim_model.state_dict(), './data/notebook2/qccnn_weights.pt')   # 保存训练好的模型参数，用于后续的推理或测试
 pd.DataFrame(metrics).to_csv('./data/notebook2/qccnn_metrics.csv', index='None')   # 保存模型训练过程，用于后续图标展示
@@ -425,7 +451,7 @@ VGG = nn.Sequential(
     vgg_block(1, 32, 2),  # 增加通道数和调整卷积层数量
     vgg_block(32, 64, 2),
     nn.Flatten(),
-    nn.Linear(64 * 4 * 4, 256),  # 调整全连接层大小
+    nn.Linear(64 * 7 * 7, 256),  # 修改为正确的输入维度
     nn.BatchNorm1d(256),  # 添加批量归一化
     nn.ReLU(),
     nn.Dropout(0.5),  # 增加dropout比例
diff --git a/data/notebook2/best_model.pt b/data/notebook2/best_model.pt
index 433ffb2..8aa5d6e 100644
Binary files a/data/notebook2/best_model.pt and b/data/notebook2/best_model.pt differ
diff --git a/data/notebook2/qccnn_best.pt b/data/notebook2/qccnn_best.pt
deleted file mode 100644
index 9aae44c..0000000
Binary files a/data/notebook2/qccnn_best.pt and /dev/null differ
diff --git a/data/notebook2/qccnn_metrics.csv b/data/notebook2/qccnn_metrics.csv
index 6fdd053..eed5d32 100644
--- a/data/notebook2/qccnn_metrics.csv
+++ b/data/notebook2/qccnn_metrics.csv
@@ -1,49 +1,61 @@
 ,epoch,train_acc,valid_acc,train_loss,valid_loss
-0,1,0.391375,0.5882056451612904,1.7793689422607422,1.3356874796652025
-1,2,0.617125,0.6365927419354839,1.059556163787842,0.9690603344671188
-2,3,0.663125,0.6577620967741935,0.9102170023918151,0.9078671547674364
-3,4,0.69,0.6819556451612904,0.8537670917510987,0.8649420180628377
-4,5,0.71125,0.7056451612903226,0.8021836678981781,0.8317571128568342
-5,6,0.719375,0.6844758064516129,0.7773117737770081,0.8293971457789021
-6,7,0.733,0.7247983870967742,0.7432277894020081,0.7570091389840649
-7,8,0.727125,0.7051411290322581,0.7392586808204651,0.7708722833664187
-8,9,0.740375,0.7091733870967742,0.716008551120758,0.7547545788749572
-9,10,0.74775,0.7091733870967742,0.6988923935890198,0.7634256799374858
-10,11,0.75025,0.7328629032258065,0.6836859595775604,0.7220065324537216
-11,12,0.75525,0.7313508064516129,0.6790840411186219,0.7320531125991575
-12,13,0.755875,0.734375,0.6686462206840516,0.7067360877990723
-13,14,0.755875,0.7338709677419355,0.659578008890152,0.6924379564100697
-14,15,0.758625,0.7217741935483871,0.6591809678077698,0.7092515037905786
-15,16,0.770125,0.7474798387096774,0.6366513178348542,0.6689942498360911
-16,17,0.76975,0.75,0.64143789935112,0.6781372427940369
-17,18,0.766625,0.7469758064516129,0.6327295961380005,0.6749444469328849
-18,19,0.772125,0.7298387096774194,0.6213552060127259,0.6950428524324971
-19,20,0.771625,0.7560483870967742,0.619529750585556,0.6620656597998834
-20,21,0.77425,0.7459677419354839,0.6170263500213623,0.6944894117693747
-21,22,0.77375,0.7379032258064516,0.610548350572586,0.698592597438443
-22,23,0.774125,0.7494959677419355,0.6073116481304168,0.6649176555295144
-23,24,0.78,0.75,0.601892656326294,0.6502023070089279
-24,25,0.777875,0.7671370967741935,0.5965238115787506,0.6416350141648324
-25,26,0.7875,0.751008064516129,0.584319313287735,0.6557927843063108
-26,27,0.784625,0.7651209677419355,0.5858220131397247,0.6265111680953733
-27,28,0.77975,0.7610887096774194,0.5928270950317382,0.6353864400617538
-28,29,0.780625,0.7520161290322581,0.5824392430782318,0.6569667564284417
-29,30,0.78475,0.7686491935483871,0.5810435285568237,0.6306867743692091
-30,31,0.789,0.7706653225806451,0.5672282783985138,0.6261125274242894
-31,32,0.78525,0.7560483870967742,0.5757509377002716,0.6505500414679127
-32,33,0.792,0.7681451612903226,0.5613697295188904,0.629849144527989
-33,34,0.793875,0.7620967741935484,0.5625830183029175,0.6189906856706066
-34,35,0.791875,0.7681451612903226,0.5602230775356293,0.6212261732547514
-35,36,0.794625,0.7615927419354839,0.5545250961780548,0.619811047469416
-36,37,0.794625,0.7701612903225806,0.5573954427242279,0.6205428954093687
-37,38,0.79425,0.764616935483871,0.5514744529724122,0.628905875067557
-38,39,0.792375,0.7842741935483871,0.5618353080749512,0.5954909113145643
-39,40,0.796375,0.7711693548387096,0.5491654114723206,0.6137347759739045
-40,41,0.799,0.7641129032258065,0.5372960684299469,0.6470560056547965
-41,42,0.800375,0.7671370967741935,0.5395989503860473,0.6211921880322118
-42,43,0.806,0.7671370967741935,0.5370515692234039,0.6075864828401997
-43,44,0.801875,0.7605846774193549,0.5388010408878326,0.5891174308715328
-44,45,0.800875,0.766633064516129,0.539761929512024,0.610026998865989
-45,46,0.802375,0.780241935483871,0.5270701496601105,0.6000283591208919
-46,47,0.799875,0.7772177419354839,0.5320828959941865,0.5864833074231302
-47,48,0.807,0.7837701612903226,0.5309389193058014,0.5761975809451072
+0,1,0.6956875,0.7605280748663101,0.8211820412079494,0.6387985139925849
+1,2,0.7645416666666667,0.787850935828877,0.6182981830437978,0.5789241027385793
+2,3,0.7837916666666667,0.7884358288770054,0.5628398391008377,0.5625918760975415
+3,4,0.7983958333333333,0.8048128342245989,0.5307413680950801,0.5154492698888727
+4,5,0.8067291666666667,0.8016377005347594,0.5082246935367585,0.5231997628900457
+5,6,0.812,0.8068181818181818,0.48933065092563627,0.516960185957465
+6,7,0.8195416666666666,0.8124164438502673,0.47495536905527114,0.4998339513406397
+7,8,0.8227708333333333,0.8236965240641712,0.46346363043785094,0.4768249858668781
+8,9,0.8255625,0.8194351604278075,0.45567453374465305,0.4821742607310494
+9,10,0.8315,0.825701871657754,0.44574370459715523,0.46175671978430316
+10,11,0.8308125,0.8133355614973262,0.4422872595389684,0.4978986528308634
+11,12,0.8348125,0.8301303475935828,0.4311412891546885,0.4576163932601398
+12,13,0.8376041666666667,0.8193516042780749,0.42240108201901116,0.4770477739247409
+13,14,0.8385625,0.8282921122994652,0.418784587085247,0.4626272647457327
+14,15,0.8436041666666667,0.8350601604278075,0.41075574096043904,0.44758253596364495
+15,16,0.8421875,0.8313001336898396,0.4109448278148969,0.446757927218223
+16,17,0.8435625,0.8346423796791443,0.40630192184448244,0.4484061913534919
+17,18,0.8458541666666667,0.8385695187165776,0.40021699021259943,0.4367518940552033
+18,19,0.847125,0.8307152406417112,0.3976554674903552,0.44405238075371095
+19,20,0.8483125,0.8386530748663101,0.3941614780028661,0.4367401426169962
+20,21,0.8502916666666667,0.8384024064171123,0.3900964806675911,0.4330310898031143
+21,22,0.850875,0.8379010695187166,0.38837515221039454,0.43586291532146737
+22,23,0.8517291666666666,0.8390708556149733,0.3842192193667094,0.43854525677341827
+23,24,0.8533958333333334,0.8319685828877005,0.37822781956195833,0.43952455710281024
+24,25,0.8521041666666667,0.8424966577540107,0.38170819969971975,0.42412006687671744
+25,26,0.8566875,0.8418282085561497,0.3745041756629944,0.42809357434351814
+26,27,0.8584583333333333,0.8440842245989305,0.3699158744017283,0.4245206856313236
+27,28,0.8557708333333334,0.841995320855615,0.37092407973607383,0.422711970892182
+28,29,0.8590625,0.8429979946524064,0.36935959541797636,0.42758023595427447
+29,30,0.8596041666666666,0.8488469251336899,0.3606020367145538,0.4196776971619397
+30,31,0.8629166666666667,0.8481784759358288,0.3595893513560295,0.41790072508355514
+31,32,0.8611875,0.8426637700534759,0.3579425945480665,0.42967644118688963
+32,33,0.8640416666666667,0.8465909090909091,0.3525094211200873,0.41836620102272953
+33,34,0.8626875,0.8418282085561497,0.3546509333451589,0.4231800782011155
+34,35,0.8648333333333333,0.8489304812834224,0.3503157667120298,0.4183035452257503
+35,36,0.8643958333333334,0.8440006684491979,0.3498730379541715,0.4190998378603216
+36,37,0.87175,0.8534425133689839,0.32715407966574034,0.4019120936406487
+37,38,0.8728541666666667,0.8485127005347594,0.32552256182829536,0.4089040935517632
+38,39,0.876375,0.852105614973262,0.32196691662073135,0.40356063794962227
+39,40,0.8771875,0.85235628342246,0.31610893772045773,0.4059840437562708
+40,41,0.8779583333333333,0.8501002673796791,0.3148737497230371,0.40458757402425144
+41,42,0.8783541666666667,0.8511029411764706,0.31425037946303686,0.40025874677507634
+42,43,0.879125,0.8485127005347594,0.31071714136004447,0.41301197538720097
+43,44,0.8840833333333333,0.8587065508021391,0.297863828599453,0.38672817749454375
+44,45,0.8851875,0.8551136363636364,0.29304139083623887,0.3945167065463602
+45,46,0.8866666666666667,0.8526069518716578,0.2949504310488701,0.4007673603009413
+46,47,0.887875,0.8566176470588235,0.2923740311563015,0.39658332620075043
+47,48,0.8865625,0.8573696524064172,0.29297807250420255,0.39060447871047543
+48,49,0.8876458333333334,0.8531918449197861,0.29198914767305056,0.4017815204227672
+49,50,0.8903125,0.8556985294117647,0.28550282220045725,0.3990818190383401
+50,51,0.8921041666666667,0.8613803475935828,0.2811181089083354,0.38583621303027965
+51,52,0.8912708333333333,0.8602941176470589,0.2825057551066081,0.38899426345519206
+52,53,0.8914375,0.8542780748663101,0.28124154203136764,0.4000103534224199
+53,54,0.8925416666666667,0.8579545454545454,0.2816228237350782,0.39209578182289306
+54,55,0.8924375,0.8584558823529411,0.2785300021370252,0.38789646502803354
+55,56,0.8932083333333334,0.8587065508021391,0.27603148049116133,0.39326987443442013
+56,57,0.8931458333333333,0.8557820855614974,0.27797147376338643,0.4011580384670094
+57,58,0.8956875,0.8595421122994652,0.2708657040695349,0.3850319666817864
+58,59,0.8921458333333333,0.8613803475935828,0.274147473325332,0.3814745011495396
+59,60,0.8947291666666667,0.8602941176470589,0.2712443074285984,0.39393253767872877
diff --git a/data/notebook2/qccnn_weights.pt b/data/notebook2/qccnn_weights.pt
index c0fead2..71029ec 100644
Binary files a/data/notebook2/qccnn_weights.pt and b/data/notebook2/qccnn_weights.pt differ
diff --git a/data/notebook2/random_qccnn_best.pt b/data/notebook2/random_qccnn_best.pt
deleted file mode 100644
index 016e9a0..0000000
Binary files a/data/notebook2/random_qccnn_best.pt and /dev/null differ
diff --git a/data/notebook2/random_qccnn_metrics.csv b/data/notebook2/random_qccnn_metrics.csv
index c29a873..5e303a0 100644
--- a/data/notebook2/random_qccnn_metrics.csv
+++ b/data/notebook2/random_qccnn_metrics.csv
@@ -1,34 +1,78 @@
 ,epoch,train_acc,valid_acc,train_loss,valid_loss
-0,1,0.561125,0.6476814516129032,1.271985122203827,0.9911825291572078
-1,2,0.676375,0.6668346774193549,0.9134431419372558,0.8937955517922679
-2,3,0.699125,0.6824596774193549,0.8243759956359863,0.8520721235582905
-3,4,0.716625,0.6995967741935484,0.7818560593128204,0.8005917687569896
-4,5,0.725125,0.7101814516129032,0.745261854171753,0.8037946531849522
-5,6,0.733375,0.7061491935483871,0.7344184167385102,0.7732808570707997
-6,7,0.737875,0.7253024193548387,0.7150477197170257,0.74149090051651
-7,8,0.74325,0.7253024193548387,0.7022454526424408,0.7516527022084882
-8,9,0.742625,0.7298387096774194,0.7041031284332275,0.7247120468847214
-9,10,0.747875,0.7116935483870968,0.6772661633491516,0.7338270487323884
-10,11,0.757875,0.7212701612903226,0.6562292041778565,0.7387931039256435
-11,12,0.761625,0.7328629032258065,0.6542983632087708,0.725050816612859
-12,13,0.75925,0.7318548387096774,0.6493379819393158,0.7001086867624714
-13,14,0.767375,0.7379032258064516,0.6460576868057251,0.6988139988914612
-14,15,0.765875,0.7288306451612904,0.6339491362571716,0.7128441218406923
-15,16,0.763125,0.7368951612903226,0.6262373595237732,0.714297366719092
-16,17,0.76925,0.7227822580645161,0.6279029569625855,0.7181522269402781
-17,18,0.77025,0.7449596774193549,0.6159816448688507,0.6757595616002237
-18,19,0.7745,0.7283266129032258,0.6136245548725128,0.6998546681096477
-19,20,0.775375,0.7469758064516129,0.6000997524261474,0.6749713065162781
-20,21,0.7805,0.748991935483871,0.5928600332736969,0.6656959902855658
-21,22,0.77525,0.7444556451612904,0.599046837568283,0.6857193170055267
-22,23,0.785,0.7384072580645161,0.5875316572189331,0.6785462142959717
-23,24,0.778375,0.765625,0.588378502368927,0.627939272311426
-24,25,0.78575,0.7459677419354839,0.5700427904129028,0.6502643679418871
-25,26,0.7805,0.75,0.5785817315578461,0.6712307555060233
-26,27,0.78675,0.7474798387096774,0.5676946561336518,0.6555941143343526
-27,28,0.787875,0.7505040322580645,0.575938116312027,0.6507430134281036
-28,29,0.789,0.7605846774193549,0.5655779435634612,0.6520830373610219
-29,30,0.790625,0.7434475806451613,0.5578647639751434,0.6789213486256138
-30,31,0.787375,0.7565524193548387,0.5687701859474182,0.649224087115257
-31,32,0.79575,0.7560483870967742,0.5432633152008056,0.6388471222692921
-32,33,0.7895,0.7595766129032258,0.5557173223495483,0.6391933618053314
+0,1,0.6058333333333333,0.7088068181818182,1.077100011587143,0.7878920158600424
+1,2,0.7224166666666667,0.7269385026737968,0.7470338317950567,0.7390924543939172
+2,3,0.7520208333333334,0.7539271390374331,0.6676628764470418,0.6713398865518723
+3,4,0.7725208333333333,0.7676303475935828,0.612583932240804,0.6310215724662026
+4,5,0.7787916666666667,0.7807486631016043,0.5905240491827329,0.594515474244235
+5,6,0.7867083333333333,0.7794117647058824,0.568865264693896,0.5900490468836086
+6,7,0.7952708333333334,0.7901069518716578,0.5486233344872793,0.553959414283222
+7,8,0.7990416666666667,0.7973763368983957,0.5307548115452131,0.5483398674962355
+8,9,0.8028958333333334,0.7942847593582888,0.5225638653437297,0.553332871613018
+9,10,0.808875,0.7990474598930482,0.5071384527683258,0.5478345090373952
+10,11,0.8109166666666666,0.8043114973262032,0.49932783031463623,0.5302995911893997
+11,12,0.8152083333333333,0.7998830213903744,0.4876467382510503,0.5369620672203003
+12,13,0.8218333333333333,0.8055648395721925,0.47221228990952174,0.5228660841357899
+13,14,0.818875,0.8045621657754011,0.47462575109799704,0.5175498661829189
+14,15,0.8245625,0.8102439839572193,0.46200620504220324,0.5206660962997274
+15,16,0.8271875,0.817346256684492,0.4543743284543355,0.4972638836837707
+16,17,0.8277916666666667,0.8086564171122995,0.45034414579470955,0.5051890009227283
+17,18,0.827625,0.8130848930481284,0.445495897213618,0.5063204698384127
+18,19,0.8327291666666666,0.8187667112299465,0.4373013471563657,0.49322714222306235
+19,20,0.835875,0.8227774064171123,0.428063592116038,0.48620352627121827
+20,21,0.8376875,0.825701871657754,0.42721951069434483,0.4824375149241106
+21,22,0.8393333333333334,0.8237800802139037,0.424452906926473,0.47128429227971774
+22,23,0.8394791666666667,0.8235294117647058,0.4193731239239375,0.4921083778621041
+23,24,0.8411041666666667,0.8219418449197861,0.4158371254205704,0.4766464689995516
+24,25,0.8403958333333333,0.8226102941176471,0.4133129887978236,0.4774917462133469
+25,26,0.8443541666666666,0.8231116310160428,0.40567960069576897,0.48197355627376126
+26,27,0.8448958333333333,0.8270387700534759,0.4060484048128128,0.48230106714256304
+27,28,0.8460625,0.8224431818181818,0.403180761963129,0.48200917403328225
+28,29,0.8495416666666666,0.8254512032085561,0.39154566899935406,0.4732334581925907
+29,30,0.849375,0.8205213903743316,0.3910754985809326,0.5025412492095468
+30,31,0.850125,0.8200200534759359,0.3879919012586276,0.48824962750475676
+31,32,0.8501666666666666,0.8283756684491979,0.3852265829642614,0.4738096959769407
+32,33,0.8533125,0.8280414438502673,0.3842975225051244,0.480888257131857
+33,34,0.8546666666666667,0.8279578877005348,0.3802658775647481,0.4736704102015113
+34,35,0.8553541666666666,0.8287934491978609,0.37693077965577443,0.47027840429448825
+35,36,0.8574583333333333,0.8280414438502673,0.37416908142964045,0.47105003949155144
+36,37,0.8575416666666666,0.8333890374331551,0.370725477874279,0.4729995278113666
+37,38,0.8593333333333333,0.8318850267379679,0.3660616307258606,0.46004778090964027
+38,39,0.86125,0.8230280748663101,0.36278441416223844,0.4792340671155542
+39,40,0.8597291666666667,0.8328877005347594,0.36306345409154894,0.4713420571490405
+40,41,0.8594583333333333,0.8338903743315508,0.35881871738036475,0.457705247131261
+41,42,0.86125,0.8338903743315508,0.35548474109172823,0.4744296157583196
+42,43,0.8638958333333333,0.8328877005347594,0.35729147561391195,0.46916984149160235
+43,44,0.864375,0.8360628342245989,0.3560072028040886,0.45340159527439483
+44,45,0.8649791666666666,0.829879679144385,0.35113127034902575,0.4705522551256068
+45,46,0.8642708333333333,0.832302807486631,0.3466692521572113,0.4682676112428706
+46,47,0.8671458333333333,0.8295454545454546,0.34586968237161636,0.4741511169601889
+47,48,0.867125,0.8333890374331551,0.3455539164741834,0.47975461392160407
+48,49,0.8672916666666667,0.8318014705882353,0.34170353756348293,0.4724194825811182
+49,50,0.8685416666666667,0.830548128342246,0.33824845480918886,0.47551511706196686
+50,51,0.8786875,0.8408255347593583,0.31427997442086536,0.45161250855195967
+51,52,0.8811666666666667,0.8393215240641712,0.3106038907766342,0.45226032346646416
+52,53,0.881875,0.8417446524064172,0.3050905155738195,0.44453552421082787
+53,54,0.8818958333333333,0.8420788770053476,0.30438282175858816,0.45232303249325984
+54,55,0.8853958333333334,0.8393215240641712,0.29754234209656716,0.45854009632120796
+55,56,0.8851041666666667,0.8404913101604278,0.29953089368343355,0.4524733354063595
+56,57,0.8845625,0.8411597593582888,0.29662314279874163,0.45404982789952486
+57,58,0.8874166666666666,0.8431651069518716,0.29427229724327725,0.45166250935850294
+58,59,0.8859375,0.8394050802139037,0.29326443988084794,0.46354111527057895
+59,60,0.888375,0.8429979946524064,0.2878838664491971,0.4655548495086119
+60,61,0.8886666666666667,0.8424966577540107,0.2877097588280837,0.45055208335267033
+61,62,0.889125,0.8429144385026738,0.2842213006118933,0.4635044741917421
+62,63,0.8885625,0.8435828877005348,0.2865720265607039,0.4594836203172245
+63,64,0.8917708333333333,0.8379010695187166,0.285613534172376,0.4660476203111404
+64,65,0.8910625,0.8383188502673797,0.28201638598243395,0.4701039531332924
+65,66,0.8908958333333333,0.8407419786096256,0.2792068747878075,0.47005820457629344
+66,67,0.8923333333333333,0.8440006684491979,0.2773614428540071,0.4646621039685081
+67,68,0.8912291666666666,0.8459224598930482,0.28061300626397134,0.45556305787142587
+68,69,0.8947083333333333,0.8403241978609626,0.27382597197095554,0.4730077214578894
+69,70,0.8913958333333334,0.8402406417112299,0.2796522062718868,0.4642943452226924
+70,71,0.8939375,0.8412433155080213,0.27323526923855146,0.4593185937022143
+71,72,0.8947708333333333,0.841326871657754,0.2720908187329769,0.4730970057892927
+72,73,0.8960208333333334,0.8417446524064172,0.26792943899333477,0.47570018247168333
+73,74,0.8949791666666667,0.8415775401069518,0.27027500013510386,0.4757912041828594
+74,75,0.9015208333333333,0.8446691176470589,0.257486442198356,0.4704125145858622
+75,76,0.9023333333333333,0.8451704545454546,0.25169081447521846,0.46167185049962234
+76,77,0.9014583333333334,0.8432486631016043,0.2498825814574957,0.462763199672342
diff --git a/data/notebook2/random_qccnn_weights.pt b/data/notebook2/random_qccnn_weights.pt
index e62e8ca..0fee8df 100644
Binary files a/data/notebook2/random_qccnn_weights.pt and b/data/notebook2/random_qccnn_weights.pt differ
diff --git a/data/notebook2/vgg_best.pt b/data/notebook2/vgg_best.pt
deleted file mode 100644
index 787171d..0000000
Binary files a/data/notebook2/vgg_best.pt and /dev/null differ
diff --git a/data/notebook2/vgg_metrics.csv b/data/notebook2/vgg_metrics.csv
index 4631c19..47dd89c 100644
--- a/data/notebook2/vgg_metrics.csv
+++ b/data/notebook2/vgg_metrics.csv
@@ -1,86 +1,101 @@
 ,epoch,train_acc,valid_acc,train_loss,valid_loss
-0,1,0.575125,0.7011088709677419,2.031563010215759,1.8535375249001287
-1,2,0.723875,0.7515120967741935,1.7430778923034669,1.713461822079074
-2,3,0.76425,0.7525201612903226,1.6991203842163085,1.7051894049490652
-3,4,0.779875,0.7716733870967742,1.6807003259658813,1.690080665772961
-4,5,0.79525,0.7696572580645161,1.6654855661392212,1.6926762519344207
-5,6,0.80175,0.7888104838709677,1.6598792200088501,1.6711357831954956
-6,7,0.80875,0.7721774193548387,1.6529147624969482,1.6847506146277151
-7,8,0.812375,0.8004032258064516,1.649049828529358,1.6597587523921844
-8,9,0.815375,0.7893145161290323,1.6458229179382324,1.6707303101016628
-9,10,0.825,0.811491935483871,1.6370085287094116,1.6518691432091497
-10,11,0.832375,0.795866935483871,1.630222158432007,1.6616583139665666
-11,12,0.83225,0.8145161290322581,1.6293829317092896,1.6454557026586225
-12,13,0.834875,0.8125,1.6268642024993896,1.6500283018235238
-13,14,0.840875,0.8104838709677419,1.6199358901977539,1.652194688397069
-14,15,0.83975,0.8069556451612904,1.6210605192184449,1.6520756982987927
-15,16,0.84175,0.8004032258064516,1.6195698108673096,1.6582853217278757
-16,17,0.838625,0.8049395161290323,1.6228710346221924,1.656198097813514
-17,18,0.83775,0.8140120967741935,1.6226149969100951,1.6462942131104008
-18,19,0.845375,0.8251008064516129,1.6150365982055663,1.6358921681680987
-19,20,0.85125,0.8296370967741935,1.6097205333709717,1.6333312757553593
-20,21,0.851875,0.8210685483870968,1.609424132347107,1.6388044895664338
-21,22,0.850375,0.8210685483870968,1.6101643466949462,1.6369191561975787
-22,23,0.858125,0.8185483870967742,1.6031983375549317,1.6421872877305554
-23,24,0.857125,0.8361895161290323,1.6038059520721435,1.62519553015309
-24,25,0.854,0.8301411290322581,1.606280044555664,1.6290803609355804
-25,26,0.858375,0.8119959677419355,1.6011180095672608,1.648349738890125
-26,27,0.861375,0.8371975806451613,1.6000273780822754,1.6237776010267195
-27,28,0.8705,0.8351814516129032,1.5907322645187378,1.6240986316434798
-28,29,0.87175,0.828125,1.5896439476013184,1.6316479982868317
-29,30,0.867375,0.8266129032258065,1.5936143741607667,1.632056209348863
-30,31,0.86575,0.8377016129032258,1.5949567575454713,1.6240303862479426
-31,32,0.86975,0.8336693548387096,1.5907692136764526,1.6251203167823054
-32,33,0.86975,0.8397177419354839,1.5919580411911012,1.6214879328204739
-33,34,0.872375,0.8422379032258065,1.5886561880111694,1.6194973222671016
-34,35,0.876,0.8331653225806451,1.585767653465271,1.6247131862948019
-35,36,0.871875,0.8392137096774194,1.5895709590911866,1.6215527211466143
-36,37,0.871125,0.8245967741935484,1.590462643623352,1.6370407227546937
-37,38,0.87325,0.8301411290322581,1.5876847248077393,1.6306157150576193
-38,39,0.8705,0.8240927419354839,1.590844289779663,1.634965923524672
-39,40,0.88125,0.8402217741935484,1.5803495874404907,1.6220719968118975
-40,41,0.881875,0.8462701612903226,1.5788077726364136,1.614263488400367
-41,42,0.87625,0.8487903225806451,1.5843929862976074,1.6131737078389814
-42,43,0.88175,0.842741935483871,1.5789586782455445,1.6184405088424683
-43,44,0.880375,0.8417338709677419,1.5802030544281005,1.6189708786626016
-44,45,0.880375,0.8492943548387096,1.5803771505355835,1.610948174230514
-45,46,0.877625,0.8523185483870968,1.5828620948791503,1.6089561793111986
-46,47,0.882125,0.8442540322580645,1.578727219581604,1.6160847948443504
-47,48,0.872875,0.8392137096774194,1.5874832077026366,1.6228746021947553
-48,49,0.884125,0.8371975806451613,1.5771635084152222,1.6237398655183855
-49,50,0.88975,0.8341733870967742,1.5712118272781372,1.6256768280459988
-50,51,0.876,0.8432459677419355,1.585065812110901,1.6188401445265739
-51,52,0.883,0.844758064516129,1.5785351629257203,1.6169025359615203
-52,53,0.889125,0.8548387096774194,1.5723771095275878,1.607426397262081
-53,54,0.88625,0.8518145161290323,1.5749757404327394,1.6076742141477522
-54,55,0.891125,0.8452620967741935,1.5695797872543336,1.6156867473356185
-55,56,0.891625,0.8497983870967742,1.5691107511520386,1.6118658050414054
-56,57,0.883375,0.8508064516129032,1.577266471862793,1.6087883749315817
-57,58,0.891625,0.8422379032258065,1.5702907581329346,1.6168161553721274
-58,59,0.89375,0.8553427419354839,1.5679762859344482,1.6071604336461713
-59,60,0.885875,0.8482862903225806,1.5749410953521727,1.6108381056016492
-60,61,0.891875,0.8417338709677419,1.5696327953338622,1.6187968907817718
-61,62,0.894,0.8392137096774194,1.5671770153045654,1.6208721822307957
-62,63,0.891125,0.8392137096774194,1.569726734161377,1.6224797733368412
-63,64,0.89325,0.8518145161290323,1.568245210647583,1.606977516605008
-64,65,0.895125,0.8462701612903226,1.5658181638717652,1.6140611633177726
-65,66,0.89175,0.8568548387096774,1.5686321334838866,1.6044510756769488
-66,67,0.896375,0.8412298387096774,1.5650024271011354,1.6208048520549652
-67,68,0.8895,0.8477822580645161,1.571241024017334,1.6136699338113107
-68,69,0.901625,0.8472782258064516,1.5600450325012207,1.612250724146443
-69,70,0.89025,0.8306451612903226,1.5705311574935914,1.630635638390818
-70,71,0.89675,0.8538306451612904,1.5637341051101685,1.6056317244806597
-71,72,0.897,0.8412298387096774,1.5633089447021484,1.6192954970944313
-72,73,0.900375,0.8503024193548387,1.560647201538086,1.6097132775091356
-73,74,0.895625,0.8533266129032258,1.565330421447754,1.6075265484471475
-74,75,0.895125,0.8543346774193549,1.565690894126892,1.6060891266792052
-75,76,0.900625,0.8608870967741935,1.560530616760254,1.6011101891917567
-76,77,0.897125,0.8563508064516129,1.563725378036499,1.6037284789546844
-77,78,0.89175,0.8482862903225806,1.5693257989883422,1.6122698245509979
-78,79,0.90225,0.8487903225806451,1.5589981718063355,1.6111104180735927
-79,80,0.893375,0.8442540322580645,1.5669814805984497,1.6161983090062295
-80,81,0.897125,0.8573588709677419,1.56400607585907,1.6040064134905416
-81,82,0.898,0.8568548387096774,1.5621892538070679,1.6044225615839804
-82,83,0.89075,0.8503024193548387,1.5697740039825439,1.6114465228972896
-83,84,0.900625,0.8598790322580645,1.5606124105453492,1.6002429416102748
-84,85,0.89825,0.8553427419354839,1.5628600664138794,1.603792409743032
+0,1,0.767875,0.8360628342245989,1.753534612496694,1.6362467181873832
+1,2,0.8333125,0.8577038770053476,1.6276749771436057,1.6030110356641962
+2,3,0.8611875,0.8703208556149733,1.60025262260437,1.5910939386184202
+3,4,0.8715833333333334,0.8587065508021391,1.5890283988316853,1.601965536408246
+4,5,0.8776041666666666,0.8759191176470589,1.5827694902420044,1.5844863908176117
+5,6,0.8843958333333334,0.8757520053475936,1.5761419967015584,1.5849310975661253
+6,7,0.88675,0.8812667112299465,1.573840765953064,1.5789309529697193
+7,8,0.8909375,0.8832720588235294,1.569781884988149,1.5779090036045422
+8,9,0.8903333333333333,0.8875334224598931,1.570390274365743,1.5736284517349406
+9,10,0.8941458333333333,0.890625,1.5664398628870646,1.5700276994450206
+10,11,0.8940208333333334,0.8915441176470589,1.5668807325363159,1.5694875258175447
+11,12,0.8958958333333333,0.8737466577540107,1.5646815473238627,1.5868047845554862
+12,13,0.8980625,0.8921290106951871,1.5629661572774252,1.5691035067971377
+13,14,0.900625,0.8943850267379679,1.560197625319163,1.566351128771981
+14,15,0.8988125,0.8884525401069518,1.561814420223236,1.572067444337243
+15,16,0.8991041666666667,0.8928810160427807,1.5614849192301432,1.5675410924748303
+16,17,0.9010208333333334,0.8932152406417112,1.5596694407463074,1.5670097658340945
+17,18,0.9034583333333334,0.8913770053475936,1.5576222189267477,1.56979562700751
+18,19,0.9042083333333333,0.8852774064171123,1.5566549801826477,1.5758734877734262
+19,20,0.9005833333333333,0.890625,1.560306779384613,1.5699780733190118
+20,21,0.91125,0.9029913101604278,1.5499087982177735,1.5577653244854932
+21,22,0.9131041666666667,0.906667780748663,1.54771670850118,1.5544780492782593
+22,23,0.9148125,0.9045788770053476,1.5459488903681438,1.5563749999286018
+23,24,0.9155833333333333,0.9033255347593583,1.5449249391555786,1.5577488563914987
+24,25,0.9170208333333333,0.9065842245989305,1.5440529939333598,1.5541542717479766
+25,26,0.9180208333333333,0.9048295454545454,1.5431619186401366,1.5560473452277361
+26,27,0.9175208333333333,0.9046624331550802,1.5432857065200805,1.5563587473038045
+27,28,0.9199583333333333,0.9085895721925134,1.5409181524912516,1.5526812178565856
+28,29,0.9184583333333334,0.9052473262032086,1.5422742649714152,1.555272618079568
+29,30,0.9212291666666667,0.9065006684491979,1.53969158522288,1.554516376021074
+30,31,0.9207916666666667,0.9100935828877005,1.5402175769805908,1.5506398645951787
+31,32,0.9214375,0.9079211229946524,1.539174134572347,1.5536619998554495
+32,33,0.9203333333333333,0.9089237967914439,1.540758513132731,1.5520596727330418
+33,34,0.920625,0.9049966577540107,1.5403572120666504,1.5559145416167968
+34,35,0.9203125,0.9095086898395722,1.5404717030525208,1.551519107053624
+35,36,0.9235,0.908673128342246,1.537489187558492,1.5522340115378885
+36,37,0.9236666666666666,0.9076704545454546,1.5371040493647257,1.5535466779362073
+37,38,0.9271458333333333,0.9108455882352942,1.5338874877293904,1.550052744182036
+38,39,0.928375,0.9119318181818182,1.5327118910153708,1.5489872643016875
+39,40,0.9302916666666666,0.9157754010695187,1.530610190709432,1.5451570577162472
+40,41,0.9313541666666667,0.9134358288770054,1.5297347887357076,1.5474902193814037
+41,42,0.9315208333333334,0.9144385026737968,1.5294018096923827,1.546497144163611
+42,43,0.9323125,0.9160260695187166,1.5288680585225423,1.5444984174667196
+43,44,0.9319791666666667,0.9151069518716578,1.5289348866144816,1.5459555961231497
+44,45,0.9333541666666667,0.9186163101604278,1.5275377634366354,1.5426336552370041
+45,46,0.932375,0.9125167112299465,1.528664146900177,1.5481219285312184
+46,47,0.932375,0.9180314171122995,1.528616013844808,1.543393959973585
+47,48,0.9353541666666667,0.9174465240641712,1.5259391600290935,1.5431863923761295
+48,49,0.9332291666666667,0.9116811497326203,1.527786935488383,1.5487734449101005
+49,50,0.9352916666666666,0.9159425133689839,1.525529363632202,1.5448988389203893
+50,51,0.9351666666666667,0.9190340909090909,1.5257512016296386,1.541433523682987
+51,52,0.9358125,0.9166945187165776,1.525342579046885,1.5442416878307568
+52,53,0.9355625,0.9197860962566845,1.5253059350649516,1.5413460323517336
+53,54,0.936625,0.9142713903743316,1.5245349135398865,1.5465437422461688
+54,55,0.9349791666666667,0.9161931818181818,1.525785480340322,1.5452430726372621
+55,56,0.9368333333333333,0.9174465240641712,1.524029705842336,1.5432787445139757
+56,57,0.9355833333333333,0.9158589572192514,1.5253107420603433,1.5446410478755115
+57,58,0.9371875,0.9207052139037433,1.524094797929128,1.5405435536634475
+58,59,0.9376458333333333,0.9199532085561497,1.5234766112963358,1.5406819996349315
+59,60,0.9391875,0.9189505347593583,1.5217385093371074,1.5418526864944295
+60,61,0.9380208333333333,0.9172794117647058,1.5229359103838602,1.54353828162433
+61,62,0.9387708333333333,0.9206216577540107,1.5224510963757834,1.5403235111644562
+62,63,0.93975,0.9156082887700535,1.5212798911730447,1.545105798996706
+63,64,0.9397708333333333,0.921875,1.5213380891482036,1.53904527202647
+64,65,0.9388541666666667,0.9197860962566845,1.5220589057604472,1.5410850934166322
+65,66,0.9384166666666667,0.9163602941176471,1.5225912384986877,1.5446943541899085
+66,67,0.9405,0.9212901069518716,1.5205388652483622,1.539675323083439
+67,68,0.9412291666666667,0.9225434491978609,1.519922403494517,1.53815110800738
+68,69,0.9401041666666666,0.9145220588235294,1.520717616558075,1.5464129702930145
+69,70,0.9411875,0.921875,1.5197122866312662,1.5391668482897753
+70,71,0.94,0.9188669786096256,1.5209512915611267,1.5418487353758379
+71,72,0.9394583333333333,0.9202874331550802,1.5213834778467814,1.540625613003491
+72,73,0.9419583333333333,0.9192012032085561,1.5191686747868856,1.5421639017880282
+73,74,0.9420208333333333,0.9211229946524064,1.5191353003184,1.539851359505067
+74,75,0.9431458333333333,0.9225434491978609,1.517710240205129,1.5381846899654776
+75,76,0.9452708333333333,0.9207887700534759,1.5158586444854736,1.5403157555483242
+76,77,0.9455625,0.9209558823529411,1.5154005877176921,1.539823994917028
+77,78,0.9458333333333333,0.9228776737967914,1.515006558418274,1.5379077584985743
+78,79,0.9461041666666666,0.9209558823529411,1.5149241960843405,1.5400247025617304
+79,80,0.9460208333333333,0.9239639037433155,1.515042960802714,1.5368424086647237
+80,81,0.9453541666666667,0.9212065508021391,1.515584836324056,1.5400443440452616
+81,82,0.9463958333333333,0.9239639037433155,1.5146392253239949,1.5368295028247936
+82,83,0.9471041666666666,0.9226270053475936,1.514121623357137,1.5383771377451279
+83,84,0.946625,0.9215407754010695,1.5143028028806051,1.5394194024131898
+84,85,0.9472916666666666,0.9249665775401069,1.5138665714263917,1.5362410226607706
+85,86,0.9468958333333334,0.9239639037433155,1.5140059588750203,1.5370322306525899
+86,87,0.947625,0.9238803475935828,1.513320836544037,1.537034041741315
+87,88,0.948375,0.9229612299465241,1.5128038795789083,1.5373781154499973
+88,89,0.9495,0.9250501336898396,1.5117765822410583,1.5355126124652312
+89,90,0.9491458333333334,0.9232118983957219,1.512057636419932,1.537534661471525
+90,91,0.9485,0.9240474598930482,1.5127018653551738,1.537239743426522
+91,92,0.9492083333333333,0.921457219251337,1.5120360770225525,1.53860371316818
+92,93,0.95,0.9224598930481284,1.5111351316769919,1.5382829611314173
+93,94,0.9503125,0.9248830213903744,1.510804171562195,1.5363529123724464
+94,95,0.9513333333333334,0.9275568181818182,1.509633824825287,1.533667146840835
+95,96,0.9501041666666666,0.9229612299465241,1.5108498128255208,1.5375941335198713
+96,97,0.9505833333333333,0.9207887700534759,1.5104362696011862,1.5394581116457036
+97,98,0.95125,0.926052807486631,1.5097919216156006,1.5350576141938805
+98,99,0.950875,0.9257185828877005,1.5103401794433593,1.5352602616988402
+99,100,0.9509166666666666,0.9251336898395722,1.5100935606956483,1.5362022280055572
diff --git a/data/notebook2/vgg_weights.pt b/data/notebook2/vgg_weights.pt
index 9c08948..c157924 100644
Binary files a/data/notebook2/vgg_weights.pt and b/data/notebook2/vgg_weights.pt differ